让量化模型也能高效微调,用进化策略突破精度瓶颈。
Quantized Evolution Strategies: High-precision Fine-tuning of Quantized LLMs at Low-precision Cost
- 提出量子化进化策略(QES),直接在量化空间微调模型参数。
- 在多个任务上超越现有零阶优化方法,实现高精度微调效果。
- 适合资源受限设备上的大模型持续优化,无需恢复高精度权重。
后训练量化(PTQ)对在内存受限设备上部署大语言模型至关重要,但使模型静态化且难以微调。传统微调依赖反向传播与连续权重计算梯度,无法适用于量化模型——其参数空间为离散且不可导。虽然进化策略(ES)可避免反向传播,但量化参数优化仍可能因梯度估计消失或不准而失败。本文提出量子化进化策略(QES),一种可在量化空间中进行全参数微调的优化范式。QES基于两项创新:(1)引入累积误差反馈以保留高精度更新信号;(2)采用无状态种子重放机制,将内存使用降至低精度推理水平。QES在多种任务上显著优于当前最先进的零阶微调方法,首次实现量化模型的直接高效微调,为全量化空间扩展大模型提供了可能。代码已开源:https://github.com/dibbla/Quantized-Evolution-Strategies。
原文摘要 · Abstract (English)
Post-Training Quantization (PTQ) is essential for deploying Large Language Models (LLMs) on memory-constrained devices, yet it renders models static and difficult to fine-tune. Standard fine-tuning paradigms, including Reinforcement Learning (RL), fundamentally rely on backpropagation and continuous weights to compute gradients. Thus they cannot be used on quantized models, where the parameter space is discrete and non-differentiable. While Evolution Strategies (ES) offer a backpropagation-free alternative, optimization of the quantized parameters can still fail due to vanishing or inaccurate gradient estimation. This paper introduces Quantized Evolution Strategies (QES), an optimization paradigm that performs full-parameter fine-tuning directly in the quantized space. QES is based on two innovations: (1) it integrates accumulated error feedback to preserve high-precision weight updating signals, and (2) it utilizes a stateless seed replay to reduce memory usage to low-precision inference levels. QES significantly outperforms the state-of-the-art zeroth-order fine-tuning methods on a variety of tasks, making direct fine-tuning for quantized models possible. It therefore opens up the possibility for scaling up LLMs entirely in the quantized space. The source code is available at https://github.com/dibbla/Quantized-Evolution-Strategies .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。