English
Related papers

Related papers: Distance-Aware Muon: Adaptive Step Scaling for Nor…

200 papers

Muon is a recently proposed optimizer that enforces orthogonality in parameter updates by projecting gradients onto the Stiefel manifold, leading to stable and efficient training in large-scale deep neural networks. Meanwhile, the…

Machine Learning · Computer Science 2026-03-18 Hideaki Iiduka

Orthogonality-based optimizers, such as Muon, have recently shown strong performance across large-scale training and community-driven efficiency challenges. However, these methods rely on a costly gradient orthogonalization step. Even…

Artificial Intelligence · Computer Science 2025-12-05 Thibaut Boissin , Thomas Massena , Franck Mamalet , Mathieu Serrurier

Muon has emerged as a strong competitor to AdamW for language model pre-training, yet its behavior at scale is sensitive to weight decay. Recent work has observed that, for Muon without decoupled weight decay, the spectral norm of weight…

Machine Learning · Computer Science 2026-05-12 Kai Lion , Florian Hübler , Bingcong Li , Antonio Orvieto , Niao He

In this paper, we propose a new and efficient nonmonotone adaptive trust region algorithm to solve unconstrained optimization problems. This algorithm incorporates two novelties: it benefits from a radius dependent shrinkage parameter for…

Optimization and Control · Mathematics 2021-05-11 Ahmad Kamandi , Keyvan Amini

Adversarial training (AT) remains one of the most reliable empirical defenses against adversarial attacks. Its robustness critically depends on how the underlying min-max objective is optimized. In practice, Stochastic Gradient Descent…

Machine Learning · Computer Science 2026-05-27 Jun Yan , Weiquan Huang , Jiankai Zuo , Yujian Mo , Xi Fang , Chengliang Wu , Zeming Wei

The Muon optimizer has demonstrated remarkable empirical success in handling matrix-structured parameters for training neural networks. However, a significant gap remains between its practical performance and theoretical understanding.…

Machine Learning · Computer Science 2026-05-12 Da Chang , Yongxiang Liu , Ganzhao Yuan

Physics-informed neural networks and neural operators often suffer from severe optimization difficulties caused by ill-conditioned gradients, multi-scale spectral behavior, and stiffness induced by physical constraints. Recently, the Muon…

Machine Learning · Computer Science 2026-02-19 Binghang Lu , Jiahao Zhang , Guang Lin

The Muon optimizer has rapidly emerged as a powerful, geometry-aware alternative to AdamW, demonstrating strong performance in large-scale training of neural networks. However, a critical theory-practice disconnect exists: Muon's efficiency…

Machine Learning · Computer Science 2025-10-24 Egor Shulgin , Sultan AlRashed , Francesco Orabona , Peter Richtárik

The Muon optimizer has received considerable attention for its strong performance in training large language models, yet the design principle behind its matrix-gradient orthogonalization remains largely elusive. In this paper, we introduce…

Optimization and Control · Mathematics 2026-04-03 Zhehang Du , Weijie Su

The Muon optimizer has recently attracted considerable attention for its strong empirical performance and use of orthogonalized updates on matrix-shaped parameters, yet its underlying mechanisms and relationship to adaptive optimizers such…

Machine Learning · Computer Science 2026-02-05 Xianbiao Qi , Marco Chen , Jiaquan Ye , Yelin He , Rong Xiao

The majority of parameters in neural networks are naturally represented as matrices. However, most commonly used optimizers treat these matrix parameters as flattened vectors during optimization, potentially overlooking their inherent…

Machine Learning · Statistics 2026-04-15 Wei Shen , Ruichuan Huang , Minhui Huang , Cong Shen , Jiawei Zhang

Orthonormalized updates accelerate training, improve stability, and enable robust hyperparameter transfer, but existing methods like Muon rely on dense matrix operations that clash with sharded weights in large-scale LLM training, causing…

Machine Learning · Computer Science 2025-09-16 Kwangjun Ahn , Byron Xu , Natalie Abreu , Ying Fan , Gagik Magakyan , Pratyusha Sharma , Zheng Zhan , John Langford

Modern deep learning commonly relies on AdamW with prescribed learning rate schedules, but recent works challenge both components: Schedule-Free optimization removes explicit schedules via iterate averaging, and Muon improves the update…

Machine Learning · Computer Science 2026-05-22 Jueun Kim , Baekrok Shin , Jihun Yun , Beomhan Baek , Minhak Song , Chulhee Yun

The Muon optimizer has recently attracted attention due to its orthogonalized first-order updates, and a deeper theoretical understanding of its convergence behavior is essential for guiding practical applications; however, existing…

Optimization and Control · Mathematics 2026-03-06 Shuntaro Nagashima , Hideaki Iiduka

Adaptive regularized framework using cubics has emerged as an alternative to line-search and trust-region algorithms for smooth nonconvex optimization, with an optimal complexity amongst second-order methods. In this paper, we propose and…

Optimization and Control · Mathematics 2018-05-30 El houcine Bergou , Youssef Diouane , Serge Gratton

In recent years, Muon has emerged as the dominant method for training large language models, and transformers more broadly. The essential difference, when compared to standard gradient descent methods, is to replace the usual update matrix…

Machine Learning · Computer Science 2026-05-25 Fangzhou Wu , Rikhav Shah , Sandeep Silwal , Qiuyi Zhang

Geometry-aware optimization algorithms, such as Muon, have achieved remarkable success in training deep neural networks (DNNs). These methods leverage the underlying geometry of DNNs by selecting appropriate norms for different layers and…

Machine Learning · Computer Science 2026-02-04 Jie Hao , Xiaochuan Gong , Jie Xu , Zhengdao Wang , Mingrui Liu

Recent advances in spectral optimization, notably Muon, have demonstrated that constraining update steps to the Stiefel manifold can significantly accelerate training and improve generalization. However, Muon implicitly assumes an isotropic…

Machine Learning · Computer Science 2026-04-02 Yechen Zhang , Shuhao Xing , Junhao Huang , Kai Lv , Yunhua Zhou , Xipeng Qiu , Qipeng Guo , Kai Chen

A central challenge in continual learning for large language models (LLMs) is catastrophic forgetting, where adapting to new tasks can substantially degrade performance on previously learned ones. Existing projection-based methods mitigate…

Machine Learning · Computer Science 2026-05-18 Binghang Lu , Zheyuan Deng , Runyu Zhang , Bing Hu , Yunhan Zhao , Yuan Tian , Changhong Mou , Guang Lin , Xiaomin Li

Recent optimizers like Muon, Scion, and Gluon have pushed the frontier of large-scale deep learning by exploiting layer-wise linear minimization oracles (LMOs) over non-Euclidean norm balls, capturing neural network structure in ways…

Machine Learning · Computer Science 2025-10-02 Kaja Gruntkowska , Alexander Gaponov , Zhirayr Tovmasyan , Peter Richtárik