arXiv:2601.17261cs.LG2026-01中稿 · ICML被引 1

通过激活结构引导零阶优化,提升大模型微调效率与精度。

AGZO: Activation-Guided Zeroth-Order Optimization for LLM Fine-Tuning

  • 利用前向传播中的激活信息构建低秩子空间,约束扰动方向。
  • 在多个基准上性能超越现有零阶方法,接近一阶微调效果。
  • 适合内存受限场景下的大模型高效微调,尤其适用于资源有限设备。

零阶(ZO)优化在严格内存约束下微调大语言模型方面展现出巨大潜力,因其避免了反向传播中存储激活值的高昂内存开销。然而,现有方法通常采用各向同性扰动,忽略了前向传播中丰富的结构信息。本文发现:线性层的梯度被限制在输入激活张量所张成的子空间内。基于此,提出激活引导的零阶优化(AGZO)。AGZO在前向传播中实时提取紧凑的、依赖激活信息的低秩子空间,并将扰动限制在此子空间内。我们建立了理论框架,证明AGZO优化的是子空间平滑后的目标函数,且更新方向与真实梯度的余弦相似度显著高于各向同性基线。实验在Qwen3和Pangu模型上进行,涵盖多种任务。结果表明,AGZO持续优于最先进的零阶基线,显著缩小与一阶微调的性能差距,同时保持与其它零阶方法相近的峰值内存占用。

原文摘要 · Abstract (English)

Zeroth-Order (ZO) optimization has emerged as a promising solution for fine-tuning LLMs under strict memory constraints, as it avoids the prohibitive memory cost of storing activations for backpropagation. However, existing ZO methods typically employ isotropic perturbations, neglecting the rich structural information available during the forward pass. In this paper, we identify a crucial link between gradient formation and activation structure: the gradient of a linear layer is confined to the subspace spanned by its input activations. Leveraging this insight, we propose Activation-Guided Zeroth-Order optimization (AGZO). Unlike prior methods, AGZO extracts a compact, activation-informed subspace on the fly during the forward pass and restricts perturbations to this low-rank subspace. We provide a theoretical framework showing that AGZO optimizes a subspace-smoothed objective and provably yields update directions with higher cosine similarity to the true gradient than isotropic baselines. Empirically, we evaluate AGZO on Qwen3 and Pangu models across various benchmarks. AGZO consistently outperforms state-of-the-art ZO baselines and significantly narrows the performance gap with first-order fine-tuning, while maintaining almost the same peak memory footprint as other ZO methods.

零阶优化大模型微调内存效率激活引导

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。