arXiv:2605.26660cs.LG2026-05

用强化学习精细分配权重位宽,实现低比特大模型量化性能突破

WINDQuant: Weight-Informed Neural Decision-Making for Global Mixed-Precision LLM Quantization

论文配图:WINDQuant: Weight-Informed Neural Decision-Making for Global Mixed-Precision LLM Quantization
图 1 · 摘自论文原文
  • 基于强化学习动态分配权重列块的位宽,实现细粒度混合精度控制
  • 在极低比特下保持模型性能,相比重训练方法优化开销显著降低
  • 适合追求高效推理部署的大模型应用,尤其适用于资源受限场景

量化是降低大语言模型(LLMs)内存占用和推理成本的有效手段,但在超低比特条件下维持性能仍具挑战。现有后训练方法常导致严重精度下降,而感知量化训练需高成本重训练与额外资源。多数混合精度策略依赖粗粒度或启发式敏感性分析,忽略权重矩阵内部的细粒度差异。本文提出 WINDQuant,一种基于强化学习的全局混合精度量化分配控制器。WINDQuant 不引入新的量化算子,而是学习在全局存储预算下,为细粒度列块分配位宽和量化方案。通过列块级操作,可在层内灵活实现细粒度精度分配,满足全局目标位宽。实现结合 PPO 与激活感知校准、轻量级单元量化拟合及所学混合精度方案的有效位显式计数。在 LLaMA 模型上的实验表明,WINDQuant 在超低比特设置下表现优异,且相较重训练方法显著降低优化开销,凸显强化学习作为自适应混合精度量化控制器的实用性。

原文摘要 · Abstract (English)

Quantization is an effective approach to reduce the memory footprint and inference cost of large language models (LLMs), yet maintaining performance in the ultra-low-bit regime remains challenging. Existing post-training methods often suffer from severe accuracy degradation, while quantization-aware training requires costly retraining and additional resources. Moreover, most mixed-precision strategies rely on coarse-grained or heuristic sensitivity analysis that overlooks fine-grained variations within weight matrices. We propose WINDQuant, a reinforcement-learning-based allocation controller for ultra-low-bit LLM quantization. Rather than introducing another low-level quantization operator, WINDQuant learns how to assign bit-widths and quantization treatments to fine-grained column chunks under a global storage budget. By operating at the column-chunk level, WINDQuant enables flexible and fine-grained precision assignment within layers under a global target bit-width. The implementation combines PPO with activation-aware calibration, lightweight per-unit quantizer fitting, and explicit effective-bit accounting of the learned mixed-precision plan. Experiments on LLaMA models demonstrate that WINDQuant achieves competitive performance in ultra-low-bit settings while reducing optimization overhead relative to retraining-based approaches, highlighting reinforcement learning as a practical controller for adaptive mixed-precision quantization.

大模型量化强化学习混合精度推理优化

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