English
Related papers

Related papers: AGZO: Activation-Guided Zeroth-Order Optimization …

200 papers

Fine-tuning large language models (LLMs) often faces GPU memory bottlenecks: the backward pass of first-order optimizers like Adam increases memory usage to more than 10 times the inference level (e.g., 633 GB for OPT-30B). Zeroth-order…

Machine Learning · Computer Science 2025-07-01 Sizhe Dang , Yangyang Guo , Yanjun Zhao , Haishan Ye , Xiaodong Zheng , Guang Dai , Ivor Tsang

Fine-tuning language models (LMs) has yielded success on diverse downstream tasks, but as LMs grow in size, backpropagation requires a prohibitively large amount of memory. Zeroth-order (ZO) methods can in principle estimate gradients using…

Machine Learning · Computer Science 2024-01-12 Sadhika Malladi , Tianyu Gao , Eshaan Nichani , Alex Damian , Jason D. Lee , Danqi Chen , Sanjeev Arora

Zeroth-order (ZO) optimization has gained attention as a memory-efficient alternative to first-order (FO) methods, particularly in settings where gradient computation is expensive or even impractical. Beyond its memory efficiency, in this…

Machine Learning · Computer Science 2026-03-13 Wanhao Yu , Zheng Wang , Shuteng Niu , Sen Lin , Li Yang

While fine-tuning large language models (LLMs) for specific tasks often yields impressive results, it comes at the cost of memory inefficiency due to back-propagation in gradient-based training. Memory-efficient Zeroth-order (MeZO)…

Machine Learning · Computer Science 2026-02-17 Yong Liu , Zirui Zhu , Chaoyu Gong , Minhao Cheng , Cho-Jui Hsieh , Yang You

Fine-tuning large language models (LLMs) using zeroth-order (ZO) optimization has emerged as a promising alternative to traditional gradient-based methods due to its reduced memory footprint requirement. However, existing ZO methods suffer…

Machine Learning · Computer Science 2025-10-22 Zhendong Mi , Qitao Tan , Grace Li Zhang , Zhaozhuo Xu , Geng Yuan , Shaoyi Huang

As the size of large language models grows exponentially, GPU memory has become a bottleneck for adapting these models to downstream tasks. In this paper, we aim to push the limits of memory-efficient training by minimizing memory usage on…

Machine Learning · Computer Science 2026-02-13 Sifeng Shang , Jiayi Zhou , Chenyu Lin , Minxian Li , Kaiyang Zhou

Fine-tuning language models (LMs) has demonstrated success in a wide array of downstream tasks. However, as LMs are scaled up, the memory requirements for backpropagation become prohibitively high. Zeroth-order (ZO) optimization methods can…

Machine Learning · Computer Science 2024-04-15 Tanmay Gautam , Youngsuk Park , Hao Zhou , Parameswaran Raman , Wooseok Ha

Zeroth-order optimization (ZO) has demonstrated remarkable promise in efficient fine-tuning tasks for Large Language Models (LLMs). In particular, recent advances incorporate the low-rankness of gradients, introducing low-rank ZO estimators…

Machine Learning · Computer Science 2025-02-03 Yan Sun , Tiansheng Huang , Liang Ding , Li Shen , Dacheng Tao

Fine-tuning Large Language Models (LLMs) is essential for adapting pre-trained models to downstream tasks. Yet traditional first-order optimizers such as Stochastic Gradient Descent (SGD) and Adam incur prohibitive memory and computational…

Zeroth-order (ZO) optimization has long been favored for its biological plausibility and its capacity to handle non-differentiable objectives, yet its computational complexity has historically limited its application in deep neural…

Machine Learning · Computer Science 2026-02-12 Sansheng Cao , Zhengyu Ma , Yonghong Tian

Fine-tuning large language models (LLMs) using zeroth-order optimization (ZO) offers a memory-efficient alternative to gradient-based methods but suffers from slower convergence and unstable optimization due to noisy gradient estimates.…

Machine Learning · Computer Science 2025-06-24 Jikai Long , Zijian Hu , Xiaodong Yu , Jianwen Xie , Zhaozhuo Xu

Fine-tuning large language models (LLMs) with zeroth-order (ZO) optimization reduces memory by approximating gradients through function evaluations. However, existing methods essentially perform updates in a one-dimensional space, and…

Machine Learning · Computer Science 2026-01-19 Jian Feng , Zhihong Huang

Fine-tuning vision language models (VLMs) has achieved remarkable performance across various downstream tasks; yet, it requires access to model gradients through backpropagation (BP), making them unsuitable for memory-constrained,…

Machine Learning · Computer Science 2025-10-27 Yifan Yang , Zhen Zhang , Rupak Vignesh Swaminathan , Jing Liu , Nathan Susanj , Zheng Zhang

Recently, zeroth-order (ZO) optimization plays an essential role in scenarios where gradient information is inaccessible or unaffordable, such as black-box systems and resource-constrained environments. While existing adaptive methods such…

Machine Learning · Computer Science 2025-06-10 Yao Shu , Qixin Zhang , Kun He , Zhongxiang Dai

Zeroth-Order optimization presents a promising memory-efficient paradigm for fine-tuning Large Language Models by relying solely on forward passes. However, its practical adoption is severely constrained by slow wall-clock convergence and…

Machine Learning · Computer Science 2026-04-21 Fei Wang , Li Shen , Liang Ding , Chao Xue , Ye Liu , Changxing Ding

Zeroth-order (ZO) optimization has become a popular technique for solving machine learning (ML) problems when first-order (FO) information is difficult or impossible to obtain. However, the scalability of ZO optimization remains an open…

Molecule optimization is an important problem in chemical discovery and has been approached using many techniques, including generative modeling, reinforcement learning, genetic algorithms, and much more. Recent work has also applied…

Biomolecules · Quantitative Biology 2022-10-31 Elvin Lo , Pin-Yu Chen

Zeroth-order (ZO) optimization is being recognized as a simple yet powerful alternative to standard backpropagation (BP)-based training. Notably, ZO optimization allows for training with only forward passes and (almost) the same memory as…

Machine Learning · Computer Science 2025-01-09 Keisuke Sugiura , Hiroki Matsutani

Lowering the memory requirement in full-parameter training on large models has become a hot research area. MeZO fine-tunes the large language models (LLMs) by just forward passes in a zeroth-order SGD optimizer (ZO-SGD), demonstrating…

Machine Learning · Computer Science 2023-12-27 Shuoran Jiang , Qingcai Chen , Youchen Pan , Yang Xiang , Yukang Lin , Xiangping Wu , Chuanyi Liu , Xiaobao Song

Zeroth-order or derivative-free optimization (MeZO) is an attractive strategy for finetuning large language models (LLMs) because it eliminates the memory overhead of backpropagation. However, it converges slowly due to the inherent curse…

Machine Learning · Computer Science 2026-04-21 Lejs Deen Behric , Liang Zhang , Bingcong Li , Kiran Koshy Thekumparampil