Parameter-Efficient Personalized Federated Learning for Accurate Cellular Traffic Prediction

Abstract

Federated learning (FL) enables cellular traffic prediction without centralizing raw base-station data, but statistical heterogeneity makes a single global model unsuitable for many clients. This paper proposes federated clustering with adaptive personalization (FedCAP), a parameter-efficient personalized FL framework that separates cluster-level representation learning from client-level adaptation. Clients are grouped using training-only daily traffic profiles, after which an LSTM backbone is trained by FedAvg within each cluster. Each client then freezes the cluster backbone and optimizes a residual bottleneck adapter locally. The adapter contains 4241 trainable parameters, 6.19% of the 68,483-parameter three-feature backbone and prediction head, and personalization transmits no model updates. In a shared-seed-42 comparison across 11 methods and four public datasets, FedCAP ranks first or second in 12 of 16 dataset–metric combinations. Across five shared seeds, its mean MAE is 6.66%, 7.47%, 2.64%, and 3.86% below FedAvg on the Milan, Trentino, Bihar, and Taiwan datasets, respectively. Holm-adjusted paired t-tests identify 6 significant dataset–metric differences, whereas exact Wilcoxon tests are not significant because each comparison contains only five nonzero seed-matched pairs; the statistical evidence is therefore interpreted conservatively.