DBellQuant: Breaking the Bell with Double-Bell Transformation for LLMs Post Training Binarization
Zijian Ye, Wei Huang, Yifei Yu, Tianhe Ren, Zhongrui Wang, Xiaojuan Qi
TL;DR
DBellQuant tackles the challenge of post-training quantization for large language models by introducing a per-channel Learnable Transformation for Dual-Bell Quantization (LTDB) that reshapes weight distributions from unimodal to bimodal, enabling near $1$-bit weight binarization. The inverse of the transformation is applied to activations, which, together with activation-aware initialization and dual loss objectives, smooths activations and suppresses outliers to support low-bit activation quantization. The method leverages two targeted losses, DTMD and DTNP, plus an early-stopping mechanism to reliably converge to a dual-bell weight distribution while preserving computation. Empirical results across multiple LLM families show state-of-the-art performance under aggressive quantization, achieving nearly 1-bit weights with 6-bit or even 4-bit activations and substantial model-size reductions, with practical speedups suitable for edge and real-world deployments. This approach broadens the feasibility of deploying large-scale models in resource-constrained environments without retraining, contributing to more sustainable and accessible AI infrastructure.
Abstract
Large language models (LLMs) demonstrate remarkable performance but face substantial computational and memory challenges that limit their practical deployment. Quantization has emerged as a promising solution; however, its effectiveness is often limited by quantization errors arising from weight distributions that are not quantization-friendly and the presence of activation outliers. To address these challenges, we introduce DBellQuant, an innovative post-training quantization (PTQ) framework that achieves nearly 1-bit weight compression and 6-bit activation quantization with minimal performance degradation. DBellQuant uses Learnable Transformation for Dual-Bell (LTDB) algorithm, which transforms single-bell weight distributions into dual-bell forms to reduce binarization errors and applies inverse transformations to smooth activations. DBellQuant sets a new state-of-the-art by preserving superior model performance under aggressive weight and activation quantization. For example, on the Wikitext2 dataset, DBellQuant achieves a perplexity of 14.39 on LLaMA2-13B with 6-bit activation quantization, significantly outperforming BiLLM's 21.35 without activation quantization, underscoring its potential in compressing LLMs for real-world applications.
