针对大模型量化中高影响参数分布不均问题,提出分层优化策略。
Layer-Wise High-Impact Parameter Ratio Optimization in Post-Training Quantization for Large Language Models
- 按层动态分配高影响力参数保留比例,考虑层间依赖关系。
- 在相同资源下比现有方法多保留更多关键参数,精度损失更小。
- 适合对量化后模型精度要求高的场景,如部署级大模型应用。
大语言模型(LLMs)虽推动了自然语言处理发展,但其庞大的参数量带来显著的计算与内存挑战。后训练量化(PTQ)成为以极低开销缓解此问题的可行方案。然而,现有方法在极低比特位宽下仍存在显著精度下降,主要源于对高影响参数处理不当。部分方法通过识别并保留高影响参数为FP16格式来缓解,但统一各层固定比例,忽略了层间敏感性差异。本文提出一种二次优化框架,实现层内差异化高影响参数保留比例,并考虑层间依赖。将高影响参数量化至中等比特位宽,通常导致性能几乎无损;其余参数则可降至极低比特位宽。在相同资源约束下,该方法可比固定保留策略保存更多高影响参数。同时,可结合需大量可学习参数的先进量化方法仅作用于高影响部分,其余使用高效方法。实验表明,本方法在保持计算效率的同时,优于当前最先进水平。
原文摘要 · Abstract (English)
Large language models (LLMs) have significantly advanced natural language processing, but their massive parameter counts create substantial computational and memory challenges during deployment. Post-training quantization (PTQ) has emerged as a promising approach to mitigate these challenges with minimal overhead. While existing PTQ methods can effectively quantize LLMs, they experience substantial accuracy loss at extremely low bit-widths, primarily due to high-impact parameters that significantly influence quantization performance. Several approaches address these issues by identifying and retaining the high-impact parameters in FP16 format. However, they apply fixed ratios of high-impact parameters across all layers, overlooking layer-wise sensitivity variations. In this paper, we propose a quadratic optimization framework that determines layer-specific ratios of high-impact parameters while considering inter-layer dependencies. We quantize high-impact parameters to moderate bit-widths, which often result in negligible performance degradation in quantized LLMs, while the remaining parameters can be quantized to extremely low bit-widths. Under the same resource-constrained budget, this allows for preserving more high-impact parameters than methods that keep selecting a few in FP16 format. Additionally, the proposed framework allows us to leverage an advanced quantization method that often requires extensive learnable parameters solely for high-impact parameters, while applying a computationally efficient method to the rest. Our approach achieves an effective balance between computational efficiency and model accuracy while maintaining high performance compared to state-of-the-art methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。