Research map / Label-efficient and robust learning / Domain adaptation and generalization
Structured Model Probing: Empowering Efficient Transfer Learning by Structured Regularization
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 structured model probing (SMP), which uses structured group-sparsity and non-linearity regularization over multi-layer ViT features to make probing adapt well to both easy and difficult transfer tasks on VTAB-1k.
Problem
Model probing (frozen pre-trained features with a light head) lags behind parameter-efficient tuning on VTAB-1k: linear probing works for domains similar to the source but fails for dissimilar ones, while more complex probes hurt easy domains.
Approach
Extract structured features from all ViT sub-structures (token- and channel-pooled), apply a group-sparsity regularizer whose groups follow the model structure, and couple a linear model with an MLP whose complexity is bounded by the selected-feature norms, so the probe automatically becomes linear for easy tasks and non-linear for difficult ones.
Key results
- SMP reaches 70.9% mean accuracy on VTAB-1k, versus 69.4% for VPT and 67.5% for Head2Toe.
- On the structured (difficult) VTAB group SMP attains 55.4% mean accuracy, against 55.0% for VPT and 45.3% for LinearStruc.
- SMP needs about 15% of fine-tuning adaptation time on average, while VPT needs 97%, and adds only 0.01G inference FLOPs versus VPT's 11.01G.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| apply SMP instead of VPT | ViT-B/16 · VTAB-1k Mean (All) | accuracy: 69.4 -> 70.9 | Helped |
| apply SMP instead of Head2Toe | ViT-B/16 · VTAB-1k Mean (All) | accuracy: 67.5 -> 70.9 | Helped |
| apply SMP instead of linear probing | ViT-B/16 · VTAB-1k Structured Mean | accuracy: 28.6 -> 55.4 | Helped |
| add structured non-linearity regularization (ablation) | structured feature probing with a linear and MLP model · VTAB-1k Mean (All) | accuracy: 69.0 -> 70.9 | Helped |
| add structured non-linearity regularization (ablation) | structured feature probing with a linear and MLP model · VTAB-1k Natural Mean | accuracy: 76.5 -> 80.7 | Helped |
| add structured non-linearity regularization (ablation) | structured feature probing with a linear and MLP model · VTAB-1k Structured Mean | accuracy: 55.0 -> 55.4 | Helped |
| replace ordinary feature sparsity regularization with structured sparsity regularization (ablation) | LinearStruc. probing · VTAB-1k Mean (All) | accuracy: 65.7 -> 66.6 | Helped |
| add feature sparsity regularization (ablation) | LinearStruc. probing · VTAB-1k Mean (All) | accuracy: 64.8 -> 65.7 | Helped |
| add structured sparsity regularization (ablation) | LinearStruc. probing · VTAB-1k Mean (All) | accuracy: 64.8 -> 66.6 | Helped |
| add non-linear transformation to structured features (ablation) | structured feature probing · VTAB-1k Structured Mean | accuracy: 45.3 -> 54.7 | Helped |
| add non-linear transformation to structured features (ablation) | structured feature probing · VTAB-1k Natural Mean | accuracy: 76.0 -> 72.8 | Hurt |
| add non-linear transformation to structured features (ablation) | structured feature probing · VTAB-1k Specialized Mean | accuracy: 84.3 -> 82.0 | Hurt |
Surprising findings
- Linear probing on the last layer is comparable to tuning methods on easy VTAB tasks, but a non-linear MLP probe degrades performance on easy tasks through overfitting.
- Adding non-linearity or more layer features helps difficult tasks but hurts easy ones, so no single fixed probe complexity works for both.
Limitations
- SMP requires 6.91M extra parameters on average, more than VPT's 0.64M.
- Gains over VPT on the mean are modest (70.9% vs 69.4%).
- Evaluation focuses on ViT-B/16 with ImageNet-21k pre-training and VTAB-1k, with limited other settings.
Experimental setup
Datasets: VTAB-1k ImageNet-21k Food101 CUB-200-2011 Stanford Cars Stanford Dogs Oxford Pets
Benchmarks: VTAB-1k Few-shot fine-grained recognition (Food101, CUB-200-2011, Stanford Cars, Stanford Dogs, Oxford Pets) 9 full-size downstream datasets
Metrics: median test accuracy over 3 seeds mean accuracy relative adaptation time extra inference FLOPs extra parameters
Baselines: Fine-tuning Bias VPT Linear Probing Side-Tuning Head2Toe LinearStruc.
Builds on: group lasso Head2Toe VPT Side-Tuning hierarchical sparse modeling
Closest papers by meaning
- Fantastic Features and Where to Find Them: A Probing Method to combine Features from Multiple Foundation Models (NeurIPS 2025 · similarity 0.84)
- RAPTOR: Ridge-Adaptive Logistic Probes (NeurIPS 2026 · similarity 0.83)
- Probe Pruning: Accelerating LLMs through Dynamic Pruning via Model-Probing (ICLR 2025 · similarity 0.83)
- Transferring Knowledge From Large Foundation Models to Small Downstream Models (ICML 2024 · similarity 0.83)
- Transferring Linear Features Across Language Models With Model Stitching (NeurIPS 2025 · similarity 0.82)
- Exploring Structured Semantic Priors Underlying Diffusion Score for Test-time Adaptation (NeurIPS 2024 · similarity 0.82)
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.
