English
Related papers

Related papers: Token Merging: Your ViT But Faster

200 papers

Masked image modeling (MIM) pre-training for large-scale vision transformers (ViTs) has enabled promising downstream performance on top of the learned self-supervised ViT features. In this paper, we question if the \textit{extremely simple}…

Computer Vision and Pattern Recognition · Computer Science 2024-05-28 Jin Gao , Shubo Lin , Shaoru Wang , Yutong Kou , Zeming Li , Liang Li , Congxuan Zhang , Xiaoqin Zhang , Yizheng Wang , Weiming Hu

Structural re-parameterization is a general training scheme for Convolutional Neural Networks (CNNs), which achieves performance improvement without increasing inference cost. As Vision Transformers (ViTs) are gradually surpassing CNNs in…

Computer Vision and Pattern Recognition · Computer Science 2023-08-28 Yongqi Huang , Peng Ye , Xiaoshui Huang , Sheng Li , Tao Chen , Tong He , Wanli Ouyang

Vision transformers (ViTs) have emerged as a prevalent architecture for vision tasks owing to their impressive performance. However, when it comes to handling long token sequences, especially in dense prediction tasks that require…

Computer Vision and Pattern Recognition · Computer Science 2023-11-03 Jin Li , Yaoming Wang , Xiaopeng Zhang , Bowen Shi , Dongsheng Jiang , Chenglin Li , Wenrui Dai , Hongkai Xiong , Qi Tian

Existing online video segmentation models typically combine a per-frame segmenter with complex specialized tracking modules. While effective, these modules introduce significant architectural complexity and computational overhead. Recent…

Computer Vision and Pattern Recognition · Computer Science 2026-03-05 Narges Norouzi , Idil Esen Zulfikar , Niccolò Cavagnero , Tommie Kerssies , Bastian Leibe , Gijs Dubbelman , Daan de Geus

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

Vision Transformer (ViT) extracts the final representation from either class token or an average of all patch tokens, following the architecture of Transformer in Natural Language Processing (NLP) or Convolutional Neural Networks (CNNs) in…

Computer Vision and Pattern Recognition · Computer Science 2022-12-09 Byungsoo Ko , Han-Gyu Kim , Byeongho Heo , Sangdoo Yun , Sanghyuk Chun , Geonmo Gu , Wonjae Kim

State-of-the-art machine learning often follows a two-stage process: $(i)$~pre-training on large, general-purpose datasets; $(ii)$~fine-tuning on task-specific data. In fine-tuning, selecting training examples that closely reflect the…

Machine Learning · Computer Science 2025-10-02 Ayush Jain , Andrea Montanari , Eren Sasoglu

Vision transformers are known to be more computationally and data-intensive than CNN models. These transformer models such as ViT, require all the input image tokens to learn the relationship among them. However, many of these tokens are…

Computer Vision and Pattern Recognition · Computer Science 2024-01-30 Foozhan Ataiefard , Walid Ahmed , Habib Hajimolahoseini , Saina Asani , Farnoosh Javadi , Mohammad Hassanpour , Omar Mohamed Awad , Austin Wen , Kangling Liu , Yang Liu

Long-context efficiency has recently become a trending topic in serving large language models (LLMs). And mixture of depths (MoD) is proposed as a perfect fit to bring down both latency and memory. In this paper, however, we discover that…

Computation and Language · Computer Science 2024-10-21 Chen Zhang , Meizhi Zhong , Qimeng Wang , Xuantao Lu , Zheyu Ye , Chengqiang Lu , Yan Gao , Yao Hu , Kehai Chen , Min Zhang , Dawei Song

Video-language models (VLMs) face rapid inference costs as visual token counts scale with video length. For example, 32 frames at $448{\times}448$ resolution already yield >8,000 visual tokens in Qwen3-VL, making LLM prefill the dominant…

Computer Vision and Pattern Recognition · Computer Science 2026-05-11 Simin Huo , Ning LI

