English

An exploration of the effect of quantisation on energy consumption and inference time of StarCoder2

Computation and Language 2024-11-21 v1 Artificial Intelligence Software Engineering

Abstract

This study examines quantisation and pruning strategies to reduce energy consumption in code Large Language Models (LLMs) inference. Using StarCoder2, we observe increased energy demands with quantization due to lower throughput and some accuracy losses. Conversely, pruning reduces energy usage but impairs performance. The results highlight challenges and trade-offs in LLM model compression. We suggest future work on hardware-optimized quantization to enhance efficiency with minimal loss in accuracy.

Keywords

Cite

@article{arxiv.2411.12758,
  title  = {An exploration of the effect of quantisation on energy consumption and inference time of StarCoder2},
  author = {Pepijn de Reus and Ana Oprescu and Jelle Zuidema},
  journal= {arXiv preprint arXiv:2411.12758},
  year   = {2024}
}