On Provable Benefits of Muon in Federated Learning
Not on the map. Why?
Abstract (arXiv)
The recently introduced optimizer, Muon, has gained increasing attention due to its superior performance across a wide range of applications. However, its effectiveness in federated learning remains unexplored. To address this gap, this paper investigates the performance of Muon in the federated learning setting. Specifically, we propose a new algorithm, FedMuon, and establish its convergence rate for nonconvex problems. Our theoretical analysis reveals multiple favorable properties of FedMuon. In particular, due to its orthonormalized update direction, the learning rate of FedMuon is independent of problem-specific parameters, and, importantly, it can naturally accommodate heavy-tailed noise. The extensive experiments on a variety of neural network architectures validate the effectiveness of the proposed algorithm.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Builds on 26 of its 37 references that are in the map. First posted on arXiv 4 Oct 2025.
Builds on
26 of 37 references.
References · 26
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2020Adaptive Federated Optimizationcited by 1,046
- 2019Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classificationcited by 851
- 2018Local SGD Converges Fast and Communicates Littlecited by 605
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2018Parallel Restarted SGD with Faster Convergence and Less Communication: Demystifying Why Model Averaging Works for Deep Learningcited by 278
- 2021Achieving Linear Speedup with Partial Worker Participation in Non-IID Federated Learningcited by 193
- 2024Old Optimizer, New Norm: An Anthologycited by 193
- 2025Training Deep Learning Models with Norm-Constrained LMOscited by 192
- 2019On the Linear Speedup Analysis of Communication Efficient Momentum SGD for Distributed Non-Convex Optimizationcited by 180
14 more
- 2024Why Transformers Need Adam: A Hessian Perspectivecited by 124
- 2025Understanding Gradient Orthogonalization for Deep Learning via Non-Euclidean Trust-Region Optimizationcited by 92
- 2025Muon Optimizes Under Spectral Norm Constraintscited by 86
- 2025A Note on the Convergence of Muoncited by 83
- 2024Heavy-Tailed Class Imbalance and Why Adam Outperforms Gradient Descent on Language Modelscited by 80
- 2025Gluon: Making Muon & Scion Great Again! (Bridging Theory and Practice of LMO-based Optimizers for LLMs)cited by 72
- 2025ASGO: Adaptive Structured Gradient Optimizationcited by 57
- 2021STEM: A Stochastic Two-Sided Momentum Algorithm Achieving Near-Optimal Sample and Communication Complexities for Federated Learningcited by 43
- 2024Nonconvex Stochastic Optimization under Heavy-Tailed Noises: Optimal Convergence without Gradient Clippingcited by 42
- 2025Lions and Muons: Optimization via Stochastic Frank-Wolfe under Heavy-Tailed Noisecited by 41
- 2023Momentum Benefits Non-IID Federated Learning Simply and Provablycited by 38
- 2022Faster Adaptive Federated Learningcited by 35
- 2025AdaGrad Meets Muon: Adaptive Stepsizes for Orthogonal Updatescited by 23
- 2025Efficient Distributed Optimization under Heavy-Tailed Noisecited by 7
Built on by
Citing papers per year: 2025: 2; 2026: 5.
7 papers.
- 2025Isotropic Curvature Model for Understanding Deep Learning Optimization: Is Gradient Orthogonalization Optimal?cited by 37
- 2025Non-Euclidean SGD for Structured Optimization: Unified Analysis and Improved Ratescited by 8
- 2026When and Why SignSGD Outperforms SGD: A Theoretical Study Based on ℓ1-norm Lower Boundscited by 2
- 2026A Note on Stability for Orthogonalized Matrix Momentum with Client Samplingcited by 1
- 2026Sign compression for Muon: SignMuon, MuonSign, and the Limits of Error Feedbackcited by 0
- 2026Federated Compositional Muon Optimizer for Matrix-Wise Modelscited by 0
- 2026FedSubMuon: Communication-Efficient Federated LLM Fine-Tuning via Structured Subspace Muoncited by 0
- 2026FedSubMuon: Communication-Efficient Federated LLM Fine-Tuning via Structured Subspace Muon5 Sep 2026
- 2026Federated Compositional Muon Optimizer for Matrix-Wise Models13 Aug 2026
- 2026Sign compression for Muon: SignMuon, MuonSign, and the Limits of Error Feedback31 Jul 2026
- 2026A Note on Stability for Orthogonalized Matrix Momentum with Client Sampling1 Jun 2026
- 2026When and Why SignSGD Outperforms SGD: A Theoretical Study Based on ℓ1-norm Lower Bounds7 May 2026
- 2025Non-Euclidean SGD for Structured Optimization: Unified Analysis and Improved Rates14 Nov 2025
- 2025Isotropic Curvature Model for Understanding Deep Learning Optimization: Is Gradient Orthogonalization Optimal?1 Nov 2025