SignMuon: Communication-Efficient Distributed Muon Optimization
Not on the map. Why?
Abstract (arXiv)
Distributed training of large neural networks is bottlenecked by full-precision gradient communication and by coordinatewise optimizers that ignore the matrix structure of weight tensors. We propose Sign-Muon, a 1-bit, matrix-aware optimizer that combines majority-vote sign aggregation from signSGD with the polar-step framework of Muon. Each worker forms a Muon-style direction by taking the polar factor of its momentum via a Newton--Schulz iteration, transmits only the entrywise signs, and aggregates by majority vote; an optional local polar step further enforces orthogonality at no extra communication cost. Under spectral-norm smoothness and bounded-variance stochastic gradients, the spectral-norm normalized sign step yields an O(1/√T) nonconvex rate for an ℓ1-based stationarity measure. With unimodal symmetric noise, majority vote across M workers cuts the stochastic term by 1/√M, matching signSGD. In the α-β model, distributed Sign-Muon needs only one integer sum-allreduce per iteration; all orthogonalization is local, giving a 32× bandwidth reduction over float32 (4× for int8). Across 330 CIFAR-10/ResNet-50 configurations Sign-Muon attains the best validation accuracy (92.15%); its 4-GPU majority-vote variant reaches 92.02% with 37% less training time at matched effective batch. On nanoGPT, Sign-Muon achieves lower perplexity and better anytime performance than other sign-based baselines, with favorable weak-scaling up to 16 GPUs.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 14 of its 35 references that are in the map. First posted on arXiv 4 May 2026.
Builds on
14 of 35 references.
References · 14
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2017Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Trainingcited by 478
- 2023Symbolic Discovery of Optimization Algorithmscited by 434
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
- 2018Sparsified SGD with Memorycited by 376
- 2017TernGrad: Ternary Gradients to Reduce Communication in Distributed Deep Learningcited by 256
- 2019PowerSGD: Practical Low-Rank Gradient Compression for Distributed Optimizationcited by 238
- 2022Effects of Spectral Normalization in Multi-agent Reinforcement Learningcited by 6
- 2023Adaptive Consensus Optimization Method for GANscited by 3
- 2023Angle based dynamic learning rate for gradient descentcited by 2
- 2023Light-weight Deep Extreme Multilabel Classificationcited by 2
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026An Isotropy-Preserving Spectral Cap for Muon: Theory and Three Case Studies
- 2025Muon: Training and Trade-offs with Latent Attention and MoE
- 2026Scaling Muon for Diffusion Transformers
- 2026Orthogonal Witness Control for Muon Optimization via Sigmoid Spectral Reshaping
- 2025Muon is Scalable for LLM Training
- 2026Training-Loss Guarantees for Muon with Finite-Step Newton--Schulz Orthogonalization
- 2026Muon Meets Mamba: Spectral Optimization for State Space Models
- 2026Muon2: Boosting Muon via Adaptive Second-Moment Preconditioning