Federated Multi-Task Learning
Not on the map. Why?
Abstract (arXiv)
Federated learning poses new statistical and systems challenges in training machine learning models over distributed networks of devices. In this work, we show that multi-task learning is naturally suited to handle the statistical challenges of this setting, and propose a novel systems-aware optimization method, MOCHA, that is robust to practical systems issues. Our method and theory for the first time consider issues of high communication cost, stragglers, and fault tolerance for distributed multi-task learning. The resulting method achieves significant speedups compared to alternatives in the federated setting, as we demonstrate through simulations on real-world federated datasets.
Key facts. Cited by 669 papers in the map (data through 7 Oct 2026). First posted on arXiv 30 May 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 181; 2022: 129; 2023: 113; 2024: 101; 2025: 100; 2026: 44.
669 papers.
Sort:
By area:
- 2021Exploiting Shared Representations for Personalized Federated Learningcited by 537
- 2021Towards Personalized Federated Learningcited by 462
- 2021Federated Learning on Non-IID Data: A Surveycited by 345
- 2021A Field Guide to Federated Optimizationcited by 312
- 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
- 2021Federated Multi-Task Learning under a Mixture of Distributionscited by 189
- 2021FedBABU: Towards Enhanced Representation for Federated Image Classificationcited by 180
- 2023Heterogeneous Federated Learning: State-of-the-art and Research Challengescited by 160
- 2021Advances in adversarial attacks and defenses in computer vision: A surveycited by 127
- 2021Distributed Learning in Wireless Networks: Recent Progress and Future Challengescited by 125
38 more
- 2024OpenFedLLM: Training Large Language Models on Decentralized Private Data via Federated Learningcited by 124
- 2022Federated Learning with Partial Model Personalizationcited by 114
- 2021FedMix: Approximation of Mixup under Mean Augmented Federated Learningcited by 108
- 2021Parameterized Knowledge Transfer for Personalized Federated Learningcited by 108
- 2022Improving Generalization in Federated Learning by Seeking Flat Minimacited by 98
- 2023Personalized Federated Learning with Feature Alignment and Classifier Collaborationcited by 97
- 2023Federated Learning for Generalization, Robustness, Fairness: A Survey and Benchmarkcited by 97
- 2021Edge Artificial Intelligence for 6G: Vision, Enabling Technologies, and Applicationscited by 96
- 2021CRFL: Certifiably Robust Federated Learning against Backdoor Attackscited by 91
- 2023Deep Model Fusion: A Surveycited by 89
- 2021Heterogeneity for the Win: One-Shot Federated Clusteringcited by 87
- 2021Federated Evaluation and Tuning for On-Device Personalization: System Design & Applicationscited by 84
- 2021FedNLP: Benchmarking Federated Learning Methods for Natural Language Processing Taskscited by 84
- 2021Federated Reconstruction: Partially Local Federated Learningcited by 78
- 2023Federated Learning for Medical Image Analysis: A Surveycited by 77
- 2021Personalized Federated Learning with Gaussian Processescited by 76
- 2021Fast Federated Learning in the Presence of Arbitrary Device Unavailabilitycited by 72
- 2021Cross-Node Federated Graph Neural Network for Spatio-Temporal Data Modelingcited by 69
- 2022Personalized Federated Learning via Variational Bayesian Inferencecited by 68
- 2021FedAUX: Leveraging Unlabeled Auxiliary Data in Federated Learningcited by 67
- 2024Learning to Route Among Specialized Experts for Zero-Shot Generalizationcited by 65
- 2021Towards Fair Federated Learning with Zero-Shot Data Augmentationcited by 64
- 2021Federated Multi-armed Bandits with Personalizationcited by 62
- 2021What Do We Mean by Generalization in Federated Learning?cited by 62
- 2021Provably Secure Federated Learning against Malicious Clientscited by 60
- 2021Federated Multi-Armed Banditscited by 59
- 2021Federated Hyperparameter Tuning: Challenges, Baselines, and Connections to Weight-Sharingcited by 58
- 2021Personalized Federated Learning through Local Memorizationcited by 57
- 2021Decentralized Federated Learning: Balancing Communication and Computing Costscited by 54
- 2021Adapt to Adaptation: Learning Personalization for Cross-Silo Federated Learningcited by 53
- 2022FederatedScope: A Flexible Federated Learning Platform for Heterogeneitycited by 53
- 2021Semi-Decentralized Federated Learning with Cooperative D2D Local Model Aggregationscited by 50
- 2021FedGEMS: Federated Learning of Larger Server Models via Selective Knowledge Fusioncited by 49
- 2021Flexible Clustered Federated Learning for Client-Level Data Distribution Shiftcited by 47
- 2021FedSoft: Soft Clustered Federated Learning with Proximal Local Updatingcited by 46
- 2022Orchestra: Unsupervised Federated Learning via Globally Consistent Clusteringcited by 45
- 2022Preserving Privacy in Federated Learning with Ensemble Cross-Domain Knowledge Distillationcited by 45
- 2021Federated Multi-Target Domain Adaptationcited by 44
50 most cited of 669 All 669 (JSON)
- 2026Encoder-Sharing Hierarchical Federated Multi-Task Learning for VANETs28 Sep 2026
- 2026Joint Domain-Class Modeling for Federated Learning Under Feature Skew19 Sep 2026
- 2026Robust Decentralized Personalized Federated Learning via Prediction-Constrained Neighborhood Collaboration7 Sep 2026
- 2026Similarity-Aware Personalized Federated Learning in Heterogeneous Environments2 Sep 2026
- 2026Beyond Non-IID: Learner--Client Distribution Mismatch in Federated Learning27 Aug 2026
- 2026Decentralized Multitask Learning over Learned Task Graphs27 Aug 2026
- 2026A Unified Framework for Fair and Personalized Decentralized Learning under Communication Constraints27 Aug 2026
- 2026Sheaf-Based Federated Representation Learning8 Aug 2026
- 2026Hierarchical Multi-Task Federated Learning in VANETs8 Aug 2026
- 2026FedLBW: A Loss-Based Weighting Strategy for Federated Learning on Non-IID Data in Wireless Networks7 Aug 2026
- 2026Federated Learning for Global Carbon Emission Forecasting: A Hybrid Time-Series Approach with Statistical and Neural Models21 Jun 2026
- 2026Post-Hoc Merging is Not Enough: Many-Shot Model Merging with Loss-Gap Balancing15 Jun 2026
38 more
- 2026FedSPC: Shared Parameter Correction for Personalized Federated Learning11 Jun 2026
- 2026FedMPT: Federated Multi-label Prompt Tuning of Vision-Language Models27 May 2026
- 2026Separate Aggregation of Split Network for Personalized Federated Learning26 May 2026
- 2026Partial Model Sharing Improves Byzantine Resilience in Federated Conformal Prediction12 May 2026
- 2026MuCALD-SplitFed: Causal-Latent Diffusion for Privacy-Preserving Multi-Task Split-Federated Medical Image Segmentation4 May 2026
- 2026Near-optimal and Efficient First-Order Algorithm for Multi-Task Learning with Shared Linear Representation1 May 2026
- 2026When To Adapt? Adapting the Model or Data in Federated Medical Imaging27 Apr 2026
- 2026Decision-Focused Federated Learning Under Heterogeneous Objectives and Constraints21 Apr 2026
- 2026FB-NLL: A Feature-Based Approach to Tackle Noisy Labels in Personalized Federated Learning21 Apr 2026
- 2026Semantic-based Distributed Learning for Diverse and Discriminative Representations20 Apr 2026
- 2026Robust Synchronisation for Federated Learning in The Face of Correlated Device Failure17 Apr 2026
- 2026FedRio: Personalized Federated Social Bot Detection via Cooperative Reinforced Contrastive Adversarial Distillation12 Apr 2026
- 2026Enhancing Cross-Problem Vehicle Routing via Federated Learning12 Apr 2026
- 2026Prototype-Regularized Federated Learning for Cross-Domain Aspect Sentiment Triplet Extraction10 Apr 2026
- 2026Task-Centric Personalized Federated Fine-Tuning of Language Models30 Mar 2026
- 2026Communication-Efficient and Robust Multi-Modal Federated Learning via Latent-Space Consensus19 Mar 2026
- 2026Few-for-Many Personalized Federated Learning12 Mar 2026
- 2026Personalized Federated Learning via Gaussian Generative Modeling12 Mar 2026
- 2026Client-Conditional Federated Learning via Local Training Data Statistics11 Mar 2026
- 2026Federated Active Learning Under Extreme Non-IID and Global Class Imbalance11 Mar 2026
- 2026FedPrism: Adaptive Personalized Federated Learning under Non-IID Data9 Mar 2026
- 2026Learning Unknown Interdependencies for Decentralized Root Cause Analysis in Nonlinear Dynamical Systems25 Feb 2026
- 2026Communication-Efficient Personalized Adaptation via Federated-Local Model Merging20 Feb 2026
- 2026Communication-Efficient Byzantine-Robust Federated Conformal Prediction via Partial Sharing20 Feb 2026
- 2026A Hybrid Federated Learning Based Ensemble Approach for Lung Disease Diagnosis Leveraging Fusion of SWIN Transformer and CNN19 Feb 2026
- 2026Federated Ensemble Learning with Progressive Model Personalization15 Feb 2026
- 2026Adaptive Personalized Federated Learning via Multi-task Averaging of Kernel Mean Embeddings13 Feb 2026
- 2026Rashomon Sets and Model Multiplicity in Federated Learning10 Feb 2026
- 2026Equitable Multi-Task Learning for AI-RANs9 Feb 2026
- 2026Toward Enhancing Representation Learning in Federated Multi-Task Settings2 Feb 2026
- 2026Mechanism Design for Federated Learning with Non-Monotonic Network Effects8 Jan 2026
- 2026Provably Convergent Decentralized Optimization over Directed Graphs under Generalized Smoothness7 Jan 2026
- 2025Evidential Trust-Aware Model Personalization in Decentralized Federated Learning for Wearable IoT22 Dec 2025
- 2025Feature-Based Semantics-Aware Scheduling for Energy-Harvesting Federated Learning1 Dec 2025
- 2025Federated style aware transformer aggregation of representations24 Nov 2025
- 2025Personalized Federated Learning with Bidirectional Communication Compression via One-Bit Random Sketching17 Nov 2025
- 2025MedFedPure: A Medical Federated Framework with MAE-based Detection and Diffusion Purification for Inference-Time Attacks7 Nov 2025
- 2025Communication-Constrained Private Decentralized Online Personalized Mean Estimation2 Nov 2025
50 newest of 669 All 669 (JSON)
- 2023FedDiff: Diffusion Model Driven Federated Learning for Multi-Modal and Multi-Clientscited by 23
- 2024ADEPT: Hierarchical Bayes Approach to Personalized Federated Unsupervised Learningcited by 1
- 2024pFedGPA: Diffusion-based Generative Parameter Aggregation for Personalized Federated Learningcited by 0
- 2025WarmFed: Federated Learning with Warm-Start for Globalization and Personalization Via Personalized Diffusion Modelscited by 0
- 2025SPIRE: Conditional Personalization for Federated Diffusion Generative Modelscited by 0
- 2026MuCALD-SplitFed: Causal-Latent Diffusion for Privacy-Preserving Multi-Task Split-Federated Medical Image Segmentationcited by 0
- 2026MuCALD-SplitFed: Causal-Latent Diffusion for Privacy-Preserving Multi-Task Split-Federated Medical Image Segmentation4 May 2026
- 2025SPIRE: Conditional Personalization for Federated Diffusion Generative Models14 Jun 2025
- 2025WarmFed: Federated Learning with Warm-Start for Globalization and Personalization Via Personalized Diffusion Models5 Mar 2025
- 2024pFedGPA: Diffusion-based Generative Parameter Aggregation for Personalized Federated Learning9 Sep 2024
- 2024ADEPT: Hierarchical Bayes Approach to Personalized Federated Unsupervised Learning19 Feb 2024
- 2023FedDiff: Diffusion Model Driven Federated Learning for Multi-Modal and Multi-Clients16 Nov 2023
- 2023Exploring the Landscape of Machine Unlearning: A Comprehensive Survey and Taxonomycited by 44
- 2021Robust Federated Learning with Attack-Adaptive Aggregationcited by 17
- 2021Meta Federated Learningcited by 8
- 2023Exploring the Landscape of Machine Unlearning: A Comprehensive Survey and Taxonomy10 May 2023
- 2021Meta Federated Learning10 Feb 2021
- 2021Robust Federated Learning with Attack-Adaptive Aggregation10 Feb 2021
- 2021Exploiting Shared Representations for Personalized Federated Learningcited by 537
- 2021Towards Personalized Federated Learningcited by 462
- 2021Federated Learning on Non-IID Data: A Surveycited by 345
- 2021A Field Guide to Federated Optimizationcited by 312
- 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
- 2021Federated Multi-Task Learning under a Mixture of Distributionscited by 189
- 2021FedBABU: Towards Enhanced Representation for Federated Image Classificationcited by 180
- 2023Heterogeneous Federated Learning: State-of-the-art and Research Challengescited by 160
- 2021Advances in adversarial attacks and defenses in computer vision: A surveycited by 127
- 2021Distributed Learning in Wireless Networks: Recent Progress and Future Challengescited by 125
12 most cited of 653 All 653 (JSON)
- 2026Encoder-Sharing Hierarchical Federated Multi-Task Learning for VANETs28 Sep 2026
- 2026Joint Domain-Class Modeling for Federated Learning Under Feature Skew19 Sep 2026
- 2026Robust Decentralized Personalized Federated Learning via Prediction-Constrained Neighborhood Collaboration7 Sep 2026
- 2026Similarity-Aware Personalized Federated Learning in Heterogeneous Environments2 Sep 2026
- 2026Beyond Non-IID: Learner--Client Distribution Mismatch in Federated Learning27 Aug 2026
- 2026Decentralized Multitask Learning over Learned Task Graphs27 Aug 2026
- 2026A Unified Framework for Fair and Personalized Decentralized Learning under Communication Constraints27 Aug 2026
- 2026Sheaf-Based Federated Representation Learning8 Aug 2026
- 2026Hierarchical Multi-Task Federated Learning in VANETs8 Aug 2026
- 2026FedLBW: A Loss-Based Weighting Strategy for Federated Learning on Non-IID Data in Wireless Networks7 Aug 2026
- 2026Federated Learning for Global Carbon Emission Forecasting: A Hybrid Time-Series Approach with Statistical and Neural Models21 Jun 2026
- 2026Post-Hoc Merging is Not Enough: Many-Shot Model Merging with Loss-Gap Balancing15 Jun 2026
12 newest of 653 All 653 (JSON)