arXiv Machine Learning

SQuAT: Spherical Lattice Vector Quantization-Aware Training

SQuAT introduces a shape‑gain vector quantization‑aware training (VQ‑QAT) method that separates weight direction and magnitude, using an $E_8$‑derived spherical codebook for directions and low‑bit scalar quantizers for gains. By applying distinct straight‑through estimators (STE) to each component, SQuAT induces tangent‑space updates that act as an implicit spherical (Riemannian) optimization, improving training dynamics over standard STE approaches. A distributed exact nearest‑neighbor assignment scheme further reduces training time by 12.8×, enabling scaling to 10 B tokens and achieving state‑of‑the‑art accuracy on several large language models.

Hugging Face Trending Papers
Oct 22

SQuAT: Spherical Lattice Vector Quantization-Aware Training

SQuAT introduces a shape‑gain vector‑quantization‑aware training (VQ‑QAT) method that separates weight direction and magnitude, using an $E_8$‑derived spherical codebook for directions and low‑bit scalar quantizers for gains. The approach employs distinct straight‑through estimators for shape and gain, enabling tangent‑space updates that act as implicit spherical optimization. Distributed exact nearest‑neighbor assignment reduces training time by 12.8×, allowing scaling to 10 B tokens and yielding state‑of‑the‑art accuracy on several large language models.

arXiv Machine Learning
Oct 22

InfoQuant: Shaping Activation Distributions for Low-Bit LLM Quantization

InfoQuant is a train‑free method that reshapes activation distributions to be more compatible with low‑bit quantizers. It uses Peak Suppression Orthogonal Transformation (PSOT) and adaptive outlier‑token selection to reduce numerical range while maintaining dispersion, thereby lowering quantization error. Across several large language model families, InfoQuant outperforms existing post‑training quantization and end‑to‑end training baselines, preserving 97% of floating‑point accuracy on average and narrowing the LLaMA‑2 13B performance gap by 42%.

By Ke Li, Dong An, Xiaoling Zang, Can Ye, Liang Xie, Qibo Qiu, Chen Shen, Xiaofei He, Wenxiao Wang
arXiv Machine Learning
Oct 8

Few Bits, One Law: Toward W2A4KV2

The paper introduces CanonQ, a unified quantization‑aware training framework that separates source canonicalization from task‑aware adaptation to enable extreme low‑bit LLM compression. By applying fixed rotations and energy normalization, CanonQ maps heterogeneous tensors to canonical coordinates, allowing frozen Gaussian‑reference codebooks to be reused across layers and models. Joint training then adapts the network to the coupled errors of weight, activation, and KV cache quantization, achieving significant performance gains—up to 14.28× lower perplexity and 57.9% higher zero‑shot accuracy on LLaMA models, and substantial improvements on instruction‑tuned MobileLLM‑Pro‑1B. "whyItMatters":"The framework demonstrates that carefully designed canonicalization and joint training can dramatically improve the efficiency and accuracy of low‑bit quantized large language models, offering a practical path to deploy powerful LLMs with reduced computational and memory footprints."

By Kai Yi, Tarek Elgamal, Sruthikesh Surineni, Vignesh Vivekraja, Soumyadeep Ghosh, Steven Li
arXiv Machine Learning
5d ago

Trainable Vector Quantization for Large Language Models

The paper introduces Trainable Vector Quantization (TVQ), a method that uses straight-through estimation to allow gradient descent optimization of vector-quantized large language models. TVQ reformulates VQ with trainable latent weights, enabling arbitrary objective minimization and is applied in a block-wise knowledge distillation framework to balance quality and training efficiency. Experiments show TVQ achieves state‑of‑the‑art low‑bit post‑training quantization, achieving near‑lossless 2‑bit quantization for LLaMA2‑70B with only a 1% accuracy drop on question‑answering benchmarks.

By Anastasiia Prutianova, Alina Bogdanova, Mikhail Shutov, Dmitrii Poimanov, Anatoly Gorkov, Ilia Zharikov