arXiv:2501.09107cs.LG2025-01被引 1

提出统计预校准方法,提升大模型量化后精度稳定性。

Rethinking Post-Training Quantization: Introducing a Statistical Pre-Calibration Approach

  • 用KL散度最小化保持权重量化前后分布一致
  • 在多个LLM上达到与校准法相当的精度
  • 适合追求高效部署的模型压缩研究者

随着大语言模型计算复杂度增加,高效部署策略如量化变得至关重要。当前先进的后训练量化(PTQ)技术依赖校准过程以维持模型精度,但在某些领域效果不佳。本文提出一种基于统计的权重自适应预校准方法,作为校准前的前置步骤,通过最小化量化权重与原始权重之间的KL散度,保留原模型的香农信息量,从而确保量化模型在多任务中具备鲁棒性与高效性。该方法性能可媲美主流校准型PTQ方案,并为后续校准提供更优初始状态。实验表明,本方法在多种LLM上实现与现有校准方法相当的精度。

原文摘要 · Abstract (English)

As Large Language Models (LLMs) become increasingly computationally complex, developing efficient deployment strategies, such as quantization, becomes crucial. State-of-the-art Post-training Quantization (PTQ) techniques often rely on calibration processes to maintain the accuracy of these models. However, while these calibration techniques can enhance performance in certain domains, they may not be as effective in others. This paper aims to draw attention to robust statistical approaches that can mitigate such issues. We propose a weight-adaptive PTQ method that can be considered a precursor to calibration-based PTQ methods, guiding the quantization process to preserve the distribution of weights by minimizing the Kullback-Leibler divergence between the quantized weights and the originally trained weights. This minimization ensures that the quantized model retains the Shannon information content of the original model to a great extent, guaranteeing robust and efficient deployment across many tasks. As such, our proposed approach can perform on par with most common calibration-based PTQ methods, establishing a new pre-calibration step for further adjusting the quantized weights with calibration. We show that our pre-calibration results achieve the same accuracy as some existing calibration-based PTQ methods on various LLMs.

量化大模型统计方法

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