Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Preprint |
| Published: |
2021
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2108.05974 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866915298235383808 |
|---|---|
| author | Malekmohammadi, Saber Shaloudegi, Kiarash Hu, Zeou Yu, Yaoliang |
| author_facet | Malekmohammadi, Saber Shaloudegi, Kiarash Hu, Zeou Yu, Yaoliang |
| contents | Over the past few years, the federated learning ($\texttt{FL}$) community has witnessed a proliferation of new $\texttt{FL}$ algorithms. However, our understating of the theory of $\texttt{FL}$ is still fragmented, and a thorough, formal comparison of these algorithms remains elusive. Motivated by this gap, we show that many of the existing $\texttt{FL}$ algorithms can be understood from an operator splitting point of view. This unification allows us to compare different algorithms with ease, to refine previous convergence results and to uncover new algorithmic variants. In particular, our analysis reveals the vital role played by the step size in $\texttt{FL}$ algorithms. The unification also leads to a streamlined and economic way to accelerate $\texttt{FL}$ algorithms, without incurring any communication overhead. We perform numerical experiments on both convex and nonconvex models to validate our findings. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2108_05974 |
| institution | arXiv |
| publishDate | 2021 |
| record_format | arxiv |
| spellingShingle | An Operator Splitting View of Federated Learning Malekmohammadi, Saber Shaloudegi, Kiarash Hu, Zeou Yu, Yaoliang Machine Learning Over the past few years, the federated learning ($\texttt{FL}$) community has witnessed a proliferation of new $\texttt{FL}$ algorithms. However, our understating of the theory of $\texttt{FL}$ is still fragmented, and a thorough, formal comparison of these algorithms remains elusive. Motivated by this gap, we show that many of the existing $\texttt{FL}$ algorithms can be understood from an operator splitting point of view. This unification allows us to compare different algorithms with ease, to refine previous convergence results and to uncover new algorithmic variants. In particular, our analysis reveals the vital role played by the step size in $\texttt{FL}$ algorithms. The unification also leads to a streamlined and economic way to accelerate $\texttt{FL}$ algorithms, without incurring any communication overhead. We perform numerical experiments on both convex and nonconvex models to validate our findings. |
| title | An Operator Splitting View of Federated Learning |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2108.05974 |