Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

AutoVP: An Automated Visual Prompting Framework and Benchmark

About

Visual prompting (VP) is an emerging parameter-efficient fine-tuning approach to adapting pre-trained vision models to solve various downstream image-classification tasks. However, there has hitherto been little systematic study of the design space of VP and no clear benchmark for evaluating its performance. To bridge this gap, we propose AutoVP, an end-to-end expandable framework for automating VP design choices, along with 12 downstream image-classification tasks that can serve as a holistic VP-performance benchmark. Our design space covers 1) the joint optimization of the prompts; 2) the selection of pre-trained models, including image classifiers and text-image encoders; and 3) model output mapping strategies, including nonparametric and trainable label mapping. Our extensive experimental results show that AutoVP outperforms the best-known current VP methods by a substantial margin, having up to 6.7% improvement in accuracy; and attains a maximum performance increase of 27.5% compared to linear-probing (LP) baseline. AutoVP thus makes a two-fold contribution: serving both as an efficient tool for hyperparameter tuning on VP design choices, and as a comprehensive benchmark that can reasonably be expected to accelerate VP's development. The source code is available at https://github.com/IBM/AutoVP.

Hsi-Ai Tsao, Lei Hsiung, Pin-Yu Chen, Sijia Liu, Tsung-Yi Ho• 2023

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-10
Accuracy95.45
875
Image ClassificationTiny ImageNet (test)
Accuracy84.81
722
Image ClassificationCIFAR-100 (test)--
395
Image ClassificationCIFAR-100
Accuracy85.96
357
Image ClassificationTiny-ImageNet
Accuracy (%)82.43
131
Fine-grained Image ClassificationOxford Pets--
20
Fine-grained Image ClassificationFlower102
Accuracy98.42
19
Texture ClassificationDTD
Accuracy70.81
12
Fine-grained Image ClassificationFood-101
Accuracy82.86
10
Remote Sensing ClassificationEuroSAT--
10
Showing 10 of 10 rows

Other info

Follow for update