atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

NOLA: Compressing LoRA using Linear Combination of Random Basis

ICLR 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: LoRA variants and theory · 1 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 NOLA, which reparameterizes LoRA's low-rank factors as linear combinations of frozen random basis matrices and trains only the mixture coefficients, breaking LoRA's rank-one parameter floor.

Problem

LoRA's parameter count is lower-bounded by rank one and is tied to layer shape and rank, limiting compression when storing many task-specific fine-tuned LLMs.

Approach

Re-parameterize LoRA's A and B matrices as linear combinations of pseudo-random fixed basis matrices (seeded), optimizing only the mixture coefficients; this decouples trainable parameter count from rank and layer dimensions, and allows coefficients to be quantized to 4 bits.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace LoRA rank 4 with NOLA QV using 0.036M parametersGPT-2 L · E2E NLG ChallengeBLEU: 70.4 -> 70.1No clear effect
Share the random basis across layers (ablation)NOLA QV · E2E NLG ChallengeBLEU: 70.46 -> 70.32No clear effect
Use rank 2 instead of rank 8 (ablation)NOLA QV on GPT-2 M · E2E NLG ChallengeBLEU: 70.03 -> 70.47Helped
Use rank 1 instead of rank 8 (ablation)NOLA QV on GPT-2 M · E2E NLG ChallengeBLEU: 70.03 -> 69.09Hurt
Reduce trainable parameters from 96K to 48K (ablation)NOLA QV on GPT-2 M · E2E NLG ChallengeBLEU: 70.03 -> 70.09No clear effect
Reduce trainable parameters from 96K to 24K (ablation)NOLA QV on GPT-2 M · E2E NLG ChallengeBLEU: 70.03 -> 68.30Hurt
Quantize coefficients to 4-bit after training (ablation)NOLA QV on GPT-2 L · E2E NLG ChallengeBLEU: 70.46 -> 70.29No clear effect
Use 3-bit quantization-aware training instead of full precision (ablation)NOLA QV on GPT-2 L · E2E NLG ChallengeBLEU: 70.46 -> 70.14No clear effect
Use 2-bit quantization-aware training (ablation)NOLA QV on GPT-2 L · E2E NLG ChallengeBLEU: 70.46 -> 68.69Hurt
Apply NOLA to MLP layers rather than QV attention projections (ablation)ViT-L NOLA · CIFAR-100 5-shotaccuracy: 74.8 -> 75.1Helped
Apply NOLA to MLP layers with 0.047M parameters instead of LoRA rank 4ViT-B · Food101 5-shotaccuracy: 60.8 (1.6) -> 64.5 (0.8)Helped
Use NOLA rather than PRANCResNet18 · ImageNet-100accuracy: 61.08% -> 64.66%Helped

Surprising findings

Limitations

Experimental setup

Datasets: E2E NLG Challenge DART WebNLG Alpaca MMLU CIFAR-10 CIFAR-100 CUB-200-2011 Caltech-101 Aircraft Food101 Pets

Benchmarks: E2E NLG Challenge DART WebNLG Alpaca MMLU CIFAR-10 CIFAR-100 CUB-200-2011 Caltech-101 Aircraft Food101 Pets

Metrics: BLEU NIST METEOR ROUGE-L CIDEr TER training loss validation loss MMLU accuracy top-1 accuracy

Baselines: LoRA Adapter PreLayer Full fine-tuning QLoRA PRANC HashedNet Linear probe Nearest neighbor

Builds on: LoRA PRANC QLoRA Straight-Through Estimator

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