Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

GRZO: Group-Relative Zeroth-Order Optimization for Large Language Model Fine-Tuning

About

Zeroth-order (ZO) optimization is a memory-efficient alternative to backpropagation for fine-tuning large language models, but its deployment is limited by the high variance of gradient estimation. We propose GRZO, a Group-Relative Zeroth-Order optimizer that draws one pseudo-independent perturbation per mini-batch example and aggregates the per-example losses through group-relative normalization, raising the effective gradient-direction count from one to the batch size at no additional forward cost while preserving inference-level memory. We prove that GRZO is directionally unbiased with variance shrinking proportionally to the batch size, yielding a tighter nonconvex convergence bound than MeZO. Across RoBERTa-large, Llama3-8B, and OPT-13B over multiple tasks, GRZO improves average accuracy on Llama3-8B by $+3.0$ over MeZO at $23\%$ lower peak GPU memory; as a drop-in replacement for the MeZO core, it lifts sparse, low-rank, and quantized ZO variants by $+6.0$ on average.

Liyan Tan, Yequan Zhao, Yifan Yang, Ruijie Zhang, Xinling Yu, Zheng Zhang• 2026

Related benchmarks

TaskDatasetResultRank
Sentiment ClassificationSST-2
Accuracy93.3
220
Natural Language InferenceSNLI
Accuracy85.4
218
Natural Language InferenceMNLI--
80
Sentiment ClassificationSST-5
Accuracy54.8
73
Question AnsweringDROP
Accuracy65
16
Natural Language InferenceRTE
Accuracy (RTE)79
9
Topic ClassificationTREC
Accuracy94.7
6
Showing 7 of 7 rows

Other info

Follow for update