Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Elastic Attention Cores for Scalable Vision Transformers

About

Vision Transformers (ViTs) achieve strong data-driven scaling by leveraging all-to-all self-attention. However, this flexibility incurs a computational cost that scales quadratically with image resolution, limiting ViTs in high-resolution domains. Underlying this approach is the assumption that pairwise token interactions are necessary for learning rich visual-semantic representations. In this work, we challenge this assumption, demonstrating that effective visual representations can be learned without any direct patch-to-patch interaction. We propose VECA (Visual Elastic Core Attention), a vision transformer architecture that uses efficient linear-time core-periphery structured attention enabled by a small set of learned cores. In VECA, these cores act as a communication interface: patch tokens exchange information exclusively through the core tokens, which are initialized from scratch and propagated across layers. Because the $N$ image patches only directly interact with a resolution invariant set of $C$ learned "core" embeddings, this yields linear complexity $O(N)$ for predetermined $C$, which bypasses quadratic scaling. Compared to prior cross-attention architectures, VECA maintains and iteratively updates the full set of $N$ input tokens, avoiding a small $C$-way bottleneck. Combined with nested training along the core axis, our model can elastically trade off compute and accuracy during inference. Across classification and dense tasks, VECA achieves performance competitive with the latest vision foundation models while reducing computational cost. Our results establish elastic core-periphery attention as a scalable alternative building block for Vision Transformers.

Alan Z. Song, Yinjie Chen, Mu Nan, Rui Zhang, Jiahang Cao, Weijian Mai, Muquan Yu, Hossein Adeli, Deva Ramanan, Michael J. Tarr, Andrew F. Luo• 2026

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K (val)
mIoU55.54
3069
Image ClassificationCIFAR-10--
875
Image ClassificationImageNet V2
Top-1 Acc76.91
749
Semantic segmentationADE20K
mIoU50.69
559
Semantic segmentationCityscapes
mIoU69.54
494
Semantic segmentationCOCO Stuff
mIoU47.92
399
Image ClassificationCIFAR-100--
357
Fine-grained Image ClassificationCUB-200 2011
Accuracy88.02
314
Semantic segmentationPascal VOC
mIoU0.8707
280
Image ClassificationImageNet-ReaL
Precision@189.71
275
Showing 10 of 32 rows

Other info

Follow for update