atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

PiSSA: Principal Singular Values and Singular Vectors Adaptation of Large Language Models

NeurIPS 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: LoRA variants and theory · 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 PiSSA, a PEFT method that initializes low-rank adapters from the principal singular values/vectors of the pretrained weight and freezes the residual, plus QPiSSA with 4-bit quantized residual, outperforming LoRA/QLoRA on NLG and NLU tasks.

Problem

LoRA initializes adapters with Gaussian noise and zeros, so gradients are small or uninformative early on, causing slow convergence and suboptimal fits; QLoRA also leaves quantization error unreduced.

Approach

Apply SVD to the pretrained weight W, initialize trainable adapters A,B from the top-r principal singular components, and freeze the residual W^res built from the remaining components; quantizing only W^res reduces quantization error relative to QLoRA.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
initialize trainable adapters from principal singular values and vectors instead of Gaussian/zero initializationLoRA fine-tuning · GSM8K, MATH, HumanEval, MBPP, and MT-Benchbenchmark scoreHelped
use principal instead of medium or minor singular values and vectors to initialize adapters (ablation)PiSSA · GSM8K and MATHaccuracyHelped
increase PiSSA rank from 64 to 128 (ablation)PiSSA · Quantization error reduction ratio (AVG), LLaMA 3-70Bquantization error reduction ratio: 13.4 -> 20.3Helped
increase QPiSSA SVD iterations from 1 to 5 (ablation)QPiSSA rank 64 · Quantization error reduction ratio (AVG), LLaMA 3-8Bquantization error reduction ratio: 7.7 -> 14.3Helped
increase Fast SVD niter from 1 to 16 at rank 8 (ablation)PiSSA Fast SVD initialization · Training settraining loss: 0.3044 -> 0.2946Helped
replace exact SVD with Fast SVD (ablation)PiSSA initialization · Training settraining loss: 0.2945; 0.2946No clear effect
train in FP32 instead of BF16 (ablation)full fine-tuning · GSM8K, LLaMA-2-7Baccuracy: 63.15 -> 68.31Helped
train in FP32 instead of BF16 (ablation)full fine-tuning · GSM8K, Mistral-7B and LLaMA-3-8Baccuracy: 73.09 -> 65.88; 81.96 -> 75.44Hurt
combine PiSSA with AdaLoRA improvements (ablation)PiSSA · GSM8K, LLaMA-3-8Baccuracy: 76.75±0.036 -> 78.59±0.199Helped
combine PiSSA with DoRA (ablation)PiSSA · GSM8K, LLaMA-3-8Baccuracy: 76.75±0.036 -> 77.51±0.257Helped
quantize the residual matrix to 4-bit instead of quantizing the full base modelQLoRA · GSM8K, LLaMA-3-70Baccuracy: 81.73% -> 86.05%Helped
increase QPiSSA SVD iterations from 1 to 5 (ablation)QPiSSA rank 64 · Quantization error reduction ratio (AVG), LLaMA-3-70Bquantization error reduction ratio: 8.6 -> 13.4Helped

Surprising findings

Limitations

Experimental setup

Datasets: MetaMathQA CodeFeedback WizardLM-Evol-Instruct GLUE GSM8K MATH HumanEval MBPP MT-Bench

Benchmarks: GSM8K MATH HumanEval MBPP MT-Bench GLUE

Metrics: accuracy average score quantization error reduction ratio training loss

Baselines: LoRA QLoRA LoftQ Full FT AdaLoRA DoRA BitFit HAdapter PAdapter

Builds on: LoRA QLoRA LoftQ AdaLoRA DoRA

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