LLM2D
联邦学习中自适应优化的参数跟踪
Parameter Tracking in Federated Learning with Adaptive Optimization
作者: Evan Chen. Jianing Zhang, Shiqiang Wang, Chaoyue Liu, Christopher Brinton
发布日期: 2/6/2025
arXiv ID: oai:arXiv.org:2502.02727v1

摘要

arXiv:2502.02727v1 类型: cross 摘要: 在联邦学习(FL)中,模型训练性能受到客户端数据异质性的影响极大。梯度跟踪(GT)最近作为一种解决方案出现,通过在局部模型更新中引入校正项来缓解这一问题。到目前为止,GT仅被考虑在基于随机梯度下降(SGD)的模型训练中,而现代FL框架越来越多地采用自适应优化器以提高收敛性能。在本文中,我们泛化了GT框架,提出了一个更灵活的参数跟踪(PT)范式,并提出了一种基于Adam的自适应优化算法FAdamET和FAdamGT,将PT整合到基于Adam的FL中。我们在非凸环境下提供了这两种算法的严格收敛分析。我们的实验结果表明,在不同数据异质性水平下,这两种算法在评估总通信成本和总计算成本时均优于现有方法,显示了在联邦自适应优化中纠正一阶信息的有效性。