无需重复微调,一键适配任意量化配置的高效大模型压缩方法
On-the-Fly Adaptation to Quantization: Configuration-Aware LoRA for Efficient Fine-Tuning of Quantized LLMs
- 根据量化配置动态生成低秩适配器,避免重复训练
- 通过帕累托优化筛选高质量训练配置集,提升适配精度
- 适合边缘设备上多场景部署的轻量化大模型应用
随着大型预训练模型不断发布,为保护隐私将其部署于边缘设备需有效压缩。现有方法将量化与高精度LoRA适配器微调结合,可大幅缩减模型尺寸并缓解量化带来的精度损失。然而,边缘设备能力各异,对每种量化配置进行独立微调在计算上不可行。本文提出CoA-LoRA,一种能动态适应任意量化配置(即模型各层的位宽选择)的方法,无需重复微调。其核心是配置感知模型,将每种配置映射为对应的低秩调整。该模型效果依赖于训练配置集的质量,即覆盖不同总位宽预算的配置集合。为此,我们设计基于帕累托的配置搜索算法,迭代优化训练配置集,获得更精确的低秩调整。实验表明,相比需为每种配置单独微调的最先进方法,CoA-LoRA无额外时间成本,性能相当甚至更优。
原文摘要 · Abstract (English)
As increasingly large pre-trained models are released, deploying them on edge devices for privacy-preserving applications requires effective compression. Recent works combine quantization with the fine-tuning of high-precision LoRA adapters, which can substantially reduce model size while mitigating the accuracy loss from quantization. However, edge devices have inherently heterogeneous capabilities, while performing configuration-wise fine-tuning for every quantization setting is computationally prohibitive. In this paper, we propose CoA-LoRA, a method that dynamically adjusts the LoRA adapter to arbitrary quantization configurations (i.e., the per-layer bit-width choices of a pre-trained model) without requiring repeated fine-tuning. This is accomplished via a configuration-aware model that maps each configuration to its low-rank adjustments. The effectiveness of this model critically depends on the training configuration set, a collection of configurations chosen to cover different total bit-width budgets. However, constructing a high-quality configuration set is non-trivial. We therefore design a Pareto-based configuration search that iteratively optimizes the training configuration set, yielding more precise low-rank adjustments. Our experiments demonstrate that, unlike the state-of-the-art methods that require fine-tuning a separate LoRA adapter for each configuration, CoA-LoRA incurs no additional time cost while achieving comparable or even superior performance to those methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。