When Quantization Affects Confidence of Large Language Models?

Irina Proskurina, Luc Brun, Guillaume Metzler, Julien Velcin
5/1/2024
cs.CLcs.AI

No official code found. Best implementation: upunaprosk/quantized-lm-confidence (3 stars, Apache-2.0, last commit 2024-12-30).

Abstract

Recent studies introduced effective compression techniques for Large Language Models (LLMs) via post-training quantization or low-bit weight representation. Although quantized weights offer storage efficiency and allow for faster inference, existing works have indicated that quantization might compromise performance and exacerbate biases in LLMs. This study investigates the confidence and calibration of quantized models, considering factors such as language model type and scale as contributors to quantization loss. Firstly, we reveal that quantization with GPTQ to 4-bit results in a decrease in confidence regarding true labels, with varying impacts observed among different language models. Secondly, we observe fluctuations in the impact on confidence across different scales. Finally, we propose an explanation for quantization loss based on confidence levels, indicating that quantization disproportionately affects samples where the full model exhibited low confidence levels in the first place.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

Code for NAACL paper When Quantization Affects Confidence of Large Language Models?

31Python, Jupyter NotebookMar 27, 20241 years agoApache-2.0
compressionefficient-modelgptqlarge-language-modelsllm+3 more

Similar Papers

Cite this paper

@article{proskurina2024when,
  title  = {When Quantization Affects Confidence of Large Language Models?},
  author = {Irina Proskurina and Luc Brun and Guillaume Metzler and Julien Velcin},
  year   = {2024},
  eprint = {2405.00632},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2405.00632v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.00632.svg)](https://codeofpaper.com/paper/2405.00632)

Discussion