atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

Expanding Sparse Tuning for Low Memory Usage

NeurIPS 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Adapter-based parameter-efficient tuning · 0 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 SNELL, a sparse-tuning PEFT method that sparsifies a kernelized low-rank (LoRA-style) adaptation matrix via competition-based thresholding, cutting memory while matching or beating prior sparse and LoRA methods on VTAB-1k and FGVC.

Problem

Sparse tuning for PEFT achieves strong accuracy but has high memory usage because the whole weight matrix is stored as optimizer parameters and tunable weight indexes must be stored, making it no cheaper than full fine-tuning on large models.

Approach

Decompose the sparsified adaptation matrix into two learnable low-rank matrices, merge them with nonlinear kernel functions (instead of an inner product) to raise the merged matrix's rank, and sparsify it with a competition-based dynamic threshold on absolute values so no binary mask needs storing.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace the LoRA inner product with a piecewise linear kernel (ablation)LoRA on VTAB-1k · VTAB-1k mean accuracy, ranks 8, 16, and 32mean accuracy: 75.1 -> 75.6; 74.9 -> 75.4; 74.2 -> 75.2Helped
Use Sigmoid or RBF kernels instead of the piecewise linear kernel (ablation)Kernelized LoRA fine-tuning · VTAB-1kperformanceHurt
Add sparse tuning to kernelized LoRA (ablation)KLoRA on VTAB-1k · VTAB-1k mean accuracy, ranks 8, 16, and 32mean accuracy: +1.0; +1.4; +1.9Helped
Increase learnable-matrix rank (ablation)KLoRA on VTAB-1k · VTAB-1k mean accuracy, ranks 8 to 32mean accuracy: 73.2 -> 72.7Hurt
Increase learnable-matrix rank (ablation)SNELL on VTAB-1k · VTAB-1k mean accuracy, ranks 8 to 32mean accuracy: 74.2 -> 74.6Helped
Use a LoRA merged adaptation matrix rather than a full-rank matrix for sparsification (ablation)Sparse tuning on VTAB-1k · VTAB-1k mean accuracy, s=0.9mean accuracy: 74.4 -> 65.7Hurt
Replace LoRA with kernelized LoRA before sparsification (ablation)Sparse tuning on VTAB-1k · VTAB-1k mean accuracy, s=0.9mean accuracy: 65.7 -> 73.9Helped
Use predefined fixed weight masks instead of competition-based dynamic masks (ablation)KLoRA-8 on FGVC · FGVC mean accuracymean accuracy: 90.3 -> 89.4Hurt
Replace the inner product with a piecewise linear kernel (ablation)LoRA on VTAB-1k · Natural, Specialized, and Structured groupsaccuracy: Natural: 80.8 -> 80.8; Specialized: 84.9 -> 85.5; Structured: 59.6 -> 60.5Mixed
Use SNELL sparse tuning with kernelized low-rank adaptationViT-B/16 on FGVC · FGVC mean accuracymean accuracy: 90.7 -> 91.8Helped
Use SNELL instead of SPT-LoRAViT-B/16 on VTAB-1k · VTAB-1k mean accuracymean accuracy: 76.4 -> 76.9Helped
Use SNELL instead of LoRA-8ViT-L/16 on VTAB-1k · VTAB-1k mean accuracymean accuracy: 73.7 -> 75.3Helped

Surprising findings

Limitations

Experimental setup

Datasets: FGVC (CUB-200-2011, NABirds, Oxford Flowers, Stanford Dogs, Stanford Cars) VTAB-1k (19 tasks) Commonsense reasoning benchmark (BoolQ, PIQA, SIQA, HellaSwag, WinoGrande, ARC, OBQA)

Benchmarks: FGVC VTAB-1k commonsense reasoning (LLaMA-2-7B)

Metrics: top-1 accuracy mean accuracy peak memory usage (MB) training time (s/img)

Baselines: Full fine-tuning LoRA-8 LoRA-16 SPT-LoRA SPT-Adapter VPT-Deep VPT-Shallow Adapter-8 Adapter-32 SSF Bias Partial-1

Builds on: LoRA SPT DyN Sparse tuning (binary mask) VPT

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