CoRe-Fed: Bridging Collaborative and Representation Fairness via Federated Embedding Distillation
Noorain Mukhtiar, Adnan Mahmood, Quan Z. Sheng
Abstract
With the proliferation of distributed data sources, Federated Learning (FL) has emerged as a key approach to enable collaborative intelligence through decentralized model training while preserving data privacy. However, conventional FL algorithms often suffer from performance disparities across clients caused by heterogeneous data distributions and unequal participation, which leads to unfair outcomes. Specifically, we focus on two core fairness challenges, i.e., representation bias, arising from misaligned client representations, and collaborative bias, stemming from inequitable contribution during aggregation, both of which degrade model performance and generalizability. To mitigate these disparities, we propose CoRe-Fed, a unified optimization framework that bridges collaborative and representation fairness via embedding-level regularization and fairness-aware aggregation. Initially, an alignment-driven mechanism promotes semantic consistency between local and global embeddings to reduce representational divergence. Subsequently, a dynamic reward-penalty-based aggregation strategy adjusts each client's weight based on participation history and embedding alignment to ensure contribution-aware aggregation. Extensive experiments across diverse models and datasets demonstrate that CoRe-Fed improves both fairness and model performance over the state-of-the-art baseline algorithms. Code - https://github.com/Noorain1/CoRe-Fed Introduction Federated Learning (FL) has gained widespread adoption as a decentralized learning framework that enables multiple devices, sensors, or edge nodes (collectively referred to as clients or participants) to collaboratively train Machine Learning (ML) models without directly sharing their raw data, thus preserving data privacy (Woisetschläger et al. 2024) . However, despite its growing popularity, traditional FL encounters several challenges, particularly, in ensuring fair and unbiased model performance across clients with heterogeneous data distribution and varying participation frequency. These discrepancies result in models that perform disproportionately well on certain clients, while neglecting others, a phenomenon referred to as performance bias.
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 91455c0f-9872-40ff-a47a-2faa6e3c5e8bBuilds on20
- Ditto: Fair and Robust Federated Learning Through PersonalizationTian Li, Shengyuan Hu, Ahmad Beirami, Virginia SmithICML 2021 · 1,313 citations
- Fair Resource Allocation in Federated LearningTian Li, Maziar Sanjabi, Ahmad Beirami, Virginia SmithICLR 2020 · 971 citations
- No Fear of Heterogeneity: Classifier Calibration for Federated Learning with Non-IID DataMi Luo, Fei Chen, Dapeng Hu, Yifan Zhang et al.NeurIPS 2021 · 510 citations
- The Pitfalls of Simplicity Bias in Neural NetworksHarshay Shah, Kaustav Tamuly, Aditi Raghunathan, Prateek Jain et al.NeurIPS 2020 · 503 citations
- Addressing Class Imbalance in Federated LearningLixu Wang, Shichao Xu, Xiao Wang, Qi ZhuAAAI 2021 · 314 citations
Related papers
- Fair Federated Learning Under Domain Skew with Local Consistency and Domain DiversityYuhang Chen, Wenke Huang, Mang YeCVPR 2024
- FairFed: Enabling Group Fairness in Federated LearningYahya H. Ezzeldin, Shen Yan, Chaoyang He, Emilio Ferrara et al.AAAI 2023 · 310 citations
- Rethinking Fair Federated Learning from Parameter and Client ViewKaiqi Guan, Wenke Huang, Xianda Guo, Yueyang Yuan et al.NeurIPS 2025 · 1 citation
- FedEBA+: Towards Fair and Effective Federated Learning via Entropy-Based ModelZhichao Wang, Lin Wang, Ye Shi, Sai Praneeth Reddy Karimireddy et al.ICML 2026
- 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
