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
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language Modeling | WikiText-2 | Perplexity (PPL)5.8 | 2862 | |
| Physical Interaction Question Answering | PIQA | Accuracy77.1 | 462 | |
| Multiple-choice Question Answering | ARC Easy | Accuracy79.8 | 269 | |
| Zero-shot Reasoning | Zero-Shot Reasoning Tasks (ARC-C, ARC-E, BoolQ, Hella, OBQA, PIQA, SIQA, Wino) | ARC-C Accuracy53.07 | 54 | |
| Classification | Covertype | Accuracy92.9 | 52 | |
| Commonsense Reasoning | HellaSwag | HS Accuracy75.4 | 44 | |
| Commonsense Reasoning | WinoGrande (WG) | Accuracy66.4 | 38 | |
| Language Modeling | Language Modeling Dataset | PPL10.16 | 31 | |
| Classification | QoE | Accuracy77.9 | 26 | |
| Regression | KVS | MSE2.41 | 26 |