Search papers, labs, and topics across Lattice.
This paper introduces CAT-Q, a novel post-training quantization method for large language models (LLMs) that significantly reduces the need for extensive quantization-aware training. By employing learnable modulation and softened ternarization, CAT-Q achieves superior performance while requiring only 512 calibration samples, resulting in a 100,000X reduction in training tokens compared to existing methods. Notably, CAT-Q can effectively quantize LLMs ranging from 1.7B to 235B parameters within a time frame of 8 to 60 hours on 8 A100-80GB GPUs, demonstrating its efficiency and scalability across diverse model architectures.
CAT-Q quantizes large language models with unprecedented efficiency, achieving superior performance while slashing training token requirements by 100,000X.
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.
