English

SUDA-Muon: Structural Design Principles and Boundaries for Fully Decentralized Muon

Optimization and Control 2026-04-28 v1

Abstract

Fully decentralized Muon is difficult because its nonlinear matrix-sign operator does not commute with linear gossip averaging. This makes decentralized Muon a structural design problem: in designing the algorithm, one must distinguish modular components from non-modular ones. We propose \sudamuon{}, which realizes this separation through a unified primal--dual communication template called SUDA; within this template, ED/D2^2, EXTRA, and gradient tracking become modular backbone choices. We prove a topology-separated non-asymptotic convergence guarantee in the nuclear-norm geometry: the dominant term scales as O((1+σ/N)K1/4)\mathcal{O}((1+\sigma/\sqrt{N})K^{-1/4}) and does not explicitly involve graph quantities, identifying the communication backbone as the modular axis in the structure design. We then establish two complementary non-modular boundaries. Internally, tracking-before-polarization is necessary for this natural no-tracking variant to avoid non-stationary fixed points under heterogeneous objectives. Externally, in the absence of a central server, a fully decentralized method cannot perform the federated average-then-polarize update; we show that this non-modular local-polarize-then-average design is the essential reason why can fail to exhibit linear speedup. Experiments on CIFAR-100 and GPT-2 fine-tuning support the same picture: the unified template makes different communication algorithms directly comparable. In mild near-IID regimes, the resulting variants perform similarly, while in the more difficult long-horizon non-IID CIFAR-100 setting, \sudamuon{} achieves higher accuracy and lower loss than \textsc{DeMuon}.

Keywords

Cite

@article{arxiv.2604.23980,
  title  = {SUDA-Muon: Structural Design Principles and Boundaries for Fully Decentralized Muon},
  author = {Hengrui Zhang and Boao Kong and Jiahe Geng and Zhengyang Huang},
  journal= {arXiv preprint arXiv:2604.23980},
  year   = {2026}
}
R2 v1 2026-07-01T12:36:15.377Z