Saved in:
Bibliographic Details
Main Authors: Tsao, Hsi-Ai, Hsiung, Lei, Chen, Pin-Yu, Liu, Sijia, Ho, Tsung-Yi
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2310.08381
Tags: Add Tag
No Tags, Be the first to tag this record!
_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