English
Related papers

Related papers: Accelerating Zeroth-Order Spectral Optimization wi…

200 papers

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…

Zeroth-order (ZO) optimization is indispensable for complex non-convex tasks where explicit gradients are computationally prohibitive or strictly inaccessible. For deploying ZO methods over distributed heterogeneous networks, the gradient…

Optimization and Control · Mathematics 2026-04-24 Yanxu Su , Xiaorui Tong , Changyin Sun

Federated fine-tuning offers a promising approach for tuning Large Language Models (LLMs) on edge devices while preserving data privacy. However, fine-tuning these models on edge devices remains challenging due to high memory,…

Machine Learning · Computer Science 2025-12-19 Mohamed Aboelenien Ahmed , Kilian Pfeiffer , Ramin Khalili , Heba Khdr , Jörg Henkel

Large Language Models (LLMs) achieve competitive performance across diverse natural language processing (NLP) tasks, yet pretraining is computationally demanding, making optimizer efficiency an important practical consideration. Muon…

Machine Learning · Computer Science 2026-01-22 Jingru Li , Yibo Fan , Huan Li

Neural network (NN) training is inherently a large-scale matrix optimization problem, yet the matrix structure of NN parameters has long been overlooked. Recently, the optimizer Muon \citep{jordanmuon}, which explicitly exploits this…

Machine Learning · Computer Science 2026-04-21 Chuan He , Zhanwang Deng , Zhaosong Lu

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 recently offered a promising alternative to AdamW for large language model training, leveraging matrix orthogonalization to produce geometry-aware updates. However, like all first-order methods, Muon can become…

Machine Learning · Computer Science 2026-05-27 Jiacheng Li , Jianchao Tan , Hongtao Xu , Jiaqi Zhang , Yifan Lu , Yerui Sun , Yuchen Xie , Xunliang Cai

Safe derivative-free optimization under unknown constraints is a fundamental challenge in modern learning and control. Existing zeroth-order (ZO) methods typically still assume access to a first-order oracle of the constraint functions or…

Optimization and Control · Mathematics 2026-01-29 Runyu Zhang , Gioele Zardini , Asuman Ozdaglar , Jeff Shamma , Na Li

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

Low-rank gradient-based optimization methods have significantly improved memory efficiency during the training of large language models (LLMs), enabling operations within constrained hardware without sacrificing performance. However, these…

Machine Learning · Computer Science 2025-10-28 Yehonathan Refael , Guy Smorodinsky , Tom Tirer , Ofir Lindenbaum

Scaling large models requires optimization strategies that ensure rapid convergence grounded in stability. Maximal Update Parametrization ($\boldsymbol{\mu}$P) provides a theoretical safeguard for width-invariant $\Theta(1)$ activation…

Machine Learning · Computer Science 2026-03-06 Tian Xie , Haoming Luo , Haoyu Tang , Yiwen Hu , Jason Klein Liu , Qingnan Ren , Yang Wang , Wayne Xin Zhao , Rui Yan , Bing Su , Chong Luo , Baining Guo

To solve unmodeled optimization problems with hard constraints, this paper proposes a novel zeroth-order approach called Safe Zeroth-order Optimization using Linear Programs (SZO-LP). The SZO-LP method solves a linear program in each…

Optimization and Control · Mathematics 2023-04-05 Baiwei Guo , Yang Wang , Yuning Jiang , Maryam Kamgarpour , Giancarlo Ferrari-Trecate

Interest in stochastic zeroth-order (SZO) methods has recently been revived in black-box optimization scenarios such as adversarial black-box attacks to deep neural networks. SZO methods only require the ability to evaluate the objective…

Machine Learning · Statistics 2020-11-11 Mayumi Ohta , Nathaniel Berger , Artem Sokolov , Stefan Riezler

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

Zeroth-order (ZO) optimization enables large-language-model fine-tuning without storing backpropagation activations, while LoRA supplies compact trainable adapters. Combining them creates a rank paradox: increasing LoRA rank improves…

Machine Learning · Computer Science 2026-05-20 Ziye Chen , Hongbin Lin , Chenyu Zhang , Xiangda Yan , Yongjie Yang , Yao Shu

The optimization of large language models (LLMs) remains a critical challenge, particularly as model scaling exacerbates sensitivity to algorithmic imprecision and training instability. Recent advances in optimizers have improved…

Machine Learning · Computer Science 2025-11-26 Wei He , Kai Han , Hang Zhou , Hanting Chen , Zhicheng Liu , Xinghao Chen , Yunhe Wang

Spectral optimizers such as Muon have recently shown strong empirical performance in large-scale language model training, but the source and extent of their advantage remain poorly understood. We study this question through the linear…

Machine Learning · Computer Science 2026-04-29 Juno Kim , Eshaan Nichani , Denny Wu , Alberto Bietti , Jason D. Lee

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

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

Zeroth-Order Optimization (ZOO) provides powerful tools for optimizing functions where explicit gradients are unavailable or expensive to compute. However, the underlying mechanisms of popular ZOO methods, particularly those employing…

Machine Learning · Computer Science 2025-06-18 Junbin Qiu , Zhengpeng Xie , Xiangda Yan , Yongjie Yang , Yao Shu