CRVQ: Channel-Relaxed Vector Quantization for Extreme Compression of LLMs

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Xu, Yuzhuang, Ji, Shiyu, Zhu, Qingfu, Che, Wanxiang
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909501346545664
author Xu, Yuzhuang
Ji, Shiyu
Zhu, Qingfu
Che, Wanxiang
author_facet Xu, Yuzhuang
Ji, Shiyu
Zhu, Qingfu
Che, Wanxiang
contents Powerful large language models (LLMs) are increasingly expected to be deployed with lower computational costs, enabling their capabilities on resource-constrained devices. Post-training quantization (PTQ) has emerged as a star approach to achieve this ambition, with best methods compressing weights to less than 2 bit on average. In this paper, we propose Channel-Relaxed Vector Quantization (CRVQ), a novel technique that significantly improves the performance of PTQ baselines at the cost of only minimal additional bits. This state-of-the-art extreme compression method achieves its results through two key innovations: (1) carefully selecting and reordering a very small subset of critical weight channels, and (2) leveraging extended codebooks to relax the constraint of critical channels. With our method, we demonstrate a 38.9\% improvement over the current strongest sub-2-bit PTQ baseline, enabling nearer lossless 1-bit compression. Furthermore, our approach offers flexible customization of quantization bit-width and performance, providing a wider range of deployment options for diverse hardware platforms.
format Preprint
id arxiv_https___arxiv_org_abs_2412_09282
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle CRVQ: Channel-Relaxed Vector Quantization for Extreme Compression of LLMs
Xu, Yuzhuang
Ji, Shiyu
Zhu, Qingfu
Che, Wanxiang
Machine Learning
Computation and Language
Powerful large language models (LLMs) are increasingly expected to be deployed with lower computational costs, enabling their capabilities on resource-constrained devices. Post-training quantization (PTQ) has emerged as a star approach to achieve this ambition, with best methods compressing weights to less than 2 bit on average. In this paper, we propose Channel-Relaxed Vector Quantization (CRVQ), a novel technique that significantly improves the performance of PTQ baselines at the cost of only minimal additional bits. This state-of-the-art extreme compression method achieves its results through two key innovations: (1) carefully selecting and reordering a very small subset of critical weight channels, and (2) leveraging extended codebooks to relax the constraint of critical channels. With our method, we demonstrate a 38.9\% improvement over the current strongest sub-2-bit PTQ baseline, enabling nearer lossless 1-bit compression. Furthermore, our approach offers flexible customization of quantization bit-width and performance, providing a wider range of deployment options for diverse hardware platforms.
title CRVQ: Channel-Relaxed Vector Quantization for Extreme Compression of LLMs
topic Machine Learning
Computation and Language
url https://arxiv.org/abs/2412.09282