atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

VB-LoRA: Extreme Parameter Efficient Fine-Tuning with Vector Banks

NeurIPS 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 VB-LoRA, a PEFT method that reparameterizes LoRA's rank-one components into sub-vectors composed from a single globally shared vector bank via a differentiable top-k admixture, achieving extreme parameter efficiency.

Problem

LoRA and its variants still incur substantial storage and transmission costs when customizing LLMs per user or task, and existing sharing schemes (LoRA local, tied/VeRA across layers) do not share parameters across matrix dimensions, modules and layers at fine granularity.

Approach

Divide-and-share: LoRA's rank-one vectors are split into equal-size sub-vectors, and each sub-vector is a sparse top-k softmax combination of vectors from one global vector bank shared across modules and layers; the logits are learned end-to-end, and only the bank, top-k indices and weights are stored.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Top-1 instead of Top-2 selection (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.3±0.7 -> 66.9±0.5Hurt
Use Top-3 instead of Top-2 selection (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.3±0.7 -> 68.1±1.3Hurt
Use Top-6 instead of Top-2 selection (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.3±0.7 -> 67.1±0.5Hurt
Select all vector-bank entries instead of Top-2 (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.3±0.7 -> 67.5±1.2Hurt
Use Noisy Top-1 instead of noise-free Top-2 selection (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.3±0.7 -> 45.3±2.2Hurt
Use Noisy Top-2 instead of noise-free Top-2 selection (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.3±0.7 -> 62.6±0.2Hurt
Use Gumbel-Softmax (τ=1/3) during training, Top-1 at inference (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.3±0.7 -> 57.1±0.6Hurt
Use Gumbel-Softmax (τ=1/3) during training, Top-2 at inference (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.3±0.7 -> 57.3±1.6Hurt
Use Straight-Through Gumbel-Softmax (τ=1/3) during training, Top-1 at inference (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.3±0.7 -> 55.6±1.6Hurt
Use Straight-Through Gumbel-Softmax (τ=1/3) during training, Top-2 at inference (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.3±0.7 -> 54.7±1.2Hurt
Set sub-vector length b=128 (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.7±0.7 -> 67.0±0.8Hurt
Set sub-vector length b=512 (ablation)VB-LoRA · CoLA, RoBERTa-largeMatthew's correlation: 68.7±0.7 -> 67.8±0.8Hurt

Surprising findings

Limitations

Experimental setup

Datasets: GLUE E2E Cleaned Alpaca MetaMathQA

Benchmarks: GLUE E2E MT-Bench GSM8K MATH

Metrics: accuracy Matthews correlation Pearson correlation BLEU NIST METEOR ROUGE-L CIDEr MT-Bench score stored parameter count

Baselines: LoRA VeRA Tied-LoRA LoRA-XS full fine-tuning Noisy Top-k Gumbel-Softmax Straight-Through Gumbel-Softmax

Builds on: LoRA VeRA Tied-LoRA Noisy Top-k gating QLoRA

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