arXiv:2606.26037stat.MLcs.CV2026-06

通过特征自适应重标注,解决联邦学习中的类别不平衡问题。

FedReLa: Imbalanced Federated Learning via Re-Labeling

论文配图:FedReLa: Imbalanced Federated Learning via Re-Labeling
图 1 · 摘自论文原文
  • 用特征依赖的重标注器动态调整样本标签。
  • 在极端类别缺失下仍显著提升少数类准确率。
  • 无需全局分布信息,兼容现有方法且不增加通信开销。

联邦学习已成为保护隐私的去中心化模型训练主流方法。全局类别不平衡与跨客户端数据异质性天然共存,本地与全局不平衡的不匹配加剧了聚合模型性能下降。全局类别分布未知给数据级方法带来巨大挑战,尤其在客户端严重缺失某些类别的极端情况下。本文提出 FedReLa,一种新型数据级方法,应对联邦学习中数据异质性与类别不平衡的共存问题。通过特征依赖的标签重分配器,FedReLa 在无需全局类别分布知识的前提下,修正了有偏的全局决策边界。该模块化、模型无关的方法可无缝集成至算法级方法,实现持续提升且无额外通信开销。大量实验表明,该方法显著提高步进式不平衡和长尾数据集上少数类及整体准确率,优于先前最先进方法。

原文摘要 · Abstract (English)

Federated learning has emerged as the foremost approach for decentralized model training with privacy preservation. The global class imbalance and cross-client data heterogeneity naturally coexist, and the mismatch between local and global imbalances exacerbates the performance degradation of the aggregated model. The agnosticism of global class distribution poses significant challenges for data-level methods, especially under extreme conditions with severe class absence across clients. In this paper, we propose FedReLa, a novel data-level approach that tackles the coexistence of data heterogeneity and class imbalance in federated learning. By re-labeling samples with a feature-dependent label re-allocator, FedReLa corrects biased global decision boundaries without requiring knowledge of the global class distribution. This modular, model-agnostic approach can be integrated with algorithmic methods to deliver consistent improvements without additional communication overhead. Through extensive experiments, our method significantly improves the accuracy of minority classes and the overall accuracy on stepwise-imbalanced and long-tailed datasets, outperforming the previous state of the art.

联邦学习类别不平衡重标注数据异质性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。