English
Related papers

Related papers: Universally Empowering Zeroth-Order Optimization v…

200 papers

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

Zeroth-order (ZO) optimization is an emerging deep neural network (DNN) training paradigm that offers computational simplicity and memory savings. However, this seemingly promising approach faces a significant and long-ignored challenge. ZO…

Machine Learning · Computer Science 2025-07-28 Qitao Tan , Sung-En Chang , Rui Xia , Huidong Ji , Chence Yang , Ci Zhang , Jun Liu , Zheng Zhan , Zhenman Fang , Zhou Zou , Yanzhi Wang , Jin Lu , Geng Yuan

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

Zeroth-order optimizers have recently emerged as a practical approach for fine-tuning large language models (LLMs), significantly reducing GPU memory consumption compared to traditional first-order methods. Yet, existing zeroth-order…

Machine Learning · Computer Science 2025-10-02 Kairun Zhang , Haoyu Li , Yanjun Zhao , Yifan Sun , Huan Zhang

Fine-tuning large language models (LLMs) using standard first-order (FO) optimization often drives training toward sharp, poorly generalizing minima. Conversely, zeroth-order (ZO) methods offer stronger exploratory behavior without relying…

Machine Learning · Computer Science 2026-01-12 Feihu Jin , Ying Tan

Parameter-efficient fine-tuning (PEFT) significantly reduces memory costs when adapting large language models (LLMs) for downstream applications. However, traditional first-order (FO) fine-tuning algorithms incur substantial memory overhead…

Machine Learning · Computer Science 2024-10-11 Yiming Chen , Yuan Zhang , Liyuan Cao , Kun Yuan , Zaiwen Wen

Language Models (LLMs) are often quantized to lower precision to reduce the memory cost and latency in inference. However, quantization often degrades model performance, thus fine-tuning is required for various down-stream tasks.…

Machine Learning · Computer Science 2025-02-19 Jiajun Zhou , Yifan Yang , Kai Zhen , Ziyue Liu , Yequan Zhao , Ershad Banijamali , Athanasios Mouchtaris , Ngai Wong , Zheng Zhang

Fine-tuning large pre-trained LLMs generally demands extensive GPU memory. Traditional first-order optimizers like SGD encounter substantial difficulties due to increased memory requirements from storing activations and gradients during…

Machine Learning · Computer Science 2025-03-18 Liangyu Wang , Jie Ren , Hang Xu , Junxiao Wang , Huanyi Xie , David E. Keyes , Di Wang

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

In the evolving landscape of natural language processing (NLP), fine-tuning pre-trained Large Language Models (LLMs) with first-order (FO) optimizers like SGD and Adam has become standard. Yet, as LLMs grow {in size}, the substantial memory…

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 is popular in real-world applications that accessing the gradient information is expensive or unavailable. Recently, adaptive ZO methods that normalize gradient estimators by the empirical standard deviation…

Optimization and Control · Mathematics 2026-02-03 Haishan Ye , Luo Luo

On-device fine-tuning is a critical capability for edge AI systems, which must support adaptation to different agentic tasks under stringent memory constraints. Conventional backpropagation (BP)-based training requires storing layer…

Machine Learning · Computer Science 2026-05-06 Prabodh Katti , Houssem Sifaou , Sangwoo Park , Bipin Rajendran , Osvaldo Simeone

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

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…

The confluence of Federated Learning (FL) and Large Language Models (LLMs) is ushering in a new era in privacy-preserving natural language processing. However, the intensive memory requirements for fine-tuning LLMs pose significant…

Machine Learning · Computer Science 2024-06-18 Zhenqing Ling , Daoyuan Chen , Liuyi Yao , Yaliang Li , Ying Shen

Zeroth-order (ZO) optimization has become increasingly popular and important in fine-tuning large language models (LLMs), especially on edge devices due to its ability to adjust the model to local data without the need for memory-intensive…

Machine Learning · Computer Science 2026-05-18 Jiahe Chen , Ziye Ma

The adaptive momentum method (AdaMM), which uses past gradients to update descent directions and learning rates simultaneously, has become one of the most popular first-order optimization methods for solving machine learning problems.…

Machine Learning · Computer Science 2019-10-17 Xiangyi Chen , Sijia Liu , Kaidi Xu , Xingguo Li , Xue Lin , Mingyi Hong , David Cox

Federated learning enables collaborative model training across numerous edge devices without requiring participants to share data; however, memory and communication constraints on these edge devices may preclude their participation in…

Machine Learning · Computer Science 2025-09-04 Gwen Legate , Irina Rish , Eugene Belilovsky

Zeroth-order (ZO) fine-tuning is attractive for large language models because it replaces backpropagation with forward objective evaluations. Existing implementations nevertheless execute ZO algorithms inside conventional training loops,…

Machine Learning · Computer Science 2026-05-28 Zelin Li , Caiwen Ding