通过激活信息构建低秩子空间,实现高效零阶微调。
ZO-Act: Efficient Zeroth-Order Fine-Tuning via One-Shot Activation-Informed Low-Rank Subspaces

- 基于输入激活构造固定低秩子空间,仅优化系数矩阵
- 在Llama-3-8B等模型上超越强基线,提升语言理解与推理能力
- 适合资源受限场景,支持量化模型微调,兼容Adam优化器
零阶(ZO)优化可在无反向传播或内存受限时微调大语言模型,但现有方法常对全量模型权重或随机构造的低维子空间进行扰动,导致估计方差高、性能有限。本文提出ZO-Act,一种基于激活信息的零阶微调方法,将扰动限制在由输入激活导出的固定低秩子空间内。对每一层线性层,ZO-Act在初始化时一次性计算小规模激活基,并仅通过前向损失评估优化轻量级系数矩阵。该方法降低有效扰动维度,暴露显式可训练变量,兼容Adam等带动量优化器,并天然支持量化大模型微调(低比特权重冻结)。我们分析了ZO-Act作为受限系数空间上的零阶优化,证明扰动低维系数可同时减少方差相关的收敛项与有限差分误差,仅引入可控的子空间近似偏差;该偏差因大模型激活与梯度的低秩特性得以缓解。在Llama-3-8B、OPT-13B及INT4 Llama-3-8B上的实验表明,该方法在语言理解、问答和常识推理任务中持续优于强基线。
原文摘要 · Abstract (English)
Zeroth-order (ZO) optimization enables fine-tuning large language models when backpropagation is unavailable or memory-prohibitive, but existing methods often perturb full model weights or randomly constructed low-dimensional subspaces, yielding high-variance estimates and limited performance. We propose ZO-Act, an activation-informed ZO fine-tuning method that restricts perturbations to a fixed low-rank subspace derived from input activations. For each linear layer, ZO-Act computes a small activation basis once at initialization and optimizes only lightweight coefficient matrices using forward-only loss evaluations. This reduces the effective perturbation dimension, exposes explicit trainable variables compatible with momentum-based optimizers such as Adam, and naturally supports quantized LLM fine-tuning by keeping low-bit weights frozen. We analyze ZO-Act as zeroth-order optimization over a restricted coefficient space and show that perturbing the low-dimensional coefficients reduces both the variance-dependent convergence term and the finite-difference error of the ZO estimator, at the cost of a controlled subspace approximation bias that is mitigated by the low-rank structure of LLM activations and gradients. Experiments on Llama-3-8B, OPT-13B, and INT4 Llama-3-8B show consistent gains over strong ZO fine-tuning baselines across language understanding, question answering, and commonsense reasoning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。