CAT-Q: Cost-efficient and Accurate Ternary Quantization for LLMs
Abstract
Lay Summary
Large language models (LLMs) are powerful but expensive to store and deploy because they typically contain billions to hundreds of billions of parameters. A promising way to reduce this cost is ternary quantization, which replaces the high-precision values of model parameters with ternary values {1,0,-1}. However, existing ternary quantization methods often require massive retraining data and significant computational resources to avoid serious model accuracy drop. We introduce CAT-Q, a simple and accurate ternary quantization method for efficiently compressing pre-trained LLMs without retraining from scratch. CAT-Q establishes a new framework to guide the conversion of the pre-trained model parameter values into a much more compact ternary form, enabling the compressed model to preserve strong performance using only a small amount of calibration data. In our experiments, CAT-Q compresses LLMs using only 512 calibration samples, while previous top-performing ternary LLM families rely on around 100 billion training tokens. This corresponds to a reduction of around 100,000× in training tokens. More importantly, CAT-Q can easily scale to compress much larger LLMs, ranging from 14B to 235B parameters, within 8 to 60 hours on 8 A100 GPUs. These results demonstrate that highly compact ternary LLMs can be produced with substantially greater efficiency and practicality than ever before.