cs.LGOct 5, 2026

Muon Is Theoretically Wrong For Convolutions, But Empirically Effective

Authors: Thibaut Boissin, Thomas Massena, Mathieu Serrurier, Franck Mamalet

Organizations: IRIT · IRIT, DTIPG - SNCF, UT3 · Institut de Recherche Technologique Saint-Exupery, France

Abstract

Muon, an optimizer known for its efficiency, has a clear interpretation for matrix-valued updates, but convolutional kernels are stored as four-dimensional tensors. Standard implementations reshape these tensors into matrices, a shortcut which breaks the theoretical understanding behind Muon. To investigate this, we formalize the corresponding optimization objective directly in convolutional operator geometry and introduce Convolutional Newton-Schulz (Conv-NS), which approximates the polar factor in this geometry while preserving kernel support. When applied in fast training experiments, Conv-NS and reshape-based Muon are both computationally efficient and achieve comparable accuracy on CIFAR-10 and ImageNet classification tasks. However, as one could expect a theoretically aligned Conv-NS to outperform reshape-based Muon, we investigate this mismatch between practice and theoretical understanding, with the hypothesis that exact convolutional orthogonalization may overconstrain updates. These findings highlight Muon's strong practical performance while opening directions for its further development on convolutions. Our code is publicly available at github conv-muon.

Explore similar work

CardsList
  1. Muon-C: Operator-Aligned Muon for Convolutional Kernels

    Sep 9, 2026Jiaxin Qing, Lexin LiMuonDeep Unfolding Networks

  2. Turbo-Muon: Almost-Orthogonal Pre-Conditioning for Fast Muon Updates

    Dec 4, 2025Thibaut Boissin, Thomas Massena, Franck Mamalet +1MuonSpectral Preconditioning

  3. Muon as a Residual Connection

    Jul 1, 2026Hao HuangMuonResidual Connections