arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.02416cs.LGcs.AI
收藏

迈向快速稳定的联邦学习:通过知识锚点应对异构性

Towards Fast and Stable Federated Learning: Confronting Heterogeneity via Knowledge Anchor

Fuzhou University(福州大学)

Jinqian Chen, Jihua Zhu, Qinghai Zheng

AI总结:

本文分析联邦本地训练中缺失与非主导类别的严重遗忘,提出 FedKA 知识锚点算法校正梯度,实现更快更稳收敛并提升基准准确率。

摘要:

联邦学习面临数据异构性这一关键挑战,对联邦模型的性能和收敛造成不利影响。已有多种方法被提出以解决该问题,但其有效性仍然有限。近期研究表明,联邦模型在本地训练中会遭受严重遗忘,进而导致全局遗忘和性能下降。尽管这些分析提供了有价值的洞见,但对脆弱类别及其影响因素仍缺乏全面理解。本文旨在通过系统分析不同通信轮次中本地训练期间每个类别的遗忘程度来弥合这一差距。我们的观察包括:(1)缺失类别和非主导类别在本地训练期间都会遭受类似的严重遗忘,而主导类别的性能则有所提升。(2)当动态减少主导类别的样本数量时,一旦其样本比例低于某个阈值,就会突然发生灾难性遗忘,这表明本地模型难以有效利用特定类别的少量样本来防止遗忘。受这些发现启发,我们提出一种新颖且简单的算法,称为 Federated Knowledge Anchor(FedKA,联邦知识锚点)。该算法假设所有客户端对每个类别都拥有单个共享样本;在每次本地训练阶段之前,通过抽取缺失类别的共享样本,并为非主导类别从每类随机选择一个样本来构建知识锚点。随后,知识锚点被用于校正每个小批量的梯度,使其朝保留缺失类别和非主导类别知识的方向更新。大量实验结果表明,所提出的 FedKA 实现了快速且稳定的收敛,并在常用基准测试上显著提高了准确率。

英文摘要:

Federated learning encounters a critical challenge of data heterogeneity, adversely affecting the performance and convergence of the federated model. Various approaches have been proposed to address this issue, yet their effectiveness is still limited. Recent studies have revealed that the federated model suffers severe forgetting in local training, leading to global forgetting and performance degradation. Although the analysis provides valuable insights, a comprehensive understanding of the vulnerable classes and their impact factors is yet to be established. In this paper, we aim to bridge this gap by systematically analyzing the forgetting degree of each class during local training across different communication rounds. Our observations are: (1) Both missing and non-dominant classes suffer similar severe forgetting during local training, while dominant classes show improvement in performance. (2) When dynamically reducing the sample size of a dominant class, catastrophic forgetting occurs abruptly when the proportion of its samples is below a certain threshold, indicating that the local model struggles to leverage a few samples of a specific class effectively to prevent forgetting. Motivated by these findings, we propose a novel and straightforward algorithm called Federated Knowledge Anchor (FedKA). Assuming that all clients have a single shared sample for each class, the knowledge anchor is constructed before each local training stage by extracting shared samples for missing classes and randomly selecting one sample per class for non-dominant classes. The knowledge anchor is then utilized to correct the gradient of each mini-batch towards the direction of preserving the knowledge of the missing and non-dominant classes. Extensive experimental results demonstrate that our proposed FedKA achieves fast and stable convergence, significantly improving accuracy on popular benchmarks.

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

↑