English

Elastic Attention Cores for Scalable Vision Transformers

Computer Vision and Pattern Recognition 2026-05-13 v1 Machine Learning

Abstract

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 NN image patches only directly interact with a resolution invariant set of CC learned "core" embeddings, this yields linear complexity O(N)O(N) for predetermined CC, which bypasses quadratic scaling. Compared to prior cross-attention architectures, VECA maintains and iteratively updates the full set of NN input tokens, avoiding a small CC-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.

Keywords

Cite

@article{arxiv.2605.12491,
  title  = {Elastic Attention Cores for Scalable Vision Transformers},
  author = {Alan Z. Song and Yinjie Chen and Mu Nan and Rui Zhang and Jiahang Cao and Weijian Mai and Muquan Yu and Hossein Adeli and Deva Ramanan and Michael J. Tarr and Andrew F. Luo},
  journal= {arXiv preprint arXiv:2605.12491},
  year   = {2026}
}

Comments

Project repository here: https://github.com/alansong1322/VECA

R2 v1 2026-07-22T07:08:19.505Z