Bias-Eliminating Augmentation Learning for Debiased Federated Learning
Yuan-Yi Xu, Ci-Siang Lin, Yu-Chiang Frank Wang
Abstract
Learning models trained on biased datasets tend to observe correlations between categorical and undesirable features, which result in degraded performances. Most existing debiased learning models are designed for centralized machine learning, which cannot be directly applied to distributed settings like federated learning (FL), which collects data at distinct clients with privacy preserved. To tackle the challenging task of debiased federated learning, we present a novel FL framework of Bias-Eliminating Augmentation Learning (FedBEAL), which learns to deploy Bias-Eliminating Augmenters (BEA) for producing client-specific bias-conflicting samples at each client. Since the bias types or attributes are not known in advance, a unique learning strategy is presented to jointly train BEA with the proposed FL framework. Extensive image classification experiments on datasets with various bias types confirm the effectiveness and applicability of our FedBEAL, which performs favorably against state-of-the-art debiasing and FL methods for debiased FL.
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 603286fc-fb46-4e3e-87ba-bdb5357ae9a6Cited by top-tier papers6
- Self-Driven Entropy Aggregation for Byzantine-Robust Heterogeneous Federated LearningWenke Huang, Zekun Shi, Mang Ye, He Li et al.ICML 2024 · 16 citations
- FedSAC: Dynamic Submodel Allocation for Collaborative Fairness in Federated LearningZihui Wang, Zheng Wang, Lingjuan Lyu, Zhaopeng Peng et al.KDD 2024 · 6 citations
- Federated Self-Explaining GNNs with Anti-shortcut AugmentationsLinan Yue, Qi Liu, Weibo Gao, Ye Liu et al.ICML 2024 · 2 citations
- FedSum: Data-Efficient Federated Learning Under Data Scarcity Scenario for Text SummarizationZhiyong Ma, Zhengping Li, Yuanjie Shi, Jian ChenAAAI 2025 · 2 citations
- DKDR: Dynamic Knowledge Distillation for Reliability in Federated LearningYueyang Yuan, Wenke Huang, Frank Wan, Kaiqi Guan et al.NeurIPS 2025 · 1 citation
Builds on25
- Supervised Contrastive LearningPrannay Khosla, Piotr Teterwak, Chen Wang, Aaron Sarna et al.NeurIPS 2020 · 7,049 citations
- CutMix: Regularization Strategy to Train Strong Classifiers With Localizable FeaturesSangdoo Yun, Dongyoon Han, Sanghyuk Chun, Seong Joon Oh et al.ICCV 2019 · 5,843 citations
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- Tackling the Objective Inconsistency Problem in Heterogeneous Federated OptimizationJianyu Wang, Qinghua Liu, Hao Liang, Gauri Joshi et al.NeurIPS 2020 · 2,231 citations
- Ensemble Distillation for Robust Model Fusion in Federated LearningTao Lin, Lingjing Kong, Sebastian U. Stich, Martin JaggiNeurIPS 2020 · 1,615 citations
Related papers
- Eliminating Domain Bias for Federated Learning in Representation SpaceJianqing Zhang, Yang Hua, Jian Cao, Hao Wang et al.NeurIPS 2023 · 105 citations
- BiasAdv: Bias-Adversarial Augmentation for Model DebiasingJongin Lim, Youngdong Kim, Byungjai Kim, Chanho Ahn et al.CVPR 2023
- Learning Debiased Representation via Disentangled Feature AugmentationJungsoo Lee, Eungyeup Kim, Juyoung Lee, Jihyeon Lee et al.NeurIPS 2021 · 203 citations
- FedMix: Approximation of Mixup under Mean Augmented Federated LearningTehrim Yoon, Sumin Shin, Sung Ju Hwang, Eunho YangICLR 2021 · 226 citations
- Bias Mitigation in Federated Learning for Edge ComputingYasmine Djebrouni, Nawel Benarba, Ousmane Touat, Pasquale De Rosa et al.UbiComp 2024 · 23 citations
