Initialisation Determines the Basin: Efficient Codebook Optimisation for Extreme LLM Quantization
Not on the map. Why?
Abstract (arXiv)
Additive quantization enables extreme LLM compression with O(1) lookup-table dequantization, making it attractive for edge deployment. Yet at 2-bit precision, it often fails catastrophically, even with extensive search and finetuning. We show that the dominant bottleneck is codebook initialisation. Greedy sequential initialisation frequently places the model in poor optimisation regions that subsequent beam search and PV-tuning struggle to overcome. We analyse this behaviour through the representational ratio rho = N/KM, which characterises the relationship between weight groups and codebook capacity, and propose OA-EM, an output-aware EM initialisation method using Hessian-weighted Mahalanobis distance. Across compression rates, search budgets, and three architectures (Llama 3.2 3B, Llama 3.1 8B, Qwen 2.5 3B), OA-EM consistently produces better solutions after PV-tuning and dominates the quality-compute frontier. The severity of the bottleneck scales with rho: moderate at 3 bpp but extreme at 2 bpp, where poor initialisation can degrade perplexity by orders of magnitude. More broadly, our results highlight the importance of optimisation geometry in compressed model spaces, where initialisation can dominate subsequent search and fine-tuning.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 18 of its 33 references that are in the map. First posted on arXiv 9 Apr 2026.
Builds on
18 of 33 references.
References · 18
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,355
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,838
- 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,234
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 960
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 798
- 2024SpinQuant: LLM quantization with learned rotationscited by 446
- 2023QuIP: 2-Bit Quantization of Large Language Models With Guaranteescited by 397
- 2023SpQR: A Sparse-Quantized Representation for Near-Lossless LLM Weight Compressioncited by 359
6 more
- 2024QuIP#: Even Better LLM Quantization with Hadamard Incoherence and Lattice Codebookscited by 354
- 2024Extreme Compression of Large Language Models via Additive Quantizationcited by 223
- 2024QTIP: Quantization with Trellises and Incoherence Processingcited by 86
- 2024GPTVQ: The Blessing of Dimensionality for LLM Quantizationcited by 78
- 2024PV-Tuning: Beyond Straight-Through Estimation for Extreme LLM Compressioncited by 59
- 2025Learning Grouped Lattice Vector Quantizers for Low-Bit LLM Compressioncited by 4
Built on by
Not cited yet.