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

OWQ: Outlier-Aware Weight Quantization for Efficient Fine-Tuning and Inference of Large Language Models

About

Large language models (LLMs) with hundreds of billions of parameters require powerful server-grade GPUs for inference, limiting their practical deployment. To address this challenge, we introduce the outlier-aware weight quantization (OWQ) method, which aims to minimize LLM's footprint through low-precision representation. OWQ prioritizes a small subset of structured weights sensitive to quantization, storing them in high-precision, while applying highly tuned quantization to the remaining dense weights. This sensitivity-aware mixed-precision scheme reduces the quantization error notably, and extensive experiments demonstrate that 3.1-bit models using OWQ perform comparably to 4-bit models optimized by OPTQ. Furthermore, OWQ incorporates a parameter-efficient fine-tuning for task-specific adaptation, called weak column tuning (WCT), enabling accurate task-specific LLM adaptation with minimal memory overhead in the optimized format. OWQ represents a notable advancement in the flexibility, efficiency, and practicality of LLM optimization literature. The source code is available at https://github.com/xvyaward/owq

Changhun Lee, Jungyu Jin, Taesu Kim, Hyungjun Kim, Eunhyeok Park• 2023

Related benchmarks

TaskDatasetResultRank
Language ModelingWikiText-2
Perplexity (PPL)5.8
2320
Zero-shot ReasoningZero-Shot Reasoning Tasks (ARC-C, ARC-E, BoolQ, Hella, OBQA, PIQA, SIQA, Wino)
ARC-C Accuracy53.07
54
ClassificationCovertype
Accuracy92.9
40
ClassificationQoE
Accuracy77.9
26
RegressionKVS
MSE2.41
26
Classificationhiggs
Accuracy75.3
26
RegressionRSS
MSE9.225
26
RegressionVoD
MSE15.184
26
ClassificationCovertype
Accuracy79.9
8
Mixed-precision QuantizationVoD
Avg Bit-width (Weights, w/o Act Quant)4.28
3
Showing 10 of 14 rows

Other info

Follow for update