Harmony in Divergence: Towards Fast, Accurate, and Memory-efficient Zeroth-order LLM Fine-tuning
Not on the map. Why?
Abstract (arXiv)
Large language models (LLMs) excel across various tasks, but standard first-order (FO) fine-tuning demands considerable memory, significantly limiting real-world deployment. Recently, zeroth-order (ZO) optimization stood out as a promising memory-efficient training paradigm, avoiding backward passes and relying solely on forward passes for gradient estimation, making it attractive for resource-constrained scenarios. However, ZO method lags far behind FO method in both convergence speed and accuracy. To bridge the gap, we introduce a novel layer-wise divergence analysis that uncovers the distinct update pattern of FO and ZO optimization. Aiming to resemble the learning capacity of FO method from the findings, we propose Divergence-driven Zeroth-Order (DiZO) optimization. DiZO conducts divergence-driven layer adaptation by incorporating projections to ZO updates, generating diverse-magnitude updates precisely scaled to layer-wise individual optimization needs. Our results demonstrate that DiZO significantly reduces the needed iterations for convergence without sacrificing throughput, cutting training GPU hours by up to 48% on various datasets. Moreover, DiZO consistently outperforms the representative ZO baselines in fine-tuning RoBERTa-large, OPT-series, and Llama-series on downstream tasks and, in some cases, even surpasses memory-intensive FO fine-tuning. Our code is released at https://github.com/Skilteee/DiZO.
Key facts. Cited by 17 papers in the map (data through 7 Oct 2026). Builds on 51 of its 71 references that are in the map. First posted on arXiv 5 Feb 2025.
Builds on
51 of 71 references.
References · 51
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,393
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,878
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
38 more
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,140
- 2015A large annotated corpus for learning natural language inferencecited by 1,837
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,793
- 2019SuperGLUE: A Stickier Benchmark for General-Purpose Language Understanding Systemscited by 1,718
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,548
- 2022SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Modelscited by 1,338
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
- 2020Making Pre-trained Language Models Better Few-shot Learnerscited by 1,255
- 2019DROP: A Reading Comprehension Benchmark Requiring Discrete Reasoning Over Paragraphscited by 883
- 2024DoRA: Weight-Decomposed Low-Rank Adaptationcited by 740
- 2017ZOO: Zeroth Order Optimization based Black-box Attacks to Deep Neural Networks without Training Substitute Modelscited by 656
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
- 2023OmniQuant: Omnidirectionally Calibrated Quantization for Large Language Modelscited by 403
- 2018WiC: the Word-in-Context Dataset for Evaluating Context-Sensitive Meaning Representationscited by 370
- 2023Fine-Tuning Language Models with Just Forward Passescited by 337
- 2019PyHessian: Neural Networks Through the Lens of the Hessiancited by 291
- 2018Explanations based on the Missing: Towards Contrastive Explanations with Pertinent Negativescited by 261
- 2021VisualGPT: Data-efficient Adaptation of Pretrained Language Models for Image Captioningcited by 190
- 2019End-to-End Open-Domain Question Answering with BERTserinicited by 152
- 2022Revisiting Weakly Supervised Pre-Training of Visual Perception Modelscited by 116
- 2024Revisiting Zeroth-Order Optimization for Memory-Efficient LLM Fine-Tuning: A Benchmarkcited by 113
- 2018Zeroth-Order Stochastic Variance Reduction for Nonconvex Optimizationcited by 109
- 2020End-to-End Object Detection with Adaptive Clustering Transformercited by 109
- 2024AI and Memory Wallcited by 89
- 2024LISA: Layerwise Importance Sampling for Memory-Efficient Large Language Model Fine-Tuningcited by 88
- 2023DeepZero: Scaling up Zeroth-Order Optimization for Deep Model Trainingcited by 76
- 2024FlightLLM: Efficient Large Language Model Inference with a Complete Mapping Flow on FPGAscited by 60
- 2024Sparse MeZO: Less Parameters for Better Performance in Zeroth-Order LLM Fine-Tuningcited by 53
- 2024Second-Order Fine-Tuning without Pain for LLMs:A Hessian Informed Zeroth-Order Optimizercited by 51
- 2024Variance-reduced Zeroth-Order Methods for Fine-Tuning Language Modelscited by 50
- 2019Model Agnostic Contrastive Explanations for Structured Datacited by 41
- 2023Understanding the Potential of FPGA-Based Spatial Acceleration for Large Language Model Inferencecited by 29
- 2024BAdam: A Memory Efficient Full Parameter Optimization Method for Large Language Modelscited by 27
- 2019Contrasting Exploration in Parameter and Action Space: A Zeroth-Order Optimization Perspectivecited by 26
- 2020Efficient On-Chip Learning for Optical Neural Networks Through Power-Aware Sparse Zeroth-Order Optimizationcited by 24
- 2021ZARTS: On Zero-order Optimization for Neural Architecture Searchcited by 17
- 2020Learning to Guide Random Searchcited by 13
- 2024Asteroid: Resource-Efficient Hybrid Pipeline Parallelism for Collaborative DNN Training on Heterogeneous Edge Devicescited by 13
50 most cited of 51 All 51 (JSON)
Built on by
Citing papers per year: 2025: 8; 2026: 9.
17 papers.
Sort:
By area:
- 2025RCR-Router: Efficient Role-Aware Context Routing for Multi-Agent LLM Systems with Structured Memorycited by 24
- 2025Structured Agent Distillation for Large Language Modelcited by 11
- 2025FZOO: Fast Zeroth-Order Optimizer for Fine-Tuning Large Language Models towards Adam-Scale Speedcited by 11
- 2026Powering Up Zeroth-Order Training via Subspace Gradient Orthogonalizationcited by 11
- 2025Perturbation-efficient Zeroth-order Optimization for Hardware-friendly On-device Trainingcited by 9
- 2025KerZOO: Kernel Function Informed Zeroth-Order Optimization for Accurate and Accelerated LLM Fine-Tuningcited by 4
- 2025Downgrade to Upgrade: Optimizer Simplification Enhances Robustness in LLM Unlearningcited by 4
- 2026CurvZO: Adaptive Curvature-Guided Sparse Zeroth-Order Optimization for Efficient LLM Fine-Tuningcited by 3
- 2025Towards Fast LLM Fine-tuning through Zeroth-Order Optimization with Projected Gradient-Aligned Perturbationscited by 2
- 2026When Should a Robot Think? Resource-Aware Reasoning via Reinforcement Learning for Embodied Robotic Decision-Makingcited by 2
- 2026AdaMeZO: Adam-style Zeroth-Order Optimizer for LLM Fine-tuning Without Maintaining the Momentscited by 2
- 2025TSLA: A Task-Specific Learning Adaptation for Semantic Segmentation on Autonomous Vehicles Platformcited by 1
5 more
- 2026Universally Empowering Zeroth-Order Optimization via Adaptive Layer-wise Samplingcited by 1
- 2026Palette: A Modular, Controllable, and Efficient Framework for On-demand Authorized Safety Alignment Relaxation in LLMscited by 1
- 2026Dominant-Layer ZO: A Single Layer Dominates Zeroth-Order Fine-Tuning of LLMscited by 1
- 2026CooperLLM: Cloud-Edge-End Cooperative Federated Fine-tuning for LLMs via ZOO-based Gradient Correctioncited by 0
- 2026Why Zeroth-Order Adaptation May Forget Less: A Randomized Shaping Theorycited by 0
- 2026Dominant-Layer ZO: A Single Layer Dominates Zeroth-Order Fine-Tuning of LLMs3 Jun 2026
- 2026Palette: A Modular, Controllable, and Efficient Framework for On-demand Authorized Safety Alignment Relaxation in LLMs22 May 2026
- 2026Why Zeroth-Order Adaptation May Forget Less: A Randomized Shaping Theory11 May 2026
- 2026AdaMeZO: Adam-style Zeroth-Order Optimizer for LLM Fine-tuning Without Maintaining the Moments1 May 2026
- 2026Universally Empowering Zeroth-Order Optimization via Adaptive Layer-wise Sampling20 Apr 2026
- 2026CurvZO: Adaptive Curvature-Guided Sparse Zeroth-Order Optimization for Efficient LLM Fine-Tuning23 Mar 2026
- 2026When Should a Robot Think? Resource-Aware Reasoning via Reinforcement Learning for Embodied Robotic Decision-Making17 Mar 2026
- 2026Powering Up Zeroth-Order Training via Subspace Gradient Orthogonalization19 Feb 2026
- 2026CooperLLM: Cloud-Edge-End Cooperative Federated Fine-tuning for LLMs via ZOO-based Gradient Correction19 Jan 2026
- 2025Towards Fast LLM Fine-tuning through Zeroth-Order Optimization with Projected Gradient-Aligned Perturbations21 Oct 2025
- 2025Downgrade to Upgrade: Optimizer Simplification Enhances Robustness in LLM Unlearning1 Oct 2025
- 2025TSLA: A Task-Specific Learning Adaptation for Semantic Segmentation on Autonomous Vehicles Platform17 Aug 2025
5 more
- 2025RCR-Router: Efficient Role-Aware Context Routing for Multi-Agent LLM Systems with Structured Memory6 Aug 2025
- 2025FZOO: Fast Zeroth-Order Optimizer for Fine-Tuning Large Language Models towards Adam-Scale Speed10 Jun 2025
- 2025KerZOO: Kernel Function Informed Zeroth-Order Optimization for Accurate and Accelerated LLM Fine-Tuning24 May 2025
- 2025Structured Agent Distillation for Large Language Model20 May 2025
- 2025Perturbation-efficient Zeroth-order Optimization for Hardware-friendly On-device Training28 Apr 2025
- 2025FZOO: Fast Zeroth-Order Optimizer for Fine-Tuning Large Language Models towards Adam-Scale Speedcited by 11
- 2026Powering Up Zeroth-Order Training via Subspace Gradient Orthogonalizationcited by 11
- 2025Perturbation-efficient Zeroth-order Optimization for Hardware-friendly On-device Trainingcited by 9
- 2025KerZOO: Kernel Function Informed Zeroth-Order Optimization for Accurate and Accelerated LLM Fine-Tuningcited by 4
- 2026CurvZO: Adaptive Curvature-Guided Sparse Zeroth-Order Optimization for Efficient LLM Fine-Tuningcited by 3
- 2025Towards Fast LLM Fine-tuning through Zeroth-Order Optimization with Projected Gradient-Aligned Perturbationscited by 2
- 2026AdaMeZO: Adam-style Zeroth-Order Optimizer for LLM Fine-tuning Without Maintaining the Momentscited by 2
- 2025TSLA: A Task-Specific Learning Adaptation for Semantic Segmentation on Autonomous Vehicles Platformcited by 1
- 2026Universally Empowering Zeroth-Order Optimization via Adaptive Layer-wise Samplingcited by 1
- 2026Dominant-Layer ZO: A Single Layer Dominates Zeroth-Order Fine-Tuning of LLMscited by 1
- 2026CooperLLM: Cloud-Edge-End Cooperative Federated Fine-tuning for LLMs via ZOO-based Gradient Correctioncited by 0
- 2026Why Zeroth-Order Adaptation May Forget Less: A Randomized Shaping Theorycited by 0
- 2026Dominant-Layer ZO: A Single Layer Dominates Zeroth-Order Fine-Tuning of LLMs3 Jun 2026
- 2026Why Zeroth-Order Adaptation May Forget Less: A Randomized Shaping Theory11 May 2026
- 2026AdaMeZO: Adam-style Zeroth-Order Optimizer for LLM Fine-tuning Without Maintaining the Moments1 May 2026
- 2026Universally Empowering Zeroth-Order Optimization via Adaptive Layer-wise Sampling20 Apr 2026
- 2026CurvZO: Adaptive Curvature-Guided Sparse Zeroth-Order Optimization for Efficient LLM Fine-Tuning23 Mar 2026
- 2026Powering Up Zeroth-Order Training via Subspace Gradient Orthogonalization19 Feb 2026
- 2026CooperLLM: Cloud-Edge-End Cooperative Federated Fine-tuning for LLMs via ZOO-based Gradient Correction19 Jan 2026
- 2025Towards Fast LLM Fine-tuning through Zeroth-Order Optimization with Projected Gradient-Aligned Perturbations21 Oct 2025
- 2025TSLA: A Task-Specific Learning Adaptation for Semantic Segmentation on Autonomous Vehicles Platform17 Aug 2025
- 2025FZOO: Fast Zeroth-Order Optimizer for Fine-Tuning Large Language Models towards Adam-Scale Speed10 Jun 2025
- 2025KerZOO: Kernel Function Informed Zeroth-Order Optimization for Accurate and Accelerated LLM Fine-Tuning24 May 2025
- 2025Perturbation-efficient Zeroth-order Optimization for Hardware-friendly On-device Training28 Apr 2025