atlas

Research map / Large language models / Instruction and fine-tuning

LoftQ: LoRA-Fine-Tuning-aware Quantization for Large Language Models

ICLR 2024 · Large language models › Instruction and fine-tuning · approach: Low-rank adaptation variants · 18 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 LoftQ, which alternates quantization and SVD-based low-rank approximation to initialize LoRA adapters for quantized pre-trained LLMs and encoder/encoder-decoder models, closing the gap to full fine-tuning especially at 2-bit.

Problem

Quantizing a pre-trained model and then applying LoRA fine-tuning yields a gap to full fine-tuning; QLoRA's zero-initialized adapters on a quantized backbone start from a discrepant point, which degrades or breaks fine-tuning at low bit-widths (below 3 bits).

Approach

Jointly approximate the pre-trained weight by a quantized backbone plus low-rank adapters via alternating optimization: quantize the residual W minus the current low-rank term, then set the adapters to the rank-r SVD of the new quantization residual; the result initializes LoRA fine-tuning on the frozen quantized backbone.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use LoftQ initialization instead of QLoRA initializationDeBERTaV3-base, 2-bit NF2, rank 16 · MNLI developmentaccuracy (m / mm): 75.4/75.6 -> 84.7/85.1Helped
Use LoftQ initialization instead of QLoRA initializationDeBERTaV3-base, 2-bit uniform, rank 32 · MNLI developmentaccuracy (m / mm): 79.9/79.5 -> 88.0/88.1Helped
Use LoftQ initialization instead of QLoRA initializationBART-large, 4-bit NF4, rank 8 · XSum testROUGE-1/2/L: 42.91/19.72/34.82 -> 44.08/20.72/35.89Helped
Use LoftQ initialization instead of QLoRA initializationBART-large, 4-bit NF4, rank 8 · CNN/DailyMail testROUGE-1/2/L: 43.10/20.22/40.06 -> 43.81/20.95/40.84Helped
Use LoftQ initialization instead of QLoRA initializationBART-large, 2-bit NF2, rank 8 · XSum testROUGE-1/2/LHelped
Use LoftQ initialization instead of QLoRA initializationLLAMA-2-7b, 2-bit NF2 · WikiText-2perplexityHelped
Use LoftQ initialization instead of QLoRA initializationLLAMA-2-13b, 4-bit NF4 · GSM8Kaccuracy: 39.9 -> 45.0Helped
Use mixed precision with 4-bit quantization in the first 4 layers and 2-bit quantization in the restLoftQ on LLAMA-2-13b · GSM8KaccuracyHelped
Perform SVD before quantization in each alternating step rather than quantization before SVD (ablation)LoftQ, DeBERTaV3-base, rank 32 · MNLI development, 2-bit uniformaccuracy (m / mm): 88.0/88.1 -> 87.8/87.7Hurt
Add weight decay to full-precision LoRA fine-tuning (ablation)LLAMA-2-13b · GSM8Kaccuracy: 43.1 -> 45.3Helped
Add weight decay to full-precision LoRA fine-tuning (ablation)LLAMA-2-7b · GSM8Kaccuracy: 36.9 -> 34.4Hurt

Surprising findings

Limitations

Experimental setup

Datasets: MNLI QNLI RTE SST-2 MRPC CoLA QQP STS-B SQuAD v1.1 ANLI XSum CNN/DailyMail

Benchmarks: GLUE SQuAD v1.1 ANLI XSum CNN/DailyMail GSM8K WikiText-2

Metrics: accuracy Matthews correlation exact match / F1 ROUGE-1/2/L perplexity

Baselines: Full fine-tuning LoRA QLoRA LoSparse

Builds on: QLoRA LoRA NormalFloat quantization HuggingFace Transformers

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