arXiv:2602.01039cs.LGcs.AI2026-02被引 1

通过分布外检测动态加权,提升异构联邦学习的模型稳定性和泛化能力。

Adaptive Dual-Weighting Framework for Federated Learning via Out-of-Distribution Detection

  • 客户端用伪分布外样本加权损失,增强对异常数据的鲁棒性。
  • 服务器根据客户端分布一致性加权聚合,提升全局模型稳定性。
  • 可无缝集成现有算法,适合真实场景中异构数据的联邦学习。

联邦学习(FL)在保护数据隐私的前提下实现跨大规模分布式节点的协作训练,是边缘-云智能服务系统的核心。然而,在实际服务部署中,用户、设备和应用场景产生的数据具有显著非独立同分布(non-IID)特性,严重损害全局模型的收敛稳定性与泛化能力。为此,我们提出FLood框架,受分布外(OOD)检测启发,采用双权重机制动态缓解异构性影响:客户端自适应地对监督损失进行加权,提高伪分布外样本权重,从而增强对分布不一致或困难样本的鲁棒学习;服务器则依据客户端的OOD置信度评分加权聚合,优先采纳分布一致性高的更新,提升全局模型鲁棒性与收敛稳定性。在多种基准与非IID设置下的大量实验表明,FLood在准确率与泛化性能上持续优于现有先进方法。此外,该框架可作为正交插件模块,无需修改核心优化逻辑即可与现有联邦算法集成,显著提升其在异构环境下的表现,为真实联邦环境中可靠智能服务的部署提供了实用且可扩展的解决方案。

原文摘要 · Abstract (English)

Federated Learning (FL) enables collaborative model training across large-scale distributed service nodes while preserving data privacy, making it a cornerstone of intelligent service systems in edge-cloud environments. However, in real-world service-oriented deployments, data generated by heterogeneous users, devices, and application scenarios are inherently non-IID. This severe data heterogeneity critically undermines the convergence stability, generalization ability, and ultimately the quality of service delivered by the global model. To address this challenge, we propose FLood, a novel FL framework inspired by out-of-distribution (OOD) detection. FLood dynamically counteracts the adverse effects of heterogeneity through a dual-weighting mechanism that jointly governs local training and global aggregation. At the client level, it adaptively reweights the supervised loss by upweighting pseudo-OOD samples, thereby encouraging more robust learning from distributionally misaligned or challenging data. At the server level, it refines model aggregation by weighting client contributions according to their OOD confidence scores, prioritizing updates from clients with higher in-distribution consistency and enhancing the global model's robustness and convergence stability. Extensive experiments across multiple benchmarks under diverse non-IID settings demonstrate that FLood consistently outperforms state-of-the-art FL methods in both accuracy and generalization. Furthermore, FLood functions as an orthogonal plug-in module: it seamlessly integrates with existing FL algorithms to boost their performance under heterogeneity without modifying their core optimization logic. These properties make FLood a practical and scalable solution for deploying reliable intelligent services in real-world federated environments.

联邦学习异构数据分布外检测模型鲁棒性

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