atlas

Research map / Architectures and efficiency / Quantization

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

AAAI 2024 · Architectures and efficiency › Quantization · approach: Post-training quantization of LLMs · 72 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes OWQ, a sensitivity-aware mixed-precision weight quantization that keeps a small set of outlier-sensitive weight columns in fp16 and quantizes the rest with OPTQ, plus Weak Column Tuning (WCT) for parameter-efficient fine-tuning of the quantized model.

Problem

Post-training weight quantization of LLMs to ~3 bits loses substantial accuracy, and quantized-base PEFT (e.g., QLoRA) is hampered by low-quality quantized dense weights; activation outliers make certain weight columns highly sensitive to quantization.

Approach

Select weight columns by a Hessian-weighted perturbation sensitivity (lambda_j ||ΔW_:,j||^2), keep those weak columns in fp16 while quantizing the rest with OPTQ plus grid-searched truncated ranges, and fine-tune only the retained fp16 weak columns (WCT) on top of the frozen low-bit matrix.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use OWQ 3.01-bit instead of OPTQ 3-bitOPT-6.7B · WikiText-2perplexity: 12.88 -> 11.21Helped
use OWQ 3.01-bit instead of OPTQ 3-bitLLaMA 7B · WikiText-2perplexity: 8.13 -> 6.66Helped
use OWQ 3.1-bit instead of OPTQ 3-bitLLaMA 7B · few-shot average of ARC-challenge, Hellaswag, and MMLUscore (%): 47.06 -> 51.93Helped
fine-tune 64 weak columns with WCT instead of QLoRALLaMA 7B · Vicuna BenchmarkGPT-4 preference wins: 54 -> 81Helped
apply truncation without protecting weak columns (ablation)OPTQ on OPT-6.7B · WikiText-2perplexity: 12.88 -> 48.26Hurt
add OWQ weak-column protection and tuned quantization configuration (ablation)OPT-6.7B quantization · WikiText-2perplexity: 12.14 -> 11.21Helped
add group-wise quantization with group size 128 (ablation)OWQ 3.01-bit on OPT-6.7B · WikiText-2perplexity: 11.22 -> 11.16No clear effect
add group-wise quantization with group size 128 (ablation)OWQ 3.01-bit on LLaMA 7B · WikiText-2perplexity: 6.65 -> 6.40Helped
increase weak-column bit budget from 3.005 to 3.30 effective bits (ablation)OWQ on OPT-6.7B · WikiText-2perplexity: 11.38 -> 11.13Helped
increase weak-column bit budget from 3.005 to 3.30 effective bits (ablation)OWQ on OPT-6.7B · latency overheadlatency overhead (%): 2.89 -> 8.69Hurt
increase weak-column bit budget from 3.005 to 3.30 effective bits (ablation)OWQ on OPT-6.7B · memory overheadmemory overhead (%): 0.21 -> 10.95Hurt
select weak columns using Hessian sensitivity and weight perturbation rather than weight error alone (ablation)OWQ on OPT-6.7B · WikiText-2perplexity: 14.77 -> 11.23Helped

Surprising findings

Limitations

Experimental setup

Datasets: C4 WikiText-2 Penn Treebank OpenAssistant Vicuna Benchmark LAMBADA Winogrande ARC-challenge HellaSwag MMLU

Benchmarks: WikiText-2 Penn Treebank C4 ARC-challenge HellaSwag MMLU LAMBADA Winogrande Vicuna Benchmark

Metrics: perplexity accuracy GPT-4 preference win count latency overhead memory overhead

Baselines: RTN OPTQ QLoRA LoRA AWQ-style outlier-magnitude methods

Builds on: OPTQ Optimal Brain Compression QLoRA LoRA

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy