atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

Efficient Adaptation of Pre-trained Vision Transformer via Householder Transformation

NeurIPS 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Efficient ViT adaptation · 3 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 HTA, a PEFT method that replaces SVD unitary matrices with Householder reflections built from single vectors, with learnable diagonal scaling, plus a rank-1 LoRA term, for adapting pre-trained ViTs and Swin Transformers.

Problem

Low-rank PEFT methods (LoRA, Adapter) use a fixed bottleneck dimensionality that cannot adapt to layer-wise variation in the needed rank, limiting flexibility when adapting pre-trained vision transformers.

Approach

View the adaptation matrix through SVD and replace the left and right unitary matrices with Householder matrices (each derived from one learnable vector), combined with a learnable diagonal scaling matrix per layer, so the effective rank can vary across layers; an additional rank-1 low-rank term is added for robustness.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use HTA instead of RLRRViT-B/16 · VTAB-1kMean Total accuracy: 74.5 -> 74.7Helped
Use HTA instead of RLRR*ViT-B/16 AugReg · VTAB-1kMean Total accuracy: 75.1 -> 75.7Helped
Use HTA instead of SSFViT-B/16 · FGVCMean Total accuracy: 90.7 -> 90.6Hurt
Use HTA* instead of RLRR*ViT-B/16 AugReg · FGVCMean Total accuracy: 91.0 -> 91.1Helped
Use HTA instead of RLRRViT-Large · VTAB-1kMean Total accuracy: 75.2 -> 75.4Helped
Use HTA instead of RLRRSwin-Base · VTAB-1kMean Total accuracy: 73.0 -> 74.2Helped
Replace LoRA with HTA on Wq and Wv (ablation)ViT-B/16 · VTAB-1kMean Total accuracy: 72.3 -> 72.7Helped
Add HTA on WFC1 and WFC2 to HTA on Wq and Wv (ablation)ViT-B/16 · VTAB-1kMean Total accuracy: 72.7 -> 73.9Helped
Replace Adapter on WFC1 and WFC2 with HTA (ablation)ViT-B/16 · VTAB-1kMean Total accuracy: 71.4 -> 73.0Helped
Add rank-1 low-rank adaptation to HTA (ablation)ViT-B/16 · VTAB-1kMean Total accuracy: 72.0 -> 74.7Helped
Increase additive adaptation dimension from 1 to 2 (ablation)HTA · VTAB-1kMean Total accuracy: 74.7 -> 74.5Hurt
Increase additive adaptation dimension from 1 to 4 (ablation)HTA · VTAB-1kMean Total accuracy: 74.7 -> 75.0Helped

Surprising findings

Limitations

Experimental setup

Datasets: VTAB-1k CUB-200-2011 NABirds Oxford Flowers Stanford Dogs Stanford Cars ImageNet-21K

Benchmarks: VTAB-1k FGVC

Metrics: top-1 accuracy mean accuracy trainable parameters (M)

Baselines: Full fine-tuning Linear probing Bias VPT LoRA Adapter AdaptFormer FacT ARC RLRR SSF

Builds on: LoRA Adapter AdaptFormer FacT ARC

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