Learning to Adapt to Position Bias in Vision Transformer Classifiers

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bruintjes, Robert-Jan, van Gemert, Jan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910953510010880
author Bruintjes, Robert-Jan
van Gemert, Jan
author_facet Bruintjes, Robert-Jan
van Gemert, Jan
contents How discriminative position information is for image classification depends on the data. On the one hand, the camera position is arbitrary and objects can appear anywhere in the image, arguing for translation invariance. At the same time, position information is key for exploiting capture/center bias, and scene layout, e.g.: the sky is up. We show that position bias, the level to which a dataset is more easily solved when positional information on input features is used, plays a crucial role in the performance of Vision Transformers image classifiers. To investigate, we propose Position-SHAP, a direct measure of position bias by extending SHAP to work with position embeddings. We show various levels of position bias in different datasets, and find that the optimal choice of position embedding depends on the position bias apparent in the dataset. We therefore propose Auto-PE, a single-parameter position embedding extension, which allows the position embedding to modulate its norm, enabling the unlearning of position information. Auto-PE combines with existing PEs to match or improve accuracy on classification datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2505_13137
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning to Adapt to Position Bias in Vision Transformer Classifiers
Bruintjes, Robert-Jan
van Gemert, Jan
Computer Vision and Pattern Recognition
How discriminative position information is for image classification depends on the data. On the one hand, the camera position is arbitrary and objects can appear anywhere in the image, arguing for translation invariance. At the same time, position information is key for exploiting capture/center bias, and scene layout, e.g.: the sky is up. We show that position bias, the level to which a dataset is more easily solved when positional information on input features is used, plays a crucial role in the performance of Vision Transformers image classifiers. To investigate, we propose Position-SHAP, a direct measure of position bias by extending SHAP to work with position embeddings. We show various levels of position bias in different datasets, and find that the optimal choice of position embedding depends on the position bias apparent in the dataset. We therefore propose Auto-PE, a single-parameter position embedding extension, which allows the position embedding to modulate its norm, enabling the unlearning of position information. Auto-PE combines with existing PEs to match or improve accuracy on classification datasets.
title Learning to Adapt to Position Bias in Vision Transformer Classifiers
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2505.13137