LLM2D
边缘设备上具有领域异质性的鲁棒联邦学习
Robust Federated Learning on Edge Devices with Domain Heterogeneity
作者: Huy Q. Le, Latif U. Khan, Choong Seon Hong
发布日期: 5/16/2025
arXiv ID: oai:arXiv.org:2505.10128v1

摘要

arXiv:2505.10128v1 宣告类型: cross 摘要: 联邦学习(FL)允许分布式边缘设备在确保数据隐私的同时进行协作训练,使其成为敏感隐私应用的流行解决方案。然而,FL面临着统计异质性的重大挑战,特别是域异质性,这阻碍了全局模式的收敛。在本研究中,我们引入了一种新的框架,通过在域异质性下改进FL全局模型的泛化能力来解决这一挑战,使用原型增强。具体地,我们提出了FedAPC(联邦增强原型对比学习),这是一种基于原型的FL框架,旨在增强特征多样性和模型鲁棒性。FedAPC 利用来自增强数据均值特征的原型来捕获更丰富的表示。通过使局部特征与全局原型对齐,我们使模型能够学习有意义的语义特征,同时减少对任何特定域的过拟合。Office-10 和 Digits 数据集上的实验结果表明,我们的框架优于现有最佳基线,展现了优越的性能。