Feb, 2024

基于随机逼近的联邦机器学习方法

TL;DR本论文研究了在随机逼近框架中的联邦学习(FL)。FL 是一种协作方式,可以在各个参与方或客户端之间训练神经网络模型,而不需要将它们的数据集中。每个客户端将在其相应数据上训练模型,并定期将权重发送至服务器进行聚合。服务器聚合这些权重,由客户端使用来重新初始化其神经网络并继续训练。该论文利用随机逼近迭代来更新客户端神经网络的权重,并展示了聚合权重跟踪自主 ODE。通过数值模拟并与 FedAvg 和 FedProx 等标准算法进行比较,观察到该算法具有鲁棒性,在客户端数据不完全相同时给出更可靠的权重估计。