提出RaBiT框架,让二比特量化模型更准更快。
RaBiT: Residual-Aware Binarization Training for Accurate and Efficient LLMs
- 通过共享权重逐层生成二值路径,避免特征冗余
- 在RTX 4090上实现4.49倍推理加速,性能媲美向量量化
- 适合追求高效部署的LLM工程师和硬件优化研究者
大型语言模型(LLM)的高效部署需要极低比特量化,但会面临效率与性能的权衡。残差二值化通过堆叠±1二值层实现硬件友好、无需矩阵乘法的推理,却存在病态特征共适应问题。我们识别出关键失败模式——路径间适应:在量化感知训练(QAT)中,平行的残差二值路径学习冗余特征,破坏误差补偿结构,限制模型表达能力。现有方法依赖启发式方案(如路径冻结),限制解空间。我们提出RaBiT,一种新量化框架,通过算法强制残差层级结构解决共适应问题。其核心机制从单一全精度权重逐次推导每条二值路径,确保每条路径修正前序路径的误差。该过程由稳健初始化稳定,优先保障功能保留而非单纯权重近似。RaBiT重新定义了2比特下的准确率-效率前沿:达到当前最优性能,媲美硬件密集型向量量化(VQ)方法,并在RTX 4090上实现4.49倍推理加速。代码已公开于https://github.com/SamsungLabs/RaBiT。
原文摘要 · Abstract (English)
Efficient deployment of large language models (LLMs) requires extreme quantization, forcing a critical trade-off between low-bit efficiency and performance. Residual binarization enables hardware-friendly, matmul-free inference by stacking binary ($\pm$1) layers, but is plagued by pathological feature co-adaptation. We identify a key failure mode, which we term inter-path adaptation: during quantization-aware training (QAT), parallel residual binary paths learn redundant features, degrading the error-compensation structure and limiting the expressive capacity of the model. While prior work relies on heuristic workarounds (e.g., path freezing) that constrain the solution space, we propose RaBiT, a novel quantization framework that resolves co-adaptation by algorithmically enforcing a residual hierarchy. Its core mechanism sequentially derives each binary path from a single shared full-precision weight, which ensures that every path corrects the error of the preceding one. This process is stabilized by a robust initialization that prioritizes functional preservation over mere weight approximation. RaBiT redefines the 2-bit accuracy-efficiency frontier: it achieves state-of-the-art performance, rivals even hardware-intensive Vector Quantization (VQ) methods, and delivers a $4.49\times$ inference speed-up over full-precision models on an RTX 4090. Code is available at https://github.com/SamsungLabs/RaBiT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。