English
Related papers

Related papers: Variance-Adaptive Muon: Accelerating LLM Pretraini…

200 papers

Recent empirical research has demonstrated that deep learning optimizers based on the linear minimization oracle (LMO) over specifically chosen Non-Euclidean norm balls, such as Muon and Scion, outperform Adam-type methods in the training…

Optimization and Control · Mathematics 2025-12-19 Xun Qian , Hussein Rammal , Dmitry Kovalev , Peter Richtárik

The choice of optimizer significantly impacts the training efficiency and computational costs of large language models (LLMs). Recently, the Muon optimizer has demonstrated promising results by orthogonalizing parameter updates, improving…

Machine Learning · Computer Science 2025-10-08 Zichong Li , Liming Liu , Chen Liang , Weizhu Chen , Tuo Zhao

Matrix-based preconditioned optimizers, such as Muon, have recently been shown to be more efficient than scalar-based optimizers for training large-scale neural networks, including large language models (LLMs). Recent benchmark studies of…

Machine Learning · Computer Science 2026-02-02 Yifeng Liu , Angela Yuan , Quanquan Gu

Muon has emerged as a promising optimizer for large-scale foundation model pre-training by exploiting the matrix structure of neural network updates through iterative orthogonalization. However, its practical efficiency is limited by the…

Machine Learning · Computer Science 2026-04-14 Ziyue Liu , Ruijie Zhang , Zhengyang Wang , Yequan Zhao , Yupeng Su , Zi Yang , Zheng Zhang

Efficient stochastic optimization typically integrates an update direction that performs well in the deterministic regime with a mechanism adapting to stochastic perturbations. While Adam uses adaptive moment estimates to promote stability,…

Machine Learning · Computer Science 2026-02-23 Minxin Zhang , Yuxuan Liu , Hayden Schaeffer

Recently, the Muon optimizer based on matrix orthogonalization has demonstrated strong results in training small-scale language models, but the scalability to larger models has not been proven. We identify two crucial techniques for scaling…

Muon is a matrix-aware optimizer that leverages Newton-Schulz (NS) iterations to enforce spectral gradient orthogonalization by driving all singular values of the momentum matrix toward 1. While this uniform spectral whitening enhances…

Machine Learning · Computer Science 2026-05-20 Chongyu Fan , Gaowen Liu , Mingyi Hong , Ramana Rao Kompella , Sijia Liu

We propose AdaMuon, a novel optimizer that combines element-wise adaptivity with orthogonal updates for large-scale neural network training. AdaMuon incorporates two tightly coupled mechanisms: (1) an element-wise second momentum estimator…

Machine Learning · Computer Science 2025-12-25 Chongjie Si , Debing Zhang , Wei Shen

The rapid progress of large language models (LLMs) is increasingly constrained by memory and deployment costs, motivating compression methods for practical deployment. Many state-of-the-art compression pipelines leverage the low-rank…

Muon has recently emerged as a strong optimizer for large language model pre-training, orthogonalizing the momentum matrix via Newton--Schulz polar iterations. A natural intuition is that polar iterations, by flattening the singular…

Machine Learning · Computer Science 2026-05-15 Ruijie Zhang , Yequan Zhao , Ziyue Liu , Zhengyang Wang , Yupeng Su , Liyan Tan , Zheng Zhang

Large models recently are widely applied in artificial intelligence, so efficient training of large models has received widespread attention. More recently, a useful Muon optimizer is specifically designed for matrix-structured parameters…

Machine Learning · Computer Science 2025-09-22 Feihu Huang , Yuning Luo , Songcan 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

Muon has emerged as an efficient alternative to Adam for pretraining, yet remains underused for fine-tuning. A key obstacle is that most open models are pretrained with Adam, and naively switching to Muon for fine-tuning leads to degraded…

Machine Learning · Computer Science 2026-05-12 Xingyu Qu , Peigeng Huang , Samuel Horvath

The Muon optimizer is consistently faster than Adam in training Large Language Models (LLMs), yet the mechanism underlying its success remains unclear. This paper demystifies this mechanism through the lens of associative memory. By…

Machine Learning · Computer Science 2025-10-07 Shuche Wang , Fengzhuo Zhang , Jiaxiang Li , Cunxiao Du , Chao Du , Tianyu Pang , Zhuoran Yang , Mingyi Hong , Vincent Y. F. Tan

Muon optimizer has demonstrated robust results in pretraining of language models but its performance in finetuning of existing public pretrained models is not yet explored. Currently, Muon is used along with AdamW introducing a scope of…

Computation and Language · Computer Science 2025-11-11 Saurabh Page , Advait Joshi , S. S. Sonawane

With the rapid development of natural language processing technology, large-scale language models (LLM) have achieved remarkable results in a variety of tasks. However, how to effectively train these huge models and improve their…

Artificial Intelligence · Computer Science 2024-12-09 Jiajing Chen , Bingying Liu , Xiaoxuan Liao , Jia Gao , Hongye Zheng , Yue Li

The Muon optimizer has emerged as a compelling alternative to Adam for training large language models, achieving remarkable computational savings through gradient orthogonalization. However, Muon's optimizer state is more sensitive to…

Machine Learning · Computer Science 2026-05-13 Yupeng Su , Ruijie Zhang , Ziyue Liu , Yequan Zhao , Zheng Zhang

Large language models (LLMs) have demonstrated impressive generalization and emergent capabilities, yet their pre-training remains computationally expensive and sensitive to optimization dynamics. While Adam-based optimizers offer fast…

Machine Learning · Computer Science 2025-10-01 Junjie Wang , Pan Zhou , Yiming Dong , Huan Li , Jia Li , Xun Zhou , Qicheng Lao , Cong Fang , Zhouchen Lin

Muon has recently emerged as a competitive alternative to AdamW for large-scale pre-training, with orthogonalization via Newton-Schulz (NS) iterations as its core operation. Existing Muon variants apply a uniform NS schedule to all…

Machine Learning · Computer Science 2026-05-19 Xinlin Zhuang , Panyi Ouyang , Yichen Li , Jiangming Shi , Yizhang Chen , Shuman Liu , Ying Qian , Weiyang Liu , Haibo Zhang , Imran Razzak

The impressive performance of large language models (LLMs) arises from their massive scale and heterogeneous module composition. However, this structural heterogeneity introduces additional optimization challenges. While adaptive optimizers…

Machine Learning · Computer Science 2026-05-08 Ziqing Wen , Zhouyang Liu , Jiahuan Wang , Ping Luo , Li Shen , Dongsheng Li , Tao Sun
‹ Prev 1 2 3 10 Next ›