mlsanityData to 14 Oct 2026

Inner Momentum for Differentially Private Muon

arXiv 2610.02738 ↗First posted 2 Oct 2026cs.LG cs.CRcited by 0

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

  1. 2016Deep Learning with Differential Privacycited by 2,728
  2. 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 1,040
  3. 2025Kimi K2: Open Agentic Intelligencecited by 567
  4. 2025Muon is Scalable for LLM Trainingcited by 428
  5. 2021Large Language Models Can Be Strong Differentially Private Learnerscited by 412
  6. 2017The E2E Dataset: New Challenges For End-to-End Generationcited by 253
  7. 2021Numerical Composition of Differential Privacycited by 180
  8. 2020DART: Open-Domain Structured Data Record to Text Generationcited by 145
  9. 2020Understanding Gradient Clipping in Private SGD: A Geometric Perspectivecited by 142
  10. 2022Automatic Clipping: Differentially Private Deep Learning Made Easier and Strongercited by 88
  11. 2026Convergence of Muon with Newton-Schulzcited by 36
  12. 2023Differentially Private SGD Without Clipping Bias: An Error-Feedback Approachcited by 14
3 more
  1. 2024DiSK: Differentially Private Optimizer with Simplified Kalman Filter for Noise Reductioncited by 11
  2. 2026DP-Muon: Differentially Private Optimization via Matrix-Orthogonalized Momentumcited by 2
  3. 2025Enhancing DPSGD via Per-Sample Momentum and Low-Pass Filteringcited by 1

Built on by

Not cited yet.