mlsanityData to 14 Oct 2026

Muon-C: Operator-Aligned Muon for Convolutional Kernels

arXiv 2609.09676 ↗First posted 9 Sep 2026cs.LG stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Muon replaces matrix momentum with an approximately orthogonal polar direction, but its geometry depends on the matrix representation. For convolution, standard unfolding describes a local patch map rather than the convolution operator. We introduce Muon-C, an operator-aligned optimizer that represents kernel momentum as frequency-wise channel-transfer matrices, polarizes these blocks independently, and uses a critical Fourier grid to return updates exactly to the original finite kernel support. We show that the new geometry arises from combining the block partition and Fourier coordinates. The exact-polar direction is a linear minimization oracle under the critically sampled convolution norm. Its worst-case guarantee relative to the continuous convolution-operator norm is never weaker than unfolding and is strictly stronger for 3×3 kernels. On CIFAR-10 flow matching with matched applied-update RMS, Muon-C reaches 9.87 FID at 40k iterations, compared with 22.26 for unfolded Muon and 51.31 for Adam. It reaches their final quality using 0.62× and 0.64× their model FLOPs, respectively. Under equal tuning budgets, Muon-C achieves 3.42 FID. Gains persist across data scales and transfer to classification across convolutional architectures.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 11 of its 17 references that are in the map. First posted on arXiv 9 Sep 2026.

Builds on

11 of 17 references.

References · 11

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  3. 2022Flow Matching for Generative Modelingcited by 5,702
  4. 2022A ConvNet for the 2020scited by 3,579
  5. 2018Spectral Normalization for Generative Adversarial Networkscited by 1,636
  6. 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
  7. 2023ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoderscited by 712
  8. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
  9. 2025Training Deep Learning Models with Norm-Constrained LMOscited by 192
  10. 2018The Singular Values of Convolutional Layerscited by 95
  11. 2024Modular Duality in Deep Learningcited by 77

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Muon: Training and Trade-offs with Latent Attention and MoE
  2. 2026Muon2: Boosting Muon via Adaptive Second-Moment Preconditioning
  3. 2026An Isotropy-Preserving Spectral Cap for Muon: Theory and Three Case Studies
  4. 2026How Much Orthogonalization Does Muon Need?
  5. 2026MuonQ: Enhancing Low-Bit Muon Quantization via Directional Fidelity Optimization
  6. 2025Effective Quantization of Muon Optimizer States
  7. 2026AF-Muon: An AdamW-Free Muon Optimizer for Tied-Embedding Models
  8. 2026Scaling Muon for Diffusion Transformers