arXiv:2411.09842cs.LG2024-11被引 2

通过模型回溯机制缓解分布式学习中的数据分布偏移问题。

FedRewind: Rewinding Continual Model Exchange for Decentralized Federated Learning

  • 节点定期将本地模型回传给接收方,形成双向信息交换。
  • 在多个基准上优于传统去中心化联邦学习方法。
  • 适合处理空间与时间双重数据分布变化的场景。

本文提出FedRewind,一种新型去中心化联邦学习方法,通过节点间模型交换缓解数据分布偏移问题。受持续学习和认知神经科学中记忆保留理论启发,该方法构建去中心化路由机制,使节点在本地训练期间定期将模型回传给此前接收模型的节点,进行有限迭代。此策略有效减小节点间数据分布差异,提升学习与泛化性能。我们在多个基准上评估该方法,结果表明其优于标准去中心化联邦学习及特定路由方案。此外,结合联邦学习与持续学习思想,该方法能应对更具挑战性的联邦持续学习任务——即同时存在空间与时间维度的数据分布变化,性能超越现有基线。

原文摘要 · Abstract (English)

In this paper, we present FedRewind, a novel approach to decentralized federated learning that leverages model exchange among nodes to address the issue of data distribution shift. Drawing inspiration from continual learning (CL) principles and cognitive neuroscience theories for memory retention, FedRewind implements a decentralized routing mechanism where nodes send/receive models to/from other nodes in the federation to address spatial distribution challenges inherent in distributed learning (FL). During local training, federation nodes periodically send their models back (i.e., rewind) to the nodes they received them from for a limited number of iterations. This strategy reduces the distribution shift between nodes' data, leading to enhanced learning and generalization performance. We evaluate our method on multiple benchmarks, demonstrating its superiority over standard decentralized federated learning methods and those enforcing specific routing schemes within the federation. Furthermore, the combination of federated and continual learning concepts enables our method to tackle the more challenging federated continual learning task, with data shifts over both space and time, surpassing existing baselines.

联邦学习持续学习去中心化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。