arXiv:2606.26650cs.CLcs.AI2026-06中稿 · ICML被引 5

无需训练即可高效压缩大模型,实现近百万倍的训练成本降低。

CAT-Q: Cost-efficient and Accurate Ternary Quantization for LLMs

论文配图:CAT-Q: Cost-efficient and Accurate Ternary Quantization for LLMs
图 1 · 摘自论文原文
  • 采用可学习调制与软化三值化协同优化,无需微调
  • 仅用512样本即可完成1.7B~8B模型三值量化,性能超基线
  • 支持14B~235B大模型,8卡80GB A100上60小时内完成

本文提出CAT-Q,一种高效且准确的三值量化方法,用于压缩和加速大语言模型。不同于依赖大量数据和昂贵量化感知训练的现有方法,CAT-Q是一种简单有效的后训练量化方案,适用于多种架构和规模的LLM。其核心包含可学习调制(LM)和软化三值化(ST)两个组件,从优化角度耦合设计。LM通过可学习因子调节预训练权重分布与三值阈值,降低对三值化的敏感性;ST引入可微过渡函数,引导三值过程稳定收敛。实验表明,对于1.7B至8B参数的预训练模型,仅需512个校准样本即可高效量化为三值模型,性能优于使用1000亿训练令牌训练的BitNet 1.58-bit v1/v2系列(1.3B~7B参数),训练量减少约10万倍。此外,首次证明CAT-Q可在8张A100-80GB GPU上,于8至60小时内完成14B至235B参数大模型的三值化。代码已开源。

原文摘要 · Abstract (English)

In this paper, we present CAT-Q, Cost-efficient and Accurate Ternary Quantization, for compressing and accelerating LLMs. Unlike existing state-of-the-art ternary quantization methods that rely on data-intensive and costly quantization-aware training to mitigate severe performance degradation, CAT-Q is a simple yet effective post-training quantization scheme that is readily applicable to LLMs with diverse architectures and model sizes. It has two key components, learnable modulation (LM) and softened ternarization (ST), which are coupled from an optimization perspective. LM leverages a composition of learnable factors to modulate the distribution of pre-trained high-precision weights and the ternary threshold, making them less sensitive to ternarization. ST further introduces a differentiable transition function to guide the ternarization process toward stable convergence. We show that, for pre-trained LLMs with 1.7B to 8B parameters, CAT-Q can efficiently quantize them into ternary models using only 512 calibration samples, while achieving superior performance than the seminal BitNet 1.58-bit v1 and v2 families (with 1.3B to 7B parameters) trained with 100B tokens, yielding about a 100,000X reduction in training tokens. Moreover, we show for the first time that CAT-Q can quantize much larger pre-trained LLMs having 14B to 235B parameters into leading ternary models within just 8 to 60 hours on 8 A100-80GB GPUs. Code is available at https://github.com/IntelChina-AI/BitTern.

模型量化三值化大模型压缩后训练

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