arXiv:2506.13771cs.LGcs.AI2025-06NeurIPS被引 13

将大模型权重压缩至0.1比特,实现31倍内存节省

LittleBit: Ultra Low-Bit Quantization via Latent Factorization

  • 用低秩分解表示权重,再对因子二值化
  • 0.1比特时性能超越0.7比特的现有方法
  • 适合部署大模型的低资源设备使用

大语言模型(LLMs)部署常受内存和计算需求限制。尽管量化可缓解此问题,但在亚1比特区间保持模型精度仍是难题。本文提出LittleBit框架,实现低至0.1比特每权重(BPW)的极端压缩,使Llama2-13B模型内存降至约0.9GB(压缩比约31倍)。通过低秩潜在矩阵分解表示权重,并对分解因子进行二值化。为弥补极端精度损失,引入多尺度补偿机制,学习行、列及潜在维度的重要性参数。两大贡献:双独立符号-值分解(Dual-SVID)用于量化感知训练初始化,残差补偿机制减少近似误差。大量实验表明,该方法在亚1比特域表现领先;例如,0.1 BPW下的性能超过当前0.7 BPW方案在Llama2-7B上的表现。建立新的大小-性能权衡——相比FP16推理可提速11.6倍,使高性能大模型适用于资源受限环境。代码已开源。

原文摘要 · Abstract (English)

The deployment of large language models (LLMs) is frequently hindered by prohibitive memory and computational requirements. While quantization mitigates these bottlenecks, maintaining model fidelity in the sub-1-bit regime remains a persistent challenge. In this paper, we introduce LittleBit, a novel framework for extreme LLM compression. We target quantization rates as low as $0.1$ bits per weight (BPW), achieving a memory reduction of approximately $31\times$, which effectively compresses Llama2-13B to under $0.9$ GB. We represent weights via low-rank latent matrix factorization and subsequently binarize the resulting factors. To counteract the information loss inherent to such drastic precision reduction, we integrate a multi-scale compensation mechanism that learns importance parameters across row, column, and latent dimensions. Two primary contributions enable effective training: Dual Sign-Value-Independent Decomposition (Dual-SVID) for quantization-aware training (QAT) initialization, and Residual Compensation to minimize approximation errors. Extensive experiments confirm the superiority of LittleBit in the sub-1-bit domain; for instance, our method at $0.1$ BPW surpasses the performance of leading techniques operating at $0.7$ BPW on Llama2-7B. We establish a new size-performance trade-off -- unlocking a potential $11.6\times$ inference speedup relative to FP16 -- and render powerful LLMs practical for resource-constrained environments. Our code is available at https://github.com/SamsungLabs/LittleBit.

模型压缩低比特量化大模型部署

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