通过子空间正交化提升零阶优化,大幅减少查询次数同时提高精度。
Powering Up Zeroth-Order Training via Subspace Gradient Orthogonalization
- 利用模型更新的低秩结构降低梯度估计方差,结合谱正交化提取有效信息。
- 在大模型微调中仅需24.7%查询量即可达MeZO同等准确率,ViT任务精度提升25.1%。
- 适合追求高效微调的大模型应用,尤其适用于无梯度计算场景。
零阶(ZO)优化通过函数值的有限差分估算梯度,无需反向传播,成为大规模模型微调的内存高效范式。然而,其面临准确率与查询效率之间的根本矛盾。本文提出统一两种互补原则:(i) 基于投影的子空间视角,利用模型更新的内在低秩结构降低梯度估计方差;(ii) 类似Muon的谱优化,通过梯度正交化从噪声中提取有意义的谱结构。由此构建统一的子空间梯度正交化框架,并实现新方法ZO-Muon,可自然理解为零阶设置下的低秩Muon优化器。大量实验表明,ZO-Muon显著加速收敛,在大型语言模型(LLMs)和视觉变换器(ViTs)上实现准确率与查询/运行效率的双赢。相比主流的MeZO基线,其在LLM微调中仅需24.7%的查询量即可达到相同的SST-2性能,且在CIFAR-100上的ViT-B微调任务中准确率提升25.1%。
原文摘要 · Abstract (English)
Zeroth-order (ZO) optimization provides a gradient-free alternative to first-order (FO) methods by estimating gradients via finite differences of function evaluations, and has recently emerged as a memory-efficient paradigm for fine-tuning large-scale models by avoiding backpropagation. However, ZO optimization has a fundamental tension between accuracy and query efficiency. In this work, we show that ZO optimization can be substantially improved by unifying two complementary principles: (i) a projection-based subspace view that reduces gradient estimation variance by exploiting the intrinsic low-rank structure of model updates, and (ii) Muon-style spectral optimization that applies gradient orthogonalization to extract informative spectral structure from noisy ZO gradients. These findings form a unified framework of subspace gradient orthogonalization, which we instantiate in a new method, ZO-Muon, admitting a natural interpretation as a low-rank Muon optimizer in the ZO setting. Extensive experiments on large language models (LLMs) and vision transformers (ViTs) demonstrate that ZO-Muon significantly accelerates convergence and achieves a win-win improvement in accuracy and query/runtime efficiency. Notably, compared to the popular MeZO baseline, ZO-Muon requires only 24.7% of the queries to reach the same SST-2 performance for LLM fine-tuning, and improves accuracy by 25.1% on ViT-B fine-tuning on CIFAR-100.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。