通过三阶段蒸馏实现大模型轻量化二值化,显著提升低比特推理性能。
LBLLM: Lightweight Binarization of Large Language Models via Three-Stage Distillation

- 分三阶段蒸馏:先初始化,再逐层压缩权重与参数,最后动态量化激活至4比特。
- 仅用0.016B token和单张显卡训练,优于现有W2A4量化方法。
- 无需高精度通道或旋转矩阵,适合边缘设备部署的高效大模型应用。
在资源受限环境部署大语言模型(LLMs)面临计算与内存开销大的挑战。本文提出LBLLM,一种轻量级二值化框架,通过创新的三阶段量化策略实现有效的W(1+1)A4量化。流程包括:(1) 通过预训练量化(PTQ)初始化高质量量化模型;(2) 在保持激活全精度的前提下,通过逐层蒸馏量化二值化权重、分组位图及量化参数;(3) 训练可学习的激活量化因子,动态将激活量化至4比特。该解耦设计缓解了权重与激活量化间的干扰,提升训练稳定性与推理准确性。LBLLM仅使用0.016B tokens和单张GPU训练,在语言建模、常识问答与语言理解任务上均超越当前最先进的二值化方法,适用于W2A4量化设置。结果表明,极端低比特量化在不引入额外高精度通道或旋转矩阵的情况下仍具可行性,为资源受限场景下的高效大模型部署提供了可行路径。
原文摘要 · Abstract (English)
Deploying large language models (LLMs) in resource-constrained environments is hindered by heavy computational and memory requirements. We present LBLLM, a lightweight binarization framework that achieves effective W(1+1)A4 quantization through a novel three-stage quantization strategy. The framework proceeds as follows: (1) initialize a high-quality quantized model via PTQ; (2) quantize binarized weights, group-wise bitmaps, and quantization parameters through layer-wise distillation while keeping activations in full precision; and (3) training learnable activation quantization factors to dynamically quantize activations to 4 bits. This decoupled design mitigates interference between weight and activation quantization, yielding greater training stability and better inference accuracy. LBLLM, trained only using 0.016B tokens with a single GPU, surpasses existing state-of-the-art binarization methods on W2A4 quantization settings across tasks of language modeling, commonsense QA, and language understanding. These results demonstrate that extreme low-bit quantization of LLMs can be both practical and highly effective without introducing any extra high-precision channels or rotational matrices commonly used in recent PTQ-based works, offering a promising path toward efficient LLM deployment in resource-limited situations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。