English
Related papers

Related papers: Learning to Merge Tokens in Vision Transformers

200 papers

Despite recent advances in subquadratic attention mechanisms or state-space models, processing long token sequences still imposes significant computational requirements. Token merging has emerged as a solution to increase computational…

Machine Learning · Computer Science 2025-08-06 Leon Götz , Marcel Kollovieh , Stephan Günnemann , Leo Schwinn

This paper investigates how to efficiently deploy vision transformers on edge devices for small workloads. Recent methods reduce the latency of transformer neural networks by removing or merging tokens, with small accuracy degradation.…

Machine Learning · Computer Science 2024-11-12 Nick John Eliopoulos , Purvish Jajal , James C. Davis , Gaowen Liu , George K. Thiravathukal , Yung-Hsiang Lu

Token compression is essential for reducing the computational and memory requirements of transformer models, enabling their deployment in resource-constrained environments. In this work, we propose an efficient and hardware-compatible token…

Computer Vision and Pattern Recognition · Computer Science 2025-04-01 Junzhu Mao , Yang Shen , Jinyang Guo , Yazhou Yao , Xiansheng Hua

We show that Transformer encoder architectures can be sped up, with limited accuracy costs, by replacing the self-attention sublayers with simple linear transformations that "mix" input tokens. These linear mixers, along with standard…

Computation and Language · Computer Science 2022-05-30 James Lee-Thorp , Joshua Ainslie , Ilya Eckstein , Santiago Ontanon

Hybrid vision transformers combine the elements of conventional neural networks (NN) and vision transformers (ViT) to enable lightweight and accurate detection. However, several challenges remain for their efficient deployment on…

Hardware Architecture · Computer Science 2025-07-22 Joren Dumoulin , Pouya Houshmand , Vikram Jain , Marian Verhelst

Vision Transformers have emerged as powerful, scalable and versatile representation learners. To capture both global and local features, a learnable [CLS] class token is typically prepended to the input sequence of patch tokens. Despite…

Computer Vision and Pattern Recognition · Computer Science 2026-02-10 Alexis Marouani , Oriane Siméoni , Hervé Jégou , Piotr Bojanowski , Huy V. Vo

Extensive work has demonstrated the effectiveness of Vision Transformers. The plain Vision Transformer tends to obtain multi-scale features by selecting fixed layers, or the last layer of features aiming to achieve higher performance in…

Computer Vision and Pattern Recognition · Computer Science 2023-05-10 Fangjian Lin , Yizhe Ma , Shengwei Tian

Vision Transformers (ViTs) can learn strong image-level representations while their patch representations become less effective for dense prediction during prolonged training. We revisit this dense degradation phenomenon and argue that it…

Computer Vision and Pattern Recognition · Computer Science 2026-05-25 Linxiang Su

Vision Transformers (ViTs) partition input images into uniformly sized patches regardless of their content, resulting in long input sequence lengths for high-resolution images. We present Adaptive Patch Transformers (APT), which addresses…

Computer Vision and Pattern Recognition · Computer Science 2026-04-24 Rohan Choudhury , JungEun Kim , Jinhyung Park , Eunho Yang , László A. Jeni , Kris M. Kitani

Recent token reduction methods for Vision Transformers (ViTs) incorporate token merging, which measures the similarities between token embeddings and combines the most similar pairs. However, their merging policies are directly dependent on…

Computer Vision and Pattern Recognition · Computer Science 2024-12-17 Dong Hoon Lee , Seunghoon Hong

Modeling visual data as tokens (i.e., image patches) using attention mechanisms, feed-forward networks or convolutions has been highly effective in recent years. Such methods usually have a common pipeline: a tokenization method, followed…

Computer Vision and Pattern Recognition · Computer Science 2023-11-21 Kumara Kahatapitiya , Michael S. Ryoo

In this paper, we introduce the big.LITTLE Vision Transformer, an innovative architecture aimed at achieving efficient visual recognition. This dual-transformer system is composed of two distinct blocks: the big performance block,…

Computer Vision and Pattern Recognition · Computer Science 2024-10-15 He Guo , Yulong Wang , Zixuan Ye , Jifeng Dai , Yuwen Xiong

This paper studies how to keep a vision backbone effective while removing token mixers in its basic building blocks. Token mixers, as self-attention for vision transformers (ViTs), are intended to perform information communication between…

Computer Vision and Pattern Recognition · Computer Science 2023-04-13 Jiahao Wang , Songyang Zhang , Yong Liu , Taiqiang Wu , Yujiu Yang , Xihui Liu , Kai Chen , Ping Luo , Dahua Lin

Transformers were initially introduced for natural language processing (NLP) tasks, but fast they were adopted by most deep learning fields, including computer vision. They measure the relationships between pairs of input tokens (words in…

Computer Vision and Pattern Recognition · Computer Science 2023-03-22 Robin Courant , Maika Edberg , Nicolas Dufour , Vicky Kalogeiton

Visual place recognition is a challenging task in the field of computer vision, and autonomous robotics and vehicles, which aims to identify a location or a place from visual inputs. Contemporary methods in visual place recognition employ…

Computer Vision and Pattern Recognition · Computer Science 2024-05-29 Shyam Sundar Kannan , Byung-Cheol Min

Transformer architecture has become the de-facto model for many machine learning tasks from natural language processing and computer vision. As such, improving its computational efficiency becomes paramount. One of the major computational…

Computation and Language · Computer Science 2022-05-17 Yue Guan , Zhengyi Li , Jingwen Leng , Zhouhan Lin , Minyi Guo

Decreasing sequence length is a common way to accelerate transformers, but prior token reduction work often targets classification and reports proxy metrics rather than end-to-end latency. For semantic segmentation, token reduction is…

Computer Vision and Pattern Recognition · Computer Science 2026-04-08 Simon Ravé , Pejman Rasti , David Rousseau

The recently developed vision transformer (ViT) has achieved promising results on image classification compared to convolutional neural networks. Inspired by this, in this paper, we study how to learn multi-scale feature representations in…

Computer Vision and Pattern Recognition · Computer Science 2021-08-24 Chun-Fu Chen , Quanfu Fan , Rameswar Panda

3D vision foundation models like Visual Geometry Grounded Transformer (VGGT) have advanced greatly in geometric perception. However, it is time-consuming and memory-intensive for long sequences, limiting application to large-scale scenes…

Computer Vision and Pattern Recognition · Computer Science 2025-12-05 Zhijian Shu , Cheng Lin , Tao Xie , Wei Yin , Ben Li , Zhiyuan Pu , Weize Li , Yao Yao , Xun Cao , Xiaoyang Guo , Xiao-Xiao Long

Transformer-based architectures are the model of choice for natural language understanding, but they come at a significant cost, as they have quadratic complexity in the input length, require a lot of training data, and can be difficult to…

Computation and Language · Computer Science 2023-11-14 Florian Mai , Arnaud Pannatier , Fabio Fehr , Haolin Chen , Francois Marelli , Francois Fleuret , James Henderson