Inner Momentum for Differentially Private Muon
Not on the map. Why?
Abstract (arXiv)
Differentially private training clips each per-example gradient before adding noise. This clipping is radial for each example, yet unequal clipping factors can distort the relative singular-vector geometry of their average. Muon is particularly exposed to this effect, since its update is an approximate polar factor UV^T that depends only on the singular vectors that clipping can shift. To curb this degradation, we propose averaging each sampled example's Muon gradient over the current model and a short history of recent models before clipping. The clipped batch matrix then separates into a common rescaling and a covariance residual R between sampled gradients and clipping values, with ||R||_F <= sigma_lambda sigma_G, bounding the clipping-induced distortion directly. We further show that a finite Newton-Schulz iteration preserves the polar factor of its input under these spectral conditions, confirming that our correction survives orthogonalization. In private GPT-2 fine-tuning on E2E and DART at epsilon in {1, 2, 4, 8}, DP-Muon-IM improves BLEU and ROUGE-L over DP-Muon in every seed-matched comparison, and non-private diagnostics show 2-4% lower pre-noise polar error.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 15 of its 22 references that are in the map. First posted on arXiv 2 Oct 2026.
Builds on
15 of 22 references.
References · 15
- 2016Deep Learning with Differential Privacycited by 2,728
- 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 1,040
- 2025Kimi K2: Open Agentic Intelligencecited by 567
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2021Large Language Models Can Be Strong Differentially Private Learnerscited by 412
- 2017The E2E Dataset: New Challenges For End-to-End Generationcited by 253
- 2021Numerical Composition of Differential Privacycited by 180
- 2020DART: Open-Domain Structured Data Record to Text Generationcited by 145
- 2020Understanding Gradient Clipping in Private SGD: A Geometric Perspectivecited by 142
- 2022Automatic Clipping: Differentially Private Deep Learning Made Easier and Strongercited by 88
- 2026Convergence of Muon with Newton-Schulzcited by 36
- 2023Differentially Private SGD Without Clipping Bias: An Error-Feedback Approachcited by 14
Built on by
Not cited yet.