Effective Heterogeneous Federated Learning via Efficient Hypernetwork-based Weight Generation
Not on the map. Why?
Abstract (arXiv)
While federated learning leverages distributed client resources, it faces challenges due to heterogeneous client capabilities. This necessitates allocating models suited to clients' resources and careful parameter aggregation to accommodate this heterogeneity. We propose HypeMeFed, a novel federated learning framework for supporting client heterogeneity by combining a multi-exit network architecture with hypernetwork-based model weight generation. This approach aligns the feature spaces of heterogeneous model layers and resolves per-layer information disparity during weight aggregation. To practically realize HypeMeFed, we also propose a low-rank factorization approach to minimize computation and memory overhead associated with hypernetworks. Our evaluations on a real-world heterogeneous device testbed indicate that system enhances accuracy by 5.12% over FedAvg, reduces the hypernetwork memory requirements by 98.22%, and accelerates its operations by 1.86x compared to a naive hypernetwork approach. These results demonstrate HypeMeFed's effectiveness in leveraging and engaging heterogeneous clients for federated learning.
Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Builds on 42 of its 77 references that are in the map. First posted on arXiv 3 Jul 2024.
Builds on
42 of 77 references.
References · 42
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2018Federated Optimization in Heterogeneous Networkscited by 2,971
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
- 2020Implicit Neural Representations with Periodic Activation Functionscited by 2,214
- 2019Similarity of Neural Network Representations Revisitedcited by 1,695
- 2016HyperNetworkscited by 1,003
- 2019Towards Federated Learning at Scale: System Designcited by 826
30 more
- 2017Channel Pruning for Accelerating Very Deep Neural Networkscited by 697
- 2020Ensemble Distillation for Robust Model Fusion in Federated Learningcited by 678
- 2017BranchyNet: Fast Inference via Early Exiting from Deep Neural Networkscited by 591
- 2019FedMD: Heterogenous Federated Learning via Model Distillationcited by 454
- 2018Snips Voice Platform: an embedded Spoken Language Understanding system for private-by-design voice interfacescited by 376
- 2020HeteroFL: Computation and Communication Efficient Federated Learning for Heterogeneous Clientscited by 334
- 2016Dynamic Filter Networkscited by 318
- 2021No Fear of Heterogeneity: Classifier Calibration for Federated Learning with Non-IID Datacited by 241
- 2021Personalized Federated Learning using Hypernetworkscited by 210
- 2021FjORD: Fair and Accurate Federated Learning under heterogeneous targets with Ordered Dropoutcited by 192
- 2022On-Device Training Under 256KB Memorycited by 158
- 2022FedRolex: Model-Heterogeneous Federated Learning with Rolling Sub-Model Extractioncited by 102
- 2023A Brief Review of Hypernetworks in Deep Learningcited by 102
- 2020Gradient Flow in Sparse Neural Networks and How Lottery Tickets Wincited by 77
- 2021Adaptive Inference through Early-Exit Networks: Design, Challenges and Directionscited by 74
- 2019Hypernetwork functional image representationcited by 65
- 2020LotteryFL: Personalized and Communication-Efficient Federated Learning with Lottery Ticket Hypothesis on Non-IID Datasetscited by 60
- 2016UniMiB SHAR: a new dataset for human activity recognition using acceleration data from smartphonescited by 54
- 2019Deep Meta Functionals for Shape Representationcited by 53
- 2022FedBalancer: Data and Pace Control for Efficient Federated Learning on Heterogeneous Clientscited by 46
- 2022Orchestra: Unsupervised Federated Learning via Globally Consistent Clusteringcited by 45
- 2021FedHM: Efficient Federated Learning for Heterogeneous Models via Low-rank Factorizationcited by 31
- 2022HINT: Hypernetwork Instruction Tuning for Efficient Zero- & Few-Shot Generalisationcited by 28
- 2021It's always personal: Using Early Exits for Efficient On-Device CNN Personalisationcited by 18
- 2022Federated Learning with Heterogeneous Architectures using Graph HyperNetworkscited by 17
- 2024MESEN: Exploit Multimodal Data to Design Unimodal Human Activity Recognition with Few Labelscited by 11
- 2024Recurrent Early Exits for Federated Learning with Heterogeneous Clientscited by 11
- 2022FLAME: Federated Learning Across Multi-device Environmentscited by 8
- 2024Improving Local Training in Federated Learning via Temperature Scalingcited by 5
- 2023Federated Learning Over Images: Vertical Decompositions and Pre-Trained Backbones Are Difficult to Beatcited by 4
Built on by
Citing papers per year: 2024: 3; 2025: 6.
9 papers.
- 2025One-shot Federated Learning Methods: A Practical Guidecited by 6
- 2024DeTrigger: A Gradient-Centric Approach to Backdoor Attack Mitigation in Federated Learningcited by 5
- 2024FedAli: Personalized Federated Learning Alignment with Prototype Layers for Generalized Mobile Servicescited by 2
- 2024Tazza: Shuffling Neural Network Parameters for Secure and Private Federated Learningcited by 2
- 2025Deploying Models to Non-participating Clients in Federated Learning without Fine-tuning: A Hypernetwork-based Approachcited by 2
- 2025Individualised Treatment Effects Estimation with Composite Treatments and Composite Outcomescited by 1
- 2025Hypernetworks for Model-Heterogeneous Personalized Federated Learningcited by 1
- 2025FedPalm: A General Federated Learning Framework for Closed- and Open-Set Palmprint Verificationcited by 0
- 2025Toward Storage-Aware Learning with Compressed Data An Empirical Exploratory Study on JPEGcited by 0
- 2025Toward Storage-Aware Learning with Compressed Data An Empirical Exploratory Study on JPEG18 Aug 2025
- 2025Deploying Models to Non-participating Clients in Federated Learning without Fine-tuning: A Hypernetwork-based Approach18 Aug 2025
- 2025Hypernetworks for Model-Heterogeneous Personalized Federated Learning30 Jul 2025
- 2025FedPalm: A General Federated Learning Framework for Closed- and Open-Set Palmprint Verification5 Mar 2025
- 2025One-shot Federated Learning Methods: A Practical Guide13 Feb 2025
- 2025Individualised Treatment Effects Estimation with Composite Treatments and Composite Outcomes12 Feb 2025
- 2024Tazza: Shuffling Neural Network Parameters for Secure and Private Federated Learning10 Dec 2024
- 2024DeTrigger: A Gradient-Centric Approach to Backdoor Attack Mitigation in Federated Learning19 Nov 2024
- 2024FedAli: Personalized Federated Learning Alignment with Prototype Layers for Generalized Mobile Services15 Nov 2024