Fusion of regional and sparse attention in Vision Transformers

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Ibtehaz, Nabil, Yan, Ning, Mortazavi, Masood, Kihara, Daisuke
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866911915915083776
author Ibtehaz, Nabil
Yan, Ning
Mortazavi, Masood
Kihara, Daisuke
author_facet Ibtehaz, Nabil
Yan, Ning
Mortazavi, Masood
Kihara, Daisuke
contents Modern vision transformers leverage visually inspired local interaction between pixels through attention computed within window or grid regions, in contrast to the global attention employed in the original ViT. Regional attention restricts pixel interactions within specific regions, while sparse attention disperses them across sparse grids. These differing approaches pose a challenge between maintaining hierarchical relationships vs. capturing a global context. In this study, drawing inspiration from atrous convolution, we propose Atrous Attention, a blend of regional and sparse attention that dynamically integrates both local and global information while preserving hierarchical structures. Based on this, we introduce a versatile, hybrid vision transformer backbone called ACC-ViT, tailored for standard vision tasks. Our compact model achieves approximately 84% accuracy on ImageNet-1K with fewer than 28.5 million parameters, outperforming the state-of-the-art MaxViT by 0.42% while requiring 8.4% fewer parameters.
format Preprint
id arxiv_https___arxiv_org_abs_2406_08859
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Fusion of regional and sparse attention in Vision Transformers
Ibtehaz, Nabil
Yan, Ning
Mortazavi, Masood
Kihara, Daisuke
Computer Vision and Pattern Recognition
Modern vision transformers leverage visually inspired local interaction between pixels through attention computed within window or grid regions, in contrast to the global attention employed in the original ViT. Regional attention restricts pixel interactions within specific regions, while sparse attention disperses them across sparse grids. These differing approaches pose a challenge between maintaining hierarchical relationships vs. capturing a global context. In this study, drawing inspiration from atrous convolution, we propose Atrous Attention, a blend of regional and sparse attention that dynamically integrates both local and global information while preserving hierarchical structures. Based on this, we introduce a versatile, hybrid vision transformer backbone called ACC-ViT, tailored for standard vision tasks. Our compact model achieves approximately 84% accuracy on ImageNet-1K with fewer than 28.5 million parameters, outperforming the state-of-the-art MaxViT by 0.42% while requiring 8.4% fewer parameters.
title Fusion of regional and sparse attention in Vision Transformers
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.08859