Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SliM-LLM: Salience-Driven Mixed-Precision Quantization for Large Language Models

About

Post-training quantization (PTQ) is an effective technique for compressing large language models (LLMs). However, while uniform-precision quantization is computationally efficient, it often compromises model performance. To address this, we propose SliM-LLM, a salience-driven mixed-precision quantization framework that allocates bit-widths at the group-wise. Our approach leverages the observation that important weights follow a structured distribution and introduces two key components: \textbf{1)} \textit{Salience-Determined Bit Allocation} adaptively assigns bit-widths to groups within each layer based on their salience; and \textbf{2)} \textit{Salience-Weighted Quantizer Calibration} optimizes quantizer parameters by incorporating element-level salience. With its structured partitioning, SliM-LLM provides a hardware-friendly solution that matches the efficiency of uniform quantization methods while improving accuracy. Experiments show that SliM-LLM achieves superior performance across various LLMs at low bit-widths. For example, a 2-bit quantized LLaMA-7B model reduces memory usage by nearly 6x compared to the floating-point baseline, decreases perplexity by 48\% compared to state-of-the-art gradient-free PTQ methods, and maintains GPU inference speed. Additionally, the extended version, SliM-LLM$^+$, which incorporates gradient-based quantization, further reduces perplexity by 35.1\%. Our code is available at https://github.com/Aaronhuang-778/SliM-LLM

Wei Huang, Haotong Qin, Yangdong Liu, Yawei Li, Qinshuo Liu, Xianglong Liu, Luca Benini, Michele Magno, Shiming Zhang, Xiaojuan Qi• 2024

Related benchmarks

TaskDatasetResultRank
Language ModelingWikiText2
Perplexity9.06
4085
Language ModelingWikiText-2
Perplexity (PPL)4.08
2862
Language ModelingC4 (val)
PPL8.4
908
Question AnsweringPIQA
Accuracy61.7
589
Visual Mathematical ReasoningMathVista
Accuracy21.2
448
Sentence CompletionHellaSwag
Accuracy44.1
440
Language ModelingWikiText2 v1 (test)
Perplexity6.07
396
Language ModelingWiki2
PPL9.35
382
Mathematical ReasoningMathQA
Accuracy23.62
354
Multi-task Language UnderstandingMMLU
Accuracy25.43
353
Showing 10 of 36 rows

Other info

Follow for update