论文标题

FedCl:联合多相课程学习以同步关联用户异质性

FedCL: Federated Multi-Phase Curriculum Learning to Synchronously Correlate User Heterogeneity

论文作者

Wang, Mingjie, Guo, Jianxiong, Jia, Weijia

论文摘要

联合学习(FL)是一种分散的学习方法,用于训练机器学习算法。在FL中,一个全局模型迭代地收集本地模型的参数,而无需访问其本地数据。但是,FL中的一个重大挑战是处理局部数据分布的异质性,这通常会导致一个不难收敛的漂移全球模型。为了解决这个问题,当前的方法采用了不同的策略,例如知识蒸馏,加权模型聚合和多任务学习。这些方法被称为异步FL,因为它们在本地或事后将用户模型对齐,而模型漂移已经发生或被低估了。在本文中,我们提出了一种主动和同步的相关方法,以应对FL中用户异质性的挑战。具体而言,我们的方法旨在通过通过动态的多相课程在每个回合中积极并同步安排用户学习步伐来将FL作为标准深度学习。全局课程由自动回归自动编码器组成,该自动编码器集成了服务器上的所有用户课程。然后将此全球课程分为多个阶段,并向用户广播,以测量和对齐域 - 不可思议的学习步伐。实证研究表明,即使在存在严重的用户异质性的情况下,我们的方法在泛化性能方面的表现也优于现有的异步方法。

Federated Learning (FL) is a decentralized learning method used to train machine learning algorithms. In FL, a global model iteratively collects the parameters of local models without accessing their local data. However, a significant challenge in FL is handling the heterogeneity of local data distribution, which often results in a drifted global model that is difficult to converge. To address this issue, current methods employ different strategies such as knowledge distillation, weighted model aggregation, and multi-task learning. These approaches are referred to as asynchronous FL, as they align user models either locally or post-hoc, where model drift has already occurred or has been underestimated. In this paper, we propose an active and synchronous correlation approach to address the challenge of user heterogeneity in FL. Specifically, our approach aims to approximate FL as standard deep learning by actively and synchronously scheduling user learning pace in each round with a dynamic multi-phase curriculum. A global curriculum is formed by an auto-regressive auto-encoder that integrates all user curricula on the server. This global curriculum is then divided into multiple phases and broadcast to users to measure and align the domain-agnostic learning pace. Empirical studies demonstrate that our approach outperforms existing asynchronous approaches in terms of generalization performance, even in the presence of severe user heterogeneity.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源