AutoVP: An Automated Visual Prompting Framework and Benchmark

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Tsao, Hsi-Ai, Hsiung, Lei, Chen, Pin-Yu, Liu, Sijia, Ho, Tsung-Yi
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914708631584768
author Tsao, Hsi-Ai
Hsiung, Lei
Chen, Pin-Yu
Liu, Sijia
Ho, Tsung-Yi
author_facet Tsao, Hsi-Ai
Hsiung, Lei
Chen, Pin-Yu
Liu, Sijia
Ho, Tsung-Yi
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2310_08381
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle AutoVP: An Automated Visual Prompting Framework and Benchmark
Tsao, Hsi-Ai
Hsiung, Lei
Chen, Pin-Yu
Liu, Sijia
Ho, Tsung-Yi
Computer Vision and Pattern Recognition
Machine Learning
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.
title AutoVP: An Automated Visual Prompting Framework and Benchmark
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2310.08381