Federated Learning with Matched Averaging
Not on the map. Why?
Abstract (arXiv)
Federated learning allows edge devices to collaboratively learn a shared model while keeping the training data on device, decoupling the ability to do model training from the need to store the data in the cloud. We propose Federated matched averaging (FedMA) algorithm designed for federated learning of modern neural network architectures e.g. convolutional neural networks (CNNs) and LSTMs. FedMA constructs the shared global model in a layer-wise manner by matching and averaging hidden elements (i.e. channels for convolution layers; hidden states for LSTM; neurons for fully connected layers) with similar feature extraction signatures. Our experiments indicate that FedMA not only outperforms popular state-of-the-art federated learning algorithms on deep CNN and LSTM architectures trained on real world datasets, but also reduces the overall communication burden.
Key facts. Cited by 564 papers in the map (data through 7 Oct 2026). First posted on arXiv 15 Feb 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 97; 2022: 112; 2023: 125; 2024: 123; 2025: 76; 2026: 31.
564 papers.
Sort:
By area:
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
- 2021Model-Contrastive Federated Learningcited by 762
- 2021Merging Models with Fisher-Weighted Averagingcited by 613
- 2021Federated Learning on Non-IID Data Silos: An Experimental Studycited by 566
- 2021FedBN: Federated Learning on Non-IID Features via Local Batch Normalizationcited by 565
- 2021Towards Personalized Federated Learningcited by 462
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
- 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 384
- 2021Data-Free Knowledge Distillation for Heterogeneous Federated Learningcited by 380
- 2021FedProto: Federated Prototype Learning across Heterogeneous Clientscited by 349
- 2021Federated Learning on Non-IID Data: A Surveycited by 345
- 2023Mix-of-Show: Decentralized Low-Rank Adaptation for Multi-Concept Customization of Diffusion Modelscited by 261
38 more
- 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 248
- 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
- 2021No Fear of Heterogeneity: Classifier Calibration for Federated Learning with Non-IID Datacited by 241
- 2021FjORD: Fair and Accurate Federated Learning under heterogeneous targets with Ordered Dropoutcited by 192
- 2021Federated Multi-Task Learning under a Mixture of Distributionscited by 189
- 2023ZipIt! Merging Models from Different Tasks without Trainingcited by 178
- 2022NOTE: Robust Continual Test-time Adaptation Against Temporal Correlationcited by 150
- 2022Federated Learning from Pre-Trained Models: A Contrastive Learning Approachcited by 141
- 2021Federated Learning for Open Bankingcited by 138
- 2022Federated Class-Incremental Learningcited by 126
- 2022FedALA: Adaptive Local Aggregation for Personalized Federated Learningcited by 121
- 2022REPAIR: REnormalizing Permuted Activations for Interpolation Repaircited by 119
- 2021Local Learning Matters: Rethinking Data Heterogeneity in Federated Learningcited by 114
- 2024Twin-Merging: Dynamic Integration of Modular Expertise in Model Mergingcited by 114
- 2021Multi-Center Federated Learning: Clients Clustering for Better Personalizationcited by 112
- 2021Rethinking Architecture Design for Tackling Data Heterogeneity in Federated Learningcited by 111
- 2021FedMix: Approximation of Mixup under Mean Augmented Federated Learningcited by 108
- 2022Federated Learning with Label Distribution Skew via Logits Calibrationcited by 108
- 2023Equivariant Architectures for Learning in Deep Weight Spacescited by 106
- 2022Neurotoxin: Durable Backdoors in Federated Learningcited by 97
- 2023Federated Learning for Generalization, Robustness, Fairness: A Survey and Benchmarkcited by 97
- 2021PPFL: Privacy-preserving Federated Learning with Trusted Execution Environmentscited by 93
- 2023Deep Model Fusion: A Surveycited by 89
- 2021Preservation of the Global Knowledge by Not-True Distillation in Federated Learningcited by 87
- 2021FedNLP: Benchmarking Federated Learning Methods for Natural Language Processing Taskscited by 84
- 2021FedProc: Prototypical Contrastive Federated Learning on Non-IID datacited by 80
- 2022NVIDIA FLARE: Federated Learning from Simulation to Real-Worldcited by 78
- 2022Deep Model Reassemblycited by 78
- 2021Personalized Federated Learning with Gaussian Processescited by 76
- 2022DisPFL: Towards Communication-Efficient Personalized Federated Learning via Decentralized Sparse Trainingcited by 74
- 2021On Large-Cohort Training for Federated Learningcited by 73
- 2022Divergence-aware Federated Self-Supervised Learningcited by 71
- 2022On the Importance and Applicability of Pre-Training for Federated Learningcited by 71
- 2024Parameter Competition Balancing for Model Mergingcited by 71
- 2023FedDisco: Federated Learning with Discrepancy-Aware Collaborationcited by 66
- 2021From Distributed Machine Learning to Federated Learning: A Surveycited by 65
- 2022FedDM: Iterative Distribution Matching for Communication-Efficient Federated Learningcited by 65
- 2022Re-basin via implicit Sinkhorn differentiationcited by 61
50 most cited of 564 All 564 (JSON)
- 2026Understanding Head Geometry and Dynamics in Federated Regression through a Natural Solution Selection Rule: An Unconstrained Feature Model Analysis30 Sep 2026
- 2026FedCMAPSS: A Benchmark for Federated Learning in Remaining Useful Life Estimation26 Aug 2026
- 2026Socialized Detector Learning: Trajectory-Guided and Reciprocal Distillation for Heterogeneous Object Detectors26 Aug 2026
- 2026FedTVD: Balancing Data Quality and Quantity for Robust Federated Learning10 Aug 2026
- 2026FedLBW: A Loss-Based Weighting Strategy for Federated Learning on Non-IID Data in Wireless Networks7 Aug 2026
- 2026Similarity Weighted Aggregation with Global Differential Privacy for Federated Brain Lesion Segmentation1 Aug 2026
- 2026FedGuide: Diffusion Prior Alignment and Value Baseline Guidance for Heterogeneous Federated Reinforcement Learning22 Jul 2026
- 2026Federated Lightweight Fine-Tuning20 Jul 2026
- 2026Robust Federated Learning Under Real-World Client Churn8 Jul 2026
- 2026Sparsity Curse: Understanding RLVR Model Parameter Space from Model Merging16 Jun 2026
- 2026FlashbackCL: Mitigating Temporal Forgetting in Federated Learning2 Jun 2026
- 2026Silent Failures in Federated Personalization of Foundation Models31 May 2026
38 more
- 2026FedEHR-Gen: Federated Synthetic Time-Series EHR Generation via Latent Space Alignment and Distribution-Aware Aggregation27 May 2026
- 2026FedHPro: Federated Hyper-Prototype Learning via Gradient Matching13 May 2026
- 2026FedVSSAM: Mitigating Flatness Incompatibility in Sharpness-Aware Federated Learning9 May 2026
- 2026Heterogeneous Model Fusion for Privacy-Aware Multi-Camera Surveillance via Synthetic Domain Adaptation4 May 2026
- 2026DeRelayL: Sustainable Decentralized Relay Learning30 Apr 2026
- 2026Auto-FlexSwitch: Efficient Dynamic Model Merging via Learnable Task Vector Compression30 Apr 2026
- 2026FedIDM: Achieving Fast and Stable Convergence in Byzantine Federated Learning through Iterative Distribution Matching16 Apr 2026
- 2026FedDAP: Domain-Aware Prototype Learning for Federated Learning under Domain Shift8 Apr 2026
- 2026FedVG: Gradient-Guided Aggregation for Enhanced Federated Learning24 Feb 2026
- 2026Heterogeneity-Aware Client Selection Methodology For Efficient Federated Learning24 Feb 2026
- 2026Training-Free Cross-Architecture Merging for Graph Neural Networks22 Feb 2026
- 2026Safeguarding Privacy: Privacy-Preserving Detection of Mind Wandering and Disengagement Using Federated Learning in Online Education10 Feb 2026
- 2026Linear Model Merging Unlocks Simple and Scalable Multimodal Data Mixture Optimization4 Feb 2026
- 2026Fedcompass: Federated Clustered and Periodic Aggregation Framework for Hybrid Classical-Quantum Models3 Feb 2026
- 2026Rethinking LoRA for Data Heterogeneous Federated Learning: Subspace and State Alignment2 Feb 2026
- 2026System 1&2 Synergy via Dynamic Model Interpolation29 Jan 2026
- 2026FedCCA: Client-Centric Adaptation against Data Heterogeneity in Federated Learning on IoT Devices25 Jan 2026
- 2026Federated Proximal Optimization for Privacy-Preserving Heart Disease Prediction: A Controlled Simulation Study on Non-IID Clinical Data23 Jan 2026
- 2026DecHW: Heterogeneous Decentralized Federated Learning Exploiting Second-Order Information16 Jan 2026
- 2025Federated Learning with Feedback Alignment14 Dec 2025
- 2025Stragglers Can Contribute More: Uncertainty-Aware Distillation for Asynchronous Federated Learning25 Nov 2025
- 2025ParaBlock: Communication-Computation Parallel Block Coordinate Federated Learning for Large Language Models25 Nov 2025
- 2025Merging without Forgetting: Continual Fusion of Task-Specific Models via Optimal Transport24 Nov 2025
- 2025Federated Learning Framework for Scalable AI in Heterogeneous HPC and Cloud Environments22 Nov 2025
- 2025A Unified Convergence Analysis for Semi-Decentralized Learning: Sampled-to-Sampled vs. Sampled-to-All Communication14 Nov 2025
- 2025FedSparQ: Adaptive Sparse Quantization with Error Feedback for Robust & Efficient Federated Learning5 Nov 2025
- 2025Federated Learning via Meta-Variational Dropout23 Oct 2025
- 2025BlendFL: Blended Federated Learning for Handling Multimodal Data Heterogeneity15 Oct 2025
- 2025Robustness and Regularization in Hierarchical Re-Basin10 Oct 2025
- 2025FedDTRE: Federated Dialogue Generation Models Powered by Trustworthiness Evaluation9 Oct 2025
- 2025Do We Really Need Permutations? Impact of Model Width on Linear Mode Connectivity9 Oct 2025
- 2025Trade-off in Estimating the Number of Byzantine Clients in Federated Learning6 Oct 2025
- 2025Beyond Static Knowledge Messengers: Towards Adaptive, Fair, and Scalable Federated Learning for Medical AI5 Oct 2025
- 2025Decentralized Dynamic Cooperation of Personalized Models for Federated Continual Learning28 Sep 2025
- 2025Adaptive Dual-Mode Distillation with Incentive Schemes for Scalable, Heterogeneous Federated Learning on Non-IID Data26 Sep 2025
- 2025Cost-Free Personalization via Information-Geometric Projection in Bayesian Federated Learning12 Sep 2025
- 2025Reconciling Communication Compression and Byzantine-Robustness in Distributed Learning23 Aug 2025
- 2025Learn Faster and Remember More: Balancing Exploration and Exploitation for Continual Test-time Adaptation18 Aug 2025
50 newest of 564 All 564 (JSON)
- 2023Mix-of-Show: Decentralized Low-Rank Adaptation for Multi-Concept Customization of Diffusion Modelscited by 261
- 2024Stable Diffusion-based Data Augmentation for Federated Learning with Non-IID Datacited by 10
- 2025FedDifRC: Unlocking the Potential of Text-to-Image Diffusion Models in Heterogeneous Federated Learningcited by 3
- 2026FedGuide: Diffusion Prior Alignment and Value Baseline Guidance for Heterogeneous Federated Reinforcement Learningcited by 0
- 2026FedGuide: Diffusion Prior Alignment and Value Baseline Guidance for Heterogeneous Federated Reinforcement Learning22 Jul 2026
- 2025FedDifRC: Unlocking the Potential of Text-to-Image Diffusion Models in Heterogeneous Federated Learning9 Jul 2025
- 2024Stable Diffusion-based Data Augmentation for Federated Learning with Non-IID Data13 May 2024
- 2023Mix-of-Show: Decentralized Low-Rank Adaptation for Multi-Concept Customization of Diffusion Models29 May 2023
- 2025Metropolis-Hastings Captioning Game: Knowledge Fusion of Vision Language Models via Decentralized Bayesian Inferencecited by 1
- 2026Linear Model Merging Unlocks Simple and Scalable Multimodal Data Mixture Optimizationcited by 1
- 2025FLIP: Towards Comprehensive and Reliable Evaluation of Federated Prompt Learningcited by 0
- 2026Linear Model Merging Unlocks Simple and Scalable Multimodal Data Mixture Optimization4 Feb 2026
- 2025Metropolis-Hastings Captioning Game: Knowledge Fusion of Vision Language Models via Decentralized Bayesian Inference13 Apr 2025
- 2025FLIP: Towards Comprehensive and Reliable Evaluation of Federated Prompt Learning28 Mar 2025
- 2021Robust Federated Learning with Attack-Adaptive Aggregationcited by 17
- 2022Backdoor Attack and Defense in Federated Generative Adversarial Network-based Medical Image Synthesiscited by 12
- 2024Accuracy-Privacy Trade-off in the Mitigation of Membership Inference Attack in Federated Learningcited by 1
- 2024Accuracy-Privacy Trade-off in the Mitigation of Membership Inference Attack in Federated Learning26 Jul 2024
- 2022Backdoor Attack and Defense in Federated Generative Adversarial Network-based Medical Image Synthesis19 Oct 2022
- 2021Robust Federated Learning with Attack-Adaptive Aggregation10 Feb 2021
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
- 2021Model-Contrastive Federated Learningcited by 762
- 2021Merging Models with Fisher-Weighted Averagingcited by 613
- 2021Federated Learning on Non-IID Data Silos: An Experimental Studycited by 566
- 2021FedBN: Federated Learning on Non-IID Features via Local Batch Normalizationcited by 565
- 2021Towards Personalized Federated Learningcited by 462
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
- 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 384
- 2021Data-Free Knowledge Distillation for Heterogeneous Federated Learningcited by 380
- 2021FedProto: Federated Prototype Learning across Heterogeneous Clientscited by 349
- 2021Federated Learning on Non-IID Data: A Surveycited by 345
- 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 248
12 most cited of 551 All 551 (JSON)
- 2026Understanding Head Geometry and Dynamics in Federated Regression through a Natural Solution Selection Rule: An Unconstrained Feature Model Analysis30 Sep 2026
- 2026FedCMAPSS: A Benchmark for Federated Learning in Remaining Useful Life Estimation26 Aug 2026
- 2026Socialized Detector Learning: Trajectory-Guided and Reciprocal Distillation for Heterogeneous Object Detectors26 Aug 2026
- 2026FedTVD: Balancing Data Quality and Quantity for Robust Federated Learning10 Aug 2026
- 2026FedLBW: A Loss-Based Weighting Strategy for Federated Learning on Non-IID Data in Wireless Networks7 Aug 2026
- 2026Similarity Weighted Aggregation with Global Differential Privacy for Federated Brain Lesion Segmentation1 Aug 2026
- 2026Federated Lightweight Fine-Tuning20 Jul 2026
- 2026Robust Federated Learning Under Real-World Client Churn8 Jul 2026
- 2026Sparsity Curse: Understanding RLVR Model Parameter Space from Model Merging16 Jun 2026
- 2026FlashbackCL: Mitigating Temporal Forgetting in Federated Learning2 Jun 2026
- 2026Silent Failures in Federated Personalization of Foundation Models31 May 2026
- 2026FedEHR-Gen: Federated Synthetic Time-Series EHR Generation via Latent Space Alignment and Distribution-Aware Aggregation27 May 2026
12 newest of 551 All 551 (JSON)