FedEBA+: Towards Fair and Effective Federated Learning via Entropy-Based Model
Zhichao Wang, Lin Wang, Ye Shi, Sai Praneeth Reddy Karimireddy, Xiaoying Tang
Abstract
Federated Learning (FL) often suffers from sacrificing global model accuracy when improving client-level fairness due to data heterogeneity, which often leads to inconsistent performance of the globally trained models, resulting in unfair outcomes among users. Existing fair FL algorithms face a bottleneck: they either sacrifice global model accuracy to promote fairness or fall short of achieving optimal fairness. In this paper, we propose a novel framework that effectively improves fairness while preserving global accuracy by integrating information-theoretic principles with model alignment. Specifically, we leverage the Maximum Entropy Principle to derive an analytic, closed-form solution for fair aggregation weights, ensuring significant fairness enhancements. We further employ a step-wise model alignment strategy that synchronizes gradient directions across heterogeneous clients, effectively mitigating the drift induced by local updates. Theoretical analysis proves that our method guarantees convergence even in non-convex settings. Importantly, we push the theoretical frontier of federated fairness by extending performance variance analysis to generalized regression, providing broader guarantees. Extensive experiments on five datasets demonstrate that our approach consistently outperforms state-of-the-art methods, achieving superior fairness without sacrificing global accuracy. Our code is available at https://github.com/T-Lab-CUHKSZ/FedEBA-Plus.
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 c6b9d410-c3bd-487a-b236-80ff5eaa26c2Cited by top-tier papers1
Ask how each one uses itBuilds on5
- Deep Learning with Differential PrivacyMartín Abadi, Andy Chu, Ian J. Goodfellow, H. Brendan McMahan et al.CCS 2016 · 7,620 citations
- Distributionally Robust Federated AveragingYuyang Deng, Mohammad Mahdi Kamani, Mehrdad MahdaviNeurIPS 2020 · 176 citations
- Energy-Based Learning for Cooperative Games, with Applications to Valuation Problems in Machine LearningYatao Bian, Yu Rong, Tingyang Xu, Jiaxiang Wu et al.ICLR 2022 · 17 citations
- Fair Federated Learning via the Proportional Veto CoreBhaskar Ray Chaudhury, Aniket Murhekar, Zhuowen Yuan, Bo Li et al.ICML 2024 · 14 citations
- Fairness in model-sharing gamesKate Donahue, Jon M. KleinbergWWW 2023 · 12 citations
Related papers
- FedLF: Layer-Wise Fair Federated LearningZibin Pan, Chi Li, Fangchen Yu, Shuyi Wang et al.AAAI 2024 · 12 citations
- A Fair Federated Learning Method for Handling Client Participation Probability Inconsistencies in Heterogeneous EnvironmentsSiyuan Wu, Yongzhe Jia, Haolong Xiang, Xiaolong Xu et al.NeurIPS 2025 · 2 citations
- LoGoFair: Post-Processing for Local and Global Fairness in Federated LearningLi Zhang, Chaochao Chen, Zhongxuan Han, Qiyong Zhong et al.AAAI 2025 · 1 citation
- Fair Federated Learning Under Domain Skew with Local Consistency and Domain DiversityYuhang Chen, Wenke Huang, Mang YeCVPR 2024
- CoRe-Fed: Bridging Collaborative and Representation Fairness via Federated Embedding DistillationNoorain Mukhtiar, Adnan Mahmood, Quan Z. ShengAAAI 2026
