用语义引导的特征混合减少联邦学习长尾数据偏差
FedSM: Robust Semantics-Guided Feature Mixup for Bias Reduction in Federated Learning with Long-Tail Data
- 基于预训练图文对齐模型,按语义相关性选择混合特征
- 在多个长尾数据集上准确率超越现有方法,抗域偏移能力强
- 本地计算为主,适合资源受限的边缘设备使用
联邦学习(FL)可在不共享私有数据的前提下实现跨客户端协同训练,但因数据非独立同分布和长尾分布导致全局模型产生偏差。本文提出一种客户端为中心的框架FedSM,通过语义引导的特征混合与轻量级分类器重训练来缓解该问题。利用预训练图像-文本对齐模型计算类别级语义相关性,指导本地特征与全局原型的混合,生成类别一致的伪特征,修正分类器偏差,尤其在数据严重倾斜时效果显著。为应对预训练模型与实际数据间的潜在域偏移,引入概率化类别选择机制,提升特征多样性以更有效缓解偏差。所有计算均在本地完成,服务器开销极小。在多种不平衡程度的长尾数据集上进行的大量实验表明,FedSM在准确率上持续优于现有先进方法,且对域偏移具有强鲁棒性和高计算效率。
原文摘要 · Abstract (English)
Federated Learning (FL) enables collaborative model training across decentralized clients without sharing private data. However, FL suffers from biased global models due to non-IID and long-tail data distributions. We propose \textbf{FedSM}, a novel client-centric framework that mitigates this bias through semantics-guided feature mixup and lightweight classifier retraining. FedSM uses a pretrained image-text-aligned model to compute category-level semantic relevance, guiding the category selection of local features to mix-up with global prototypes to generate class-consistent pseudo-features. These features correct classifier bias, especially when data are heavily skewed. To address the concern of potential domain shift between the pretrained model and the data, we propose probabilistic category selection, enhancing feature diversity to effectively mitigate biases. All computations are performed locally, requiring minimal server overhead. Extensive experiments on long-tail datasets with various imbalanced levels demonstrate that FedSM consistently outperforms state-of-the-art methods in accuracy, with high robustness to domain shift and computational efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。