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
3089
Image ClassificationCIFAR-10--
973
Image ClassificationImageNet V2
Top-1 Acc76.91
767
Semantic segmentationADE20K
mIoU50.69
699
Semantic segmentationCityscapes
mIoU69.54
526
Semantic segmentationCOCO Stuff
mIoU47.92
421
Image ClassificationCIFAR-100--
375
Fine-grained Image ClassificationCUB-200 2011
Accuracy88.02
317
Semantic segmentationPascal VOC
mIoU0.8707
295
Image ClassificationImageNet-ReaL
Precision@189.71
287
Showing 10 of 32 rows

Other info

Follow for update