LLM2D
Home
Arxiv
返回列表
分层联邦 ADMM
Hierarchical Federated ADMM
作者:
Seyed Mohammad Azimi-Abarghouyi, Nicola Bastianello, Karl H. Johansson, Viktoria Fodor
发布日期:
9/30/2024
arXiv ID:
oai:arXiv.org:2409.18796v1
摘要
本文抛弃了广泛使用的基于梯度下降的分层联邦学习 (FL) 算法,提出了一种基于交替方向乘子法 (ADMM) 的新型分层 FL 框架。在此框架内,我们提出了两种新颖的 FL 算法,它们都在顶层使用 ADMM:一种在底层使用 ADMM,另一种使用传统的基于梯度下降的方法。该框架增强了隐私,实验表明,与传统算法相比,所提出的算法在学习收敛和准确性方面具有优越性。此外,即使局部步骤数量非常有限,底层的梯度下降也能表现良好,而两层的 ADMM 则在其他情况下表现更好。
查看原文
下载 PDF