English

CoRe-Fed: Bridging Collaborative and Representation Fairness via Federated Embedding Distillation

Machine Learning 2026-02-03 v1

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.

Keywords

Cite

@article{arxiv.2602.00647,
  title  = {CoRe-Fed: Bridging Collaborative and Representation Fairness via Federated Embedding Distillation},
  author = {Noorain Mukhtiar and Adnan Mahmood and Quan Z. Sheng},
  journal= {arXiv preprint arXiv:2602.00647},
  year   = {2026}
}

Comments

7 pages (main content), 2 pages (references), Accepted in AAAI 2026

R2 v1 2026-07-01T09:29:17.455Z