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

Dominant-Layer ZO: A Single Layer Dominates Zeroth-Order Fine-Tuning of LLMs

About

Zeroth-order (ZO) optimization enables memory-efficient fine-tuning of large language models (LLMs) using only forward passes, but it remains unclear how useful adaptation is distributed across layers. In this work, we reveal a surprising phenomenon: ZO fine-tuning is sharply dominated by a single decoding layer. Across multiple LLM families and downstream tasks, fine-tuning this dominant layer alone consistently matches or even exceeds full-model ZO fine-tuning. We further show that the dominant layer is task-agnostic but model-specific, and can be identified before training through a simple inference-only analysis of activation outliers. Specifically, the dominant layer consistently aligns with the first activation-outlier layer in the pre-trained model. To explain this phenomenon, we analyze how perturbation effects propagate under ZO optimization. We find that the dominant layer combines two key properties: high perturbation sensitivity and early placement in the residual stream, allowing perturbation-induced effects to propagate and accumulate through remaining subsequent decoding layers. As a result, this layer produces disproportionately strong and stable optimization signals under forward-only updates. Extensive experiments on LLaMA2-7B and Qwen3-8B across nine benchmarks show that dominant-layer ZO fine-tuning improves average performance over full-model MeZO and LoRA-based ZO fine-tuning while achieving up to 4.52$\times$ training speedup.

Wanhao Yu, Ziyan Wang, Zheng Wang, Abeer Matar Almalky, Yihang Zuo, Shuteng Niu, Sen Lin, Adnan Siraj Rakin, Deliang Fan, Li Yang• 2026

Related benchmarks

TaskDatasetResultRank
Text ClassificationBoolQ
Accuracy77.8
124
Text ClassificationRTE
Accuracy67.51
110
ClassificationSST2
Accuracy91.02
108
ClassificationCB
Accuracy69.64
76
ClassificationWSC
Accuracy64.42
65
GenerationSQuAD
F1 Score89.2
58
GenerationDROP
F1 Score42.1
49
Multiple-ChoiceCOPA
Accuracy87
42
ClassificationMultiRC
Accuracy66.52
35
General language understanding and generationNLP Evaluation Suite SST-2, RTE, CB, BoolQ, WSC, MultiRC, COPA, SQuAD, DROP (test)
SST-2 Accuracy94.15
6
Showing 10 of 10 rows

Other info

Follow for update