arXiv:2511.18066cs.LGcs.CV2025-11被引 1

解决联邦学习中类别不平衡下的测试时自适应问题。

pFedBBN: A Personalized Federated Test-Time Adaptation with Balanced Batch Normalization for Class-Imbalanced Data

  • 在本地适配中使用均衡批归一化,让所有类别平等处理。
  • 无需标签数据即可提升少数类识别性能,优于现有方法。
  • 适合隐私敏感场景,支持无监督个性化推理。

联邦学习中的测试时自适应(TTA)对应对客户端间未见数据分布至关重要,尤其在存在领域偏移和类别分布不均的情况下。类别不平衡(CI)是联邦学习的核心挑战,少数但关键的类别在各客户端数据中严重缺失。现有方法虽在训练阶段通过可靠聚合和局部类别分布对齐缓解此问题,但通常依赖标注数据或客户端协作,且无法处理推理时的无监督动态域适应与分布偏移。本文揭示了现有先进TTA在联邦类别不平衡场景下的失效,并提出pFedBBN:一种个性化联邦测试时自适应框架,通过本地适配中引入均衡批归一化(BBN),使所有类别受同等对待,缓解预测偏差;同时基于BBN相似性引导客户端协作,确保具有相似平衡表示的客户端相互强化,保持与特定域特征一致。该方法支持完全无监督本地适配,并引入类别感知模型聚合策略,实现个性化推理而不泄露隐私。pFedBBN通过均衡特征归一化与域感知协作,同时应对分布偏移与类别不平衡,无需任何客户端的标签或原始数据。跨多种基线的大量实验表明,pFedBBN持续提升鲁棒性及少数类表现,优于当前最先进联邦学习与测试时自适应方法。

原文摘要 · Abstract (English)

Test-time adaptation (TTA) in federated learning (FL) is crucial for handling unseen data distributions across clients, particularly when faced with domain shifts and skewed class distributions. Class Imbalance (CI) remains a fundamental challenge in FL, where rare but critical classes are often severely underrepresented in individual client datasets. Although prior work has addressed CI during training through reliable aggregation and local class distribution alignment, these methods typically rely on access to labeled data or coordination among clients, and none address class unsupervised adaptation to dynamic domains or distribution shifts at inference time under federated CI constraints. Revealing the failure of state-of-the-art TTA in federated client adaptation in CI scenario, we propose pFedBBN,a personalized federated test-time adaptation framework that employs balanced batch normalization (BBN) during local client adaptation to mitigate prediction bias by treating all classes equally, while also enabling client collaboration guided by BBN similarity, ensuring that clients with similar balanced representations reinforce each other and that adaptation remains aligned with domain-specific characteristics. pFedBBN supports fully unsupervised local adaptation and introduces a class-aware model aggregation strategy that enables personalized inference without compromising privacy. It addresses both distribution shifts and class imbalance through balanced feature normalization and domain-aware collaboration, without requiring any labeled or raw data from clients. Extensive experiments across diverse baselines show that pFedBBN consistently enhances robustness and minority-class performance over state-of-the-art FL and TTA methods.

联邦学习测试时自适应类别不平衡无监督学习

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