Robustness Tokens: Towards Adversarial Robustness of Transformers

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Pulfer, Brian, Belousov, Yury, Voloshynovskiy, Slava
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916651712118784
author Pulfer, Brian
Belousov, Yury
Voloshynovskiy, Slava
author_facet Pulfer, Brian
Belousov, Yury
Voloshynovskiy, Slava
contents Recently, large pre-trained foundation models have become widely adopted by machine learning practitioners for a multitude of tasks. Given that such models are publicly available, relying on their use as backbone models for downstream tasks might result in high vulnerability to adversarial attacks crafted with the same public model. In this work, we propose Robustness Tokens, a novel approach specific to the transformer architecture that fine-tunes a few additional private tokens with low computational requirements instead of tuning model parameters as done in traditional adversarial training. We show that Robustness Tokens make Vision Transformer models significantly more robust to white-box adversarial attacks while also retaining the original downstream performances.
format Preprint
id arxiv_https___arxiv_org_abs_2503_10191
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Robustness Tokens: Towards Adversarial Robustness of Transformers
Pulfer, Brian
Belousov, Yury
Voloshynovskiy, Slava
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Recently, large pre-trained foundation models have become widely adopted by machine learning practitioners for a multitude of tasks. Given that such models are publicly available, relying on their use as backbone models for downstream tasks might result in high vulnerability to adversarial attacks crafted with the same public model. In this work, we propose Robustness Tokens, a novel approach specific to the transformer architecture that fine-tunes a few additional private tokens with low computational requirements instead of tuning model parameters as done in traditional adversarial training. We show that Robustness Tokens make Vision Transformer models significantly more robust to white-box adversarial attacks while also retaining the original downstream performances.
title Robustness Tokens: Towards Adversarial Robustness of Transformers
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.10191