Nov, 2023

稳定 FDG: 基于风格和注意力的联合域泛化学习

TL;DR传统联邦学习算法假设训练和测试数据分布相同,然而现实中领域漂移的发生要求联邦学习方法具备领域泛化能力。本文提出了基于样式和注意力的学习策略 StableFDG,以实现联邦领域泛化,并引入了两个关键贡献:一是基于样式的学习,使每个客户端能够在本地数据集中探索新的样式,通过样式共享、转移和探索策略改进领域多样性;二是基于注意力的特征强调器,捕捉同一类别样本特征的相似性,并强调重要 / 共同特征,以更好地学习数据贫乏联邦学习场景中每个类别的领域不变特征。实验结果表明,StableFDG 在各种领域泛化基准数据集上优于现有的基线算法,证明其有效性。