arXiv:2509.20214cs.LGcs.AI2025-09NeurIPS被引 6

提出可实现最优比特分配的分数比特量化方法,提升大模型部署效率。

Q-Palette: Fractional-Bit Quantizers Toward Optimal Bit Allocation for Efficient LLM Deployment

  • 基于高斯化权重设计分数比特量化器,逼近理论最优性能。
  • 在8比特预算下,相比基线模型误差降低37.6%,推理速度提升2.4倍。
  • 适用于边缘设备上小批量个性化推理场景,兼顾精度与速度。

本文研究仅量化权重的后训练量化(PTQ),即在无需微调或极少校准数据的情况下对大语言模型(LLM)权重进行量化。该技术对降低模型推理的内存占用和延迟至关重要,尤其适用于内存受限的小批量推理场景,如边缘设备上的个性化推理。尽管重要,但大模型中不规则的权重分布及重尾异常值使量化复杂化,促使近期采用旋转方法将权重转换为近似高斯分布,从而减少异常值并降低量化误差。本文首先推导出在给定比特预算下高斯化权重的信息论最优比特分配,揭示细粒度分数比特量化器接近高斯失真-率极限是实现近优性能的关键。为连接理论与实践,我们提出Q-Palette,一个包含多种分数比特量化器的灵活工具包,涵盖逼近最优失真的网格编码量化器,以及面向快速推理优化的向量与标量量化器,均通过优化的CUDA内核高效实现于多种位宽。此外,以Q-Palette为基础,我们提出一种新型混合量化框架,联合优化量化器选择与层融合策略,在资源约束下实现性能最优化。代码已开源:https://github.com/snu-mllab/Q-Palette。

原文摘要 · Abstract (English)

We study weight-only post-training quantization (PTQ), which quantizes the weights of a large language model (LLM) without retraining, using little or no calibration data. Weight-only PTQ is crucial for reducing the memory footprint and latency of LLM inference, especially in memory-bound, small-batch inference scenarios, such as personalized inference on edge devices. Despite its importance, irregular weight distributions with heavy-tailed outliers in LLMs complicate quantization, recently motivating rotation-based methods that transform weights into near-Gaussian distributions, which are more regular with fewer outliers, thereby reducing quantization error. In this work, we first derive the information-theoretically optimal bit allocation for Gaussianized weights under given bit budgets, revealing that fine-grained fractional-bit quantizers approaching the Gaussian distortion-rate bound are essential to achieve near-optimal quantization performance. To bridge this theoretical insight and practical implementation, we introduce Q-Palette, a versatile collection of fractional-bit quantizers that range from trellis-coded quantizers offering near-optimal distortion to simpler vector and scalar quantizers optimized for faster inference, all efficiently implemented with optimized CUDA kernels across various bitwidths. Furthermore, leveraging Q-Palette as a foundational component, we propose a novel mixed-scheme quantization framework, jointly optimizing quantizer choices and layer fusion decisions given resource constraints. The code is available at https://github.com/snu-mllab/Q-Palette.

量化大模型边缘计算分数比特

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