Benchmarking Algorithms for Federated Domain Generalization
Ruqi Bai, Saurabh Bagchi, David I. Inouye
Abstract
While prior federated learning (FL) methods mainly consider client heterogeneity, we focus on the Federated Domain Generalization (DG) task, which introduces train-test heterogeneity in the FL context. Existing evaluations in this field are limited in terms of the scale of the clients and dataset diversity. Thus, we propose a Federated DG benchmark that aim to test the limits of current methods with high client heterogeneity, large numbers of clients, and diverse datasets. Towards this objective, we introduce a novel data partition method that allows us to distribute any domain dataset among few or many clients while controlling client heterogeneity. We then introduce and apply our methodology to evaluate 14 DG methods, which include centralized DG methods adapted to the FL context, FL methods that handle client heterogeneity, and methods designed specifically for Federated DG on 7 datasets. Our results suggest that, despite some progress, significant performance gaps remain in Federated DG, especially when evaluating with a large number of clients, high client heterogeneity, or more realistic datasets. Furthermore, our extendable benchmark code will be publicly released to aid in benchmarking future Federated DG approaches.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 31bfb629-6457-420b-a2a8-4a37ec28e99fCited by top-tier papers7
- DapperFL: Domain Adaptive Federated Learning with Model Fusion Pruning for Edge DevicesYongzhe Jia, Xuyun Zhang, Hongsheng Hu, Kim-Kwang Raymond Choo et al.NeurIPS 2024 · 14 citations
- Federated Unsupervised Domain Generalization Using Global and Local Alignment of GradientsFarhad Pourpanah, Mahdiyar Molahasani, Milad Soltany, Michael A. Greenspan et al.AAAI 2025 · 10 citations
- Learning to Specialize: Joint Gating-Expert Training for Adaptive MoEs in Decentralized SettingsYehya Farhat, Hamza ElMokhtar Shili, Fangshuo Liao, Chen Dun et al.NeurIPS 2025 · 2 citations
- HFedATM: Hierarchical Federated Domain Generalization via Optimal Transport and Regularized Mean AggregationThinh Nguyen, Trung Phan, Binh T. Nguyen, Khoa D. Doan et al.CVPR 2026
- DiPrompT: Disentangled Prompt Tuning for Multiple Latent Domain Generalization in Federated LearningSikai Bai, Jie Zhang, Song Guo, Shuaicheng Li et al.CVPR 2024
Builds on11
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- On the Convergence of FedAvg on Non-IID DataXiang Li, Kaixuan Huang, Wenhao Yang, Shusen Wang et al.ICLR 2020 · 2,930 citations
- Adaptive Federated OptimizationSashank J. Reddi, Zachary Charles, Manzil Zaheer, Zachary Garrett et al.ICLR 2021 · 1,917 citations
- WILDS: A Benchmark of in-the-Wild Distribution ShiftsPang Wei Koh, Shiori Sagawa, Henrik Marklund, Sang Michael Xie et al.ICML 2021 · 1,773 citations
- Adversarial Domain Adaptation with Domain MixupMinghao Xu, Jian Zhang, Bingbing Ni, Teng Li et al.AAAI 2020 · 499 citations
Related papers
- Factorized-FL: Personalized Federated Learning with Parameter Factorization & Similarity MatchingWonyong Jeong, Sung Ju HwangNeurIPS 2022 · 37 citations
- Two Heads Are Better Than One: Generalized Cross-Domain Federated Learning via Dual-PrototypeMingsheng Cao, Tianci Chen, Ming Hu, Zhuang Qi et al.KDD 2026
- FedProto: Federated Prototype Learning across Heterogeneous ClientsYue Tan, Guodong Long, Lu Liu, Tianyi Zhou et al.AAAI 2022 · 851 citations
- FLHetBench: Benchmarking Device and State Heterogeneity in Federated LearningJunyuan Zhang, Shuang Zeng, Miao Zhang, Runxi Wang et al.CVPR 2024 · 6 citations
- Efficient Federated Domain TranslationZeyu Zhou, Sheikh Shams Azam, Christopher G. Brinton, David I. InouyeICLR 2023
