Nov, 2023

FediOS: 在特征不平衡联邦学习中解耦正交子空间实现个性化

TL;DR个性化联邦学习(pFL)通过协同训练多个客户端来增强定制本地模型的能力。本文提出了一种名为 FediOS 的有效 pFL 方法,通过重新设计特征偏斜 pFL 的架构解耦设计,将模型分为两个特征提取器(通用和个性化)和一个共享预测头。通过正交投影将通用特征映射到一个公共子空间,并将个性化特征分散到不同的子空间以实现解耦。此外,在推理过程中,还训练了一个共享预测头来平衡通用特征和个性化特征的重要性。四个视觉数据集上的广泛实验证明了我们的方法在特征偏斜异质性下达到了最先进的 pFL 性能。