atlas

Research map / Label-efficient and robust learning / Zero-/few-shot and meta-learning

Quick-Tune: Quickly Learning Which Pretrained Model to Finetune and How

ICLR 2024 · Label-efficient and robust learning › Zero-/few-shot and meta-learning · approach: Normalization and unsupervised meta-learning · 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.

Quick-Tune jointly selects a pretrained image classifier and its finetuning hyperparameters via meta-learned gray-box Bayesian optimization with a cost-aware acquisition, using a 20k-configuration meta-dataset over 24 timm models and Meta-Album datasets.

Problem

Practitioners must choose which pretrained model to finetune from a growing model hub and how to set its finetuning hyperparameters, and trial-and-error is time-consuming and suboptimal.

Approach

Combine combined algorithm selection and hyperparameter optimization (CASH) for finetuning with gray-box learning-curve prediction, meta-learned deep-kernel GP and MLP cost estimator from a large meta-dataset of finetuning runs, and a cost-sensitive expected improvement acquisition.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Jointly search pretrained models and finetuning hyperparameters instead of using fixed default configurationsPretrained image classifiers · Meta-Album Micro, Mini, and Extendednormalized regret: BEiT+Default HP: 0.229_{\pm 0.081}, 0.281_{\pm 0.108}, 0.225_{ \pm 0.059 }; XCiT+Default HP: 0.223_{\pm 0.075}, 0.290_{\Helped
Use Quick-Tune CASH on a model hub instead of finetuning DINOv2 with LoRADINOv2 and Quick-Tune · Meta-Album Micro, Mini, and Extended at 4 and 24 hoursnormalized regret: QuickTune: 0.072_{\pm 0.024}, 0.039_{\pm 0.014}, 0.042_{\pm 0.016}; Dinov2 + LoRA: 0.541_{\pm 0.093}, 0.049_{\pm 0.018},Helped
Use Quick-Tune CASH on a model hub instead of DINOv2 linear probingDINOv2 and Quick-Tune · Meta-Album Micro, Mini, and Extended at 4 and 24 hoursnormalized regret: QuickTune: 0.072_{\pm 0.024}, 0.039_{\pm 0.014}, 0.042_{\pm 0.016}; Dinov2 + Linear Probing: 0.081_{\pm 0.041}, 0.067_{\Helped
Include meta-learning of the performance and cost predictors (ablation)Quick-Tune · Meta-Album Micro, Mini, and Extendednormalized regretHelped
Include cost-awareness in the acquisition function (ablation)Quick-Tune · Meta-Album Micro, Mini, and Extendednormalized regretHelped
Use a diverse hub of 24 or 15 models instead of a small or single-model pool (ablation)Quick-Tune · Meta-Album Micro, Mini, and Extendednormalized regretHelped
Jointly optimize model selection and hyperparameters rather than use a two-stage approachQuick-Tune · Meta-Albumnormalized regretHelped

Surprising findings

Limitations

Experimental setup

Datasets: Meta-Album micro Meta-Album mini Meta-Album extended Imagenette iNaturalist

Benchmarks: Meta-Album micro Meta-Album mini Meta-Album extended Imagenette iNaturalist

Metrics: normalized regret rank

Baselines: BEiT default HP XCiT default HP DLA default HP ASHA BOHB DEHB DyHPO Random Search LogME LEEP NCE DINOv2 + LoRA

Builds on: Expected Improvement deep-kernel Gaussian process DyHPO Meta-Album timm

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