arXiv:2607.02893cs.LGcs.CL2026-07

让语言模型的权重用不同位数存储,更省空间还更准。

Variable Bit-width Quantization: Learning Per-Group Precision for "Bigger-but-Smaller" Language Models

  • 按权重分组动态分配1-8位精度,训练时自动学习最优配置。
  • 131M模型仅1.82位平均比特,性能超55M全精度模型,存空间少3.8倍。
  • 适合做轻量化部署,尤其在苹果芯片上推理速度提升4.7倍。

低比特量化能缩小语言模型,但传统方法对所有权重使用统一精度。本文提出变量比特量化(VBQ),一种训练阶段的方法:每64个连续权重可自主选择1、2、4、8位中的任意精度,通过Gumbel-Softmax松弛实现联合优化,获得清晰的任务对齐信号。VBQ发现投影层内存在高度异质的精度分布——69%的组压缩至1位,语言模型头平均仅1.09位,而首个MLP块保持约2.5位。该模式稳定,可固化为固定方案复用。此方案实现“大而小”模型:131M模型均值1.82位,在TinyStories上达到困惑度4.2,优于55M FP16模型(PPL 4.4),存储量减少3.8倍;1.46B模型在FineWeb-Edu上匹配593M FP16控制组,存储减少3.7倍,参数多2.5倍。质量/字节效率比FP16高3.9-8.4倍。该方案可直接映射到紧凑存储,加速推理:在苹果硅芯片上,自定义融合解量化乘法核使内存带宽瓶颈下的自回归解码更快,速度提升随规模增长(131M持平,1.0B提升1.9倍,9B达4.7倍)。分布分析显示深层逐步修复浅层引入的量化误差。这一优势来自从头训练,目前扩展至15亿以上参数的搜索仍具挑战。VBQ将精度视为可学习的非均匀资源,证明不均分配优于均分。

原文摘要 · Abstract (English)

Low-bit quantization shrinks language models but treats precision as a single global hyper-parameter: every weight uses the same bit-width. We introduce Variable Bit-width Quantization (VBQ), a training-time method in which each contiguous group of 64 weights learns its own resolution from {1,2,4,8} bits via a Gumbel-Softmax relaxation, trained jointly by an alternating optimization that gives the precision logits a clean, task-aligned signal. VBQ discovers a consistent, strongly heterogeneous allocation within individual projection types, not merely across layers, impossible to express with per-layer methods: 69% of groups collapse to 1 bit, the LM head averages 1.09 bits, while the first MLP block keeps ~2.5 bits. This pattern is stable enough to freeze into a fixed recipe and reuse without further search. The recipe yields a "bigger-but-smaller" regime: a 131M model at 1.82 mean bits reaches perplexity 4.2 on TinyStories, beating a 55M FP16 model (PPL 4.4) at 3.8x less storage, and lets a 1.46B model on FineWeb-Edu match a 593M FP16 control at ~3.7x less storage with 2.5x more parameters. As quality-per-byte, VBQ is 3.9-8.4x more efficient than FP16. The recipe maps directly to packed low-bit storage, so it also accelerates inference: with custom fused dequantize-and-multiply kernels, memory-bandwidth-bound autoregressive decode is faster at equal output, and the speedup grows with scale (parity at 131M, 1.9x at 1.0B, 4.7x at 9B on Apple silicon). A distributional analysis (KL divergence and argmax-flip rate) reveals a striking mechanism: deeper layers progressively self-heal the quantization error injected by early layers. The win is a from-scratch, train-time phenomenon; scaling the search economically beyond 1.5B parameters remains open. VBQ reframes precision as a learnable, non-uniform resource and shows that spending a fixed bit budget unevenly beats spending it uniformly.

量化模型压缩高效推理语言模型

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