只调一个解码层,就能让大模型零阶微调效果更好更快。
Dominant-Layer ZO: A Single Layer Dominates Zeroth-Order Fine-Tuning of LLMs

- 只优化一个关键解码层,即可达到全模型微调效果。
- 在9个基准上平均性能超越全模型方法,速度提升最高达4.52倍。
- 该关键层可通过预训练时的激活异常分析提前识别,无需训练。
零阶(ZO)优化通过仅使用前向传播实现大语言模型(LLMs)的高效微调,但其适应性在各层间的分布尚不明确。本文发现:ZO微调高度依赖单一解码层。在多个模型家族和下游任务中,仅微调这一主导层的表现始终与全模型ZO微调相当甚至更优。该主导层具有任务无关性但模型特异性,可通过预训练模型中仅基于推理的激活异常分析提前确定,且始终对应首个激活异常层。我们进一步分析扰动传播机制,发现该层兼具高扰动敏感性和残差流中的早期位置,使扰动效应能有效传递并累积至后续解码层,从而生成强而稳定的优化信号。在LLaMA2-7B与Qwen3-8B上的实验证明,主导层ZO微调在9个基准上平均性能优于全模型MeZO和基于LoRA的ZO微调,同时实现最高4.52倍的训练加速。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。