The mixture of Expert (MoE) parallelism is a recent advancement that scales up the model size with constant computational cost. MoE selects different sets of parameters (i.e., experts) for each incoming token, resulting in a…

Machine Learning · Computer Science 2022-12-13 Chaoyang He , Shuai Zheng , Aston Zhang , George Karypis , Trishul Chilimbi , Mahdi Soltanolkotabi , Salman Avestimehr

The quadratic computational complexity to the number of tokens limits the practical applications of Vision Transformers (ViTs). Several works propose to prune redundant tokens to achieve efficient ViTs. However, these methods generally…

Computer Vision and Pattern Recognition · Computer Science 2023-03-31 Shuning Chang , Pichao Wang , Ming Lin , Fan Wang , David Junhao Zhang , Rong Jin , Mike Zheng Shou

The recent success of specialized Large Language Models (LLMs) in domains such as mathematical reasoning and coding has led to growing interest in methods for merging these expert LLMs into a unified Mixture-of-Experts (MoE) model, with the…

Computation and Language · Computer Science 2025-02-18 Yuhang Zhou , Giannis Karamanolakis , Victor Soto , Anna Rumshisky , Mayank Kulkarni , Furong Huang , Wei Ai , Jianhua Lu

Learning with large-scale unlabeled data has become a powerful tool for pre-training Visual Transformers (VTs). However, prior works tend to overlook that, in real-world scenarios, the input data may be corrupted and unreliable.…

Computer Vision and Pattern Recognition · Computer Science 2023-04-13 Tianjiao Li , Lin Geng Foo , Ping Hu , Xindi Shang , Hossein Rahmani , Zehuan Yuan , Jun Liu

Mixture-of-Experts (MoE) architectures enhance the efficiency of large language models by activating only a subset of experts per token. However, standard MoE employs a fixed Top-K routing strategy, leading to redundant computation and…

Artificial Intelligence · Computer Science 2026-05-15 Juntong Wu , Jialiang Cheng , Qishen Yin , Yue Dai , Yuliang Yan , Fuyu Lv , Ou Dan , Li Yuan

Recently, the remarkable success of pre-trained Vision Transformers (ViTs) from image-text matching has sparked an interest in image-to-video adaptation. However, most current approaches retain the full forward pass for each frame, leading…

Computer Vision and Pattern Recognition · Computer Science 2024-08-15 Guozhen Zhang , Jingyu Liu , Shengming Cao , Xiaotong Zhao , Kevin Zhao , Kai Ma , Limin Wang

Mixture of Experts (MoE) models have emerged as the de facto architecture for scaling up language models without significantly increasing the computational cost. Recent MoE models demonstrate a clear trend towards high expert granularity…

Machine Learning · Computer Science 2026-03-30 Wentao Guo , Mayank Mishra , Xinle Cheng , Ion Stoica , Tri Dao

Is it always necessary to compute tokens from shallow to deep layers in Transformers? The continued success of vanilla Transformers and their variants suggests an undoubted "yes". In this work, however, we attempt to break the depth-ordered…

Computation and Language · Computer Science 2024-07-10 Zhuocheng Gong , Ang Lv , Jian Guan , Junxi Yan , Wei Wu , Huishuai Zhang , Minlie Huang , Dongyan Zhao , Rui Yan

Contrastive image-text pre-trained models such as CLIP have shown remarkable adaptability to downstream tasks. However, they face challenges due to the high computational requirements of the Vision Transformer (ViT) backbone. Current…

Computer Vision and Pattern Recognition · Computer Science 2024-12-02 Cheng-En Wu , Jinhong Lin , Yu Hen Hu , Pedro Morgado

Unlike offline processing, streaming video vision-language models face two fundamental constraints: causality and accumulation. Causality prevents access to future frames that offline methods exploit, while accumulation causes tokens to…

Computer Vision and Pattern Recognition · Computer Science 2026-03-17 Xueyi Chen , Keda Tao , Kele Shao , Huan Wang