arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.05049cs.LG

有原则的联邦领域适应:梯度投影与自动加权

Principled Federated Domain Adaptation: Gradient Projection and Auto-Weighting

UIUC(伊利诺伊大学厄巴纳-香槟分校);Stanford(斯坦福大学)

Enyi Jiang, Yibo Jacky Zhang, Sanmi Koyejo

AI总结:

针对联邦领域适应中领域偏移与数据稀缺导致传统方法失效的问题,提出轻量级聚合规则FedGP与自动加权方案,通过理论框架与实验验证其在提升目标客户端性能上的有效性。

摘要:

联邦领域适应(FDA)描述了这样的联邦学习(FL)场景:源客户端和服务器协同工作,以提升数据有限的特定目标客户端的性能。源域与目标域之间的领域偏移,加上目标客户端的数据稀缺,使得FDA成为一个极具挑战性的问题,例如,联邦平均和微调等常见技术会因领域偏移和数据稀缺而失效。为了从理论上理解该问题,我们引入了刻画FDA设定的新指标,以及一个包含新颖定理的理论框架,用于分析服务器聚合规则的性能。此外,我们提出了一种新型轻量级聚合规则Federated Gradient Projection(FedGP),它在领域偏移和数据稀缺的情况下显著提升了目标性能。并且,我们的理论提出了一种自动加权方案,能够找到源梯度和目标梯度的最优组合。该方案同时改进了FedGP和一种较简单的启发式聚合规则。大量实验验证了理论见解,并证明了所提方法在实际中的有效性。

英文摘要:

Federated Domain Adaptation (FDA) describes the federated learning (FL) setting where source clients and a server work collaboratively to improve the performance of a target client where limited data is available. The domain shift between the source and target domains, coupled with limited data of the target client, makes FDA a challenging problem, e.g., common techniques such as federated averaging and fine-tuning fail due to domain shift and data scarcity. To theoretically understand the problem, we introduce new metrics that characterize the FDA setting and a theoretical framework with novel theorems for analyzing the performance of server aggregation rules. Further, we propose a novel lightweight aggregation rule, Federated Gradient Projection ($\texttt{FedGP}$), which significantly improves the target performance with domain shift and data scarcity. Moreover, our theory suggests an $\textit{auto-weighting scheme}$ that finds the optimal combinations of the source and target gradients. This scheme improves both $\texttt{FedGP}$ and a simpler heuristic aggregation rule. Extensive experiments verify the theoretical insights and illustrate the effectiveness of the proposed methods in practice.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