Scaling laws for post-training quantized large language models
Authors: Zifei Xu, Alexander Lan, Wanzin Yazar, Tristan Webb, Sayeh Sharify, Xin Wang
AI-generated Key Points
- Study title: "Scaling laws for post-training quantized large language models"
- Investigates predictability of post-training weight quantization performance for large language models (LLMs)
- Well-trained LLMs have predictable generalization abilities based on model size
- Quality of compressed LLMs is often unpredictable and requires individual validation
- Conducted systematic empirical study on multiple LLM families using popular weight quantization techniques and low-precision tensor data types
- Identified key scaling factors related to the local loss landscape to predict performance of quantized LLMs
- Developed a statistical model based on these factors
- Explored complexities and trade-offs in post-training weight quantization, highlighting challenges in finding optimal quantization formats and model parameter counts within fixed constraints
- Findings shed light on how properties such as pre-trained negative log-likelihood (NLL) loss scale with total parameter counts in transformer layers' weight tensors
- Provided insights into local radial loss landscape mapping and illustrated trade-off between larger models quantized to lower bit formats versus smaller models quantized to higher bit formats
- Accepted at the 4th NeurIPS Workshop on Efficient Natural Language and Speech Processing (ENLSP) in 2024
- Contributes valuable insights into improving predictability and efficiency of post-training weight quantization for large language models
Abstract: Generalization abilities of well-trained large language models (LLMs) are known to scale predictably as a function of model size. In contrast to the existence of practical scaling laws governing pre-training, the quality of LLMs after post-training compression remains highly unpredictable, often requiring case-by-case validation in practice. In this work, we attempted to close this gap for post-training weight quantization of LLMs by conducting a systematic empirical study on multiple LLM families quantized to numerous low-precision tensor data types using popular weight quantization techniques. We identified key scaling factors pertaining to characteristics of the local loss landscape, based on which the performance of quantized LLMs can be reasonably well predicted by a statistical model.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.