ADAPT to Robustify Prompt Tuning Vision Transformers

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Eskandar, Masih, Imtiaz, Tooba, Wang, Zifeng, Dy, Jennifer
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917915742175232
author Eskandar, Masih
Imtiaz, Tooba
Wang, Zifeng
Dy, Jennifer
author_facet Eskandar, Masih
Imtiaz, Tooba
Wang, Zifeng
Dy, Jennifer
contents The performance of deep models, including Vision Transformers, is known to be vulnerable to adversarial attacks. Many existing defenses against these attacks, such as adversarial training, rely on full-model fine-tuning to induce robustness in the models. These defenses require storing a copy of the entire model, that can have billions of parameters, for each task. At the same time, parameter-efficient prompt tuning is used to adapt large transformer-based models to downstream tasks without the need to save large copies. In this paper, we examine parameter-efficient prompt tuning of Vision Transformers for downstream tasks under the lens of robustness. We show that previous adversarial defense methods, when applied to the prompt tuning paradigm, suffer from gradient obfuscation and are vulnerable to adaptive attacks. We introduce ADAPT, a novel framework for performing adaptive adversarial training in the prompt tuning paradigm. Our method achieves competitive robust accuracy of ~40% w.r.t. SOTA robustness methods using full-model fine-tuning, by tuning only ~1% of the number of parameters.
format Preprint
id arxiv_https___arxiv_org_abs_2403_13196
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ADAPT to Robustify Prompt Tuning Vision Transformers
Eskandar, Masih
Imtiaz, Tooba
Wang, Zifeng
Dy, Jennifer
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
The performance of deep models, including Vision Transformers, is known to be vulnerable to adversarial attacks. Many existing defenses against these attacks, such as adversarial training, rely on full-model fine-tuning to induce robustness in the models. These defenses require storing a copy of the entire model, that can have billions of parameters, for each task. At the same time, parameter-efficient prompt tuning is used to adapt large transformer-based models to downstream tasks without the need to save large copies. In this paper, we examine parameter-efficient prompt tuning of Vision Transformers for downstream tasks under the lens of robustness. We show that previous adversarial defense methods, when applied to the prompt tuning paradigm, suffer from gradient obfuscation and are vulnerable to adaptive attacks. We introduce ADAPT, a novel framework for performing adaptive adversarial training in the prompt tuning paradigm. Our method achieves competitive robust accuracy of ~40% w.r.t. SOTA robustness methods using full-model fine-tuning, by tuning only ~1% of the number of parameters.
title ADAPT to Robustify Prompt Tuning Vision Transformers
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.13196