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
2862
Physical Interaction Question AnsweringPIQA
Accuracy77.1
462
Multiple-choice Question AnsweringARC Easy
Accuracy79.8
269
Zero-shot ReasoningZero-Shot Reasoning Tasks (ARC-C, ARC-E, BoolQ, Hella, OBQA, PIQA, SIQA, Wino)
ARC-C Accuracy53.07
54
ClassificationCovertype
Accuracy92.9
52
Commonsense ReasoningHellaSwag
HS Accuracy75.4
44
Commonsense ReasoningWinoGrande (WG)
Accuracy66.4
38
Language ModelingLanguage Modeling Dataset
PPL10.16
31
ClassificationQoE
Accuracy77.9
26
RegressionKVS
MSE2.41
26
Showing 10 of 22 rows

Other info

Follow for update