用卷积码实现2-2.75比特极低比特量化,让大模型单卡部署
CCQ: Convolutional Code for Extreme Low-bit Quantization in LLMs
- 创新使用卷积码与混合编码,构建无查表映射的压缩方案
- 2比特量化下保持高精度,3000亿参数模型可单卡运行
- 开源2比特模型及推理引擎,适合资源受限场景部署
大型语言模型(LLMs)的快速扩展推高了推理成本,形成显著部署障碍。虽然8比特或4比特量化可缓解此问题,但低于3比特的量化方法常导致精度、可扩展性和效率严重下降。本文提出卷积码量化(CCQ),一种面向推理优化的量化方法,可将LLMs压缩至2.0-2.75比特,同时保持极小精度损失。不同于易出错的标量量化或速度慢的向量量化,CCQ结合硬件感知的位移编码/解码机制,引入卷积码、混合编码和码簇结构,协同突破精度-速度瓶颈。我们构建无需查表的编码空间,实现码本与权重向量间的线性映射,从而优化推理性能。同时,借鉴向量量化中的数据映射思想,最小化超低比特条件下的性能下降。实验表明,CCQ在多个基准上表现优异:将DeepSeek-V3(671B参数)压缩至184GB,将ERNIE-4.5-300B-A47B压缩至89GB,实现ERNIE 4.5的单卡部署并消除卡间通信。2比特版ERNIE-4.5-300B-A47B模型及其推理引擎已开源。
原文摘要 · Abstract (English)
The rapid scaling of Large Language Models (LLMs) elevates inference costs and compounds substantial deployment barriers. While quantization to 8 or 4 bits mitigates this, sub-3-bit methods face severe accuracy, scalability, and efficiency degradation. We propose Convolutional Code Quantization (CCQ), an inference-optimized quantization approach compressing LLMs to 2.0-2.75 bits with minimal accuracy loss. Departing from error-prone scalar quantization or slow vector quantization, CCQ integrates a hardware-aware bit-shift encoding and decoding solution with Convolutional Code, Hybrid Encoding, and Code Cluster, jointly overcoming accuracy-speed bottlenecks. We construct a lookup-free encoding space, enabling a linear mapping between the codebook and weight vectors, thereby optimizing inference performance. Meanwhile, by drawing on the concept of data mapping from vector quantization, we minimize the performance degradation of the model under extremely low-bit conditions. Experiments demonstrate that CCQ achieves outstanding performance on LLMs across various benchmarks. We compress DeepSeek-V3 (671B total parameters) to 184GB and ERNIE-4.5-300B-A47B to 89GB, enabling single-GPU deployment of ERNIE 4.5 and eliminating inter-card communication. The 2-bit ERNIE-4.5-300B-A47B model and inference engine have been open-sourced.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。