Capacity Confounds and Coverage Guarantees in Adaptive Sub-model Federated Learning
Not on the map. Why?
Abstract (arXiv)
Sub-model federated learning lets resource-constrained clients train width-reduced versions of a global model, but existing methods allocate capacity by device resources alone. A natural next step, allocating capacity by each client's data heterogeneity as estimated from the updates the server already observes, is suggested by recent methods that size sub-models from training-derived signals. We ask whether that step is possible, using HAS-FL, an adaptive capacity-allocation framework, as a test case. First, validated against ground-truth label-distribution divergence on reproducible partitions, update-divergence estimates of client heterogeneity are dominated by capacity rather than data: on both image benchmarks and every seed, the estimates correlate strongly and negatively with device capacity, and once capacity is controlled for their association with data heterogeneity is near zero or negative. Any method estimating client statistics from sub-model updates is exposed to this previously undocumented confound. Second, adaptive allocation has a hidden failure mode: when every client is capped below full width, the uncovered parameters stay at random initialization and progressively corrupt the global model. A simple coverage guarantee removes the failure and explains why uniform allocation collapses. Third, a matched-budget control settles what adaptivity contributes: random allocation to the same average budget matches the adaptive policy to within seed-to-seed variation on both image benchmarks, and on the naturally partitioned text benchmark the adaptive policy is the weakest of the three strategies while consuming the most capacity. Sub-model training admits constrained clients at quadratically reduced cost but gives up substantial accuracy relative to full-model training, and what protects that accuracy is parameter coverage rather than allocation intelligence.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 48 of its 65 references that are in the map. First posted on arXiv 7 Aug 2026.
Builds on
48 of 65 references.
References · 48
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2019Advances and Open Problems in Federated Learningcited by 3,046
- 2018Federated Optimization in Heterogeneous Networkscited by 2,987
- 2019Federated Learning: Challenges, Methods, and Future Directionscited by 1,538
- 2019On the Convergence of FedAvg on Non-IID Datacited by 1,229
- 2020Adaptive Federated Optimizationcited by 1,046
- 2020Tackling the Objective Inconsistency Problem in Heterogeneous Federated Optimizationcited by 906
- 2018LEAF: A Benchmark for Federated Settingscited by 792
- 2020Ensemble Distillation for Robust Model Fusion in Federated Learningcited by 682
- 2020Ditto: Fair and Robust Federated Learning Through Personalizationcited by 672
- 2020The Future of Digital Health with Federated Learningcited by 633
- 2020Personalized Federated Learning with Moreau Envelopescited by 618
36 more
- 2018Federated Learning for Mobile Keyboard Predictioncited by 572
- 2021Federated Learning on Non-IID Data Silos: An Experimental Studycited by 568
- 2021Exploiting Shared Representations for Personalized Federated Learningcited by 541
- 2020An Efficient Framework for Clustered Federated Learningcited by 511
- 2019Clustered Federated Learning: Model-Agnostic Distributed Multi-Task Optimization under Privacy Constraintscited by 503
- 2020Adaptive Personalized Federated Learningcited by 358
- 2019FedPAQ: A Communication-Efficient Federated Learning Method with Periodic Averaging and Quantizationcited by 356
- 2021FedProto: Federated Prototype Learning across Heterogeneous Clientscited by 352
- 2021Federated Learning on Non-IID Data: A Surveycited by 346
- 2020HeteroFL: Computation and Communication Efficient Federated Learning for Heterogeneous Clientscited by 337
- 2017Are GANs Created Equal? A Large-Scale Studycited by 267
- 2017The Marginal Value of Adaptive Gradient Methods in Machine Learningcited by 250
- 2017Sparse Communication for Distributed Gradient Descentcited by 249
- 2019PowerSGD: Practical Low-Rank Gradient Compression for Distributed Optimizationcited by 238
- 2019Model Pruning Enables Efficient Federated Learning on Edge Devicescited by 222
- 2021FjORD: Fair and Accurate Federated Learning under heterogeneous targets with Ordered Dropoutcited by 192
- 2021FedKD: Communication Efficient Federated Learning via Knowledge Distillationcited by 187
- 2021Federated Learning for Smart Healthcare: A Surveycited by 184
- 2020FetchSGD: Communication-Efficient Federated Learning with Sketchingcited by 179
- 2023Heterogeneous Federated Learning: State-of-the-art and Research Challengescited by 161
- 2020Federated Learning with Compression: Unified Analysis and Sharp Guaranteescited by 160
- 2021FedPara: Low-Rank Hadamard Product for Communication-Efficient Federated Learningcited by 141
- 2020Federated Learning for Internet of Things: Recent Advances, Taxonomy, and Open Challengescited by 139
- 2022FedRolex: Model-Heterogeneous Federated Learning with Rolling Sub-Model Extractioncited by 102
- 2023Personalized Federated Learning with Feature Alignment and Classifier Collaborationcited by 97
- 2020FedGroup: Efficient Clustered Federated Learning via Decomposed Data-Driven Measurecited by 46
- 2022FedBalancer: Data and Pace Control for Efficient Federated Learning on Heterogeneous Clientscited by 46
- 2021AutoFL: Enabling Heterogeneity-Aware Energy Efficient Federated Learningcited by 39
- 2023Efficient Personalized Federated Learning via Sparse Model-Adaptationcited by 38
- 2023FedCP: Separating Feature Information for Personalized Federated Learning via Conditional Policycited by 36
- 2024FIARSE: Model-Heterogeneous Federated Learning via Importance-Aware Submodel Extractioncited by 21
- 2021Client Selection in Federated Learning based on Gradients Importancecited by 9
- 2024Improving Local Training in Federated Learning via Temperature Scalingcited by 5
- 2024Learnable Sparse Customization in Heterogeneous Edge Computingcited by 2
- 2025Towards One-shot Federated Learning: Advances, Challenges, and Future Directionscited by 2
- 2025On the Convergence and Stability of Distributed Sub-model Trainingcited by 1
Built on by
Not cited yet.