RAPID: Layer-Wise Redundancy-Aware Pruning and Importance-Driven Token Merging for Efficient ViT
On the map: Efficient LLM inference and serving → Token pruning and merging for ViTs
Abstract (arXiv)
Vision Transformers (ViTs) achieve strong performance but suffer from high computational costs due to quadratic self-attention complexity. Although token reduction techniques such as pruning and merging mitigate this, they typically overlook how representations evolve across network depth. We propose RAPID, a depth-aware token reduction framework that adapts reduction strategies to the layer-wise characteristics of token representations. The primary methodological contribution is a bifurcated strategy: in shallow-to-middle layers, RAPID employs a redundancy-similarity aware pruning metric to eliminate over-represented local patterns. As features transition to global semantic concepts in deeper layers, the framework shifts to an importance-similarity aware merging mechanism. This stage leverages classification (CLS) token attention weights to protect semantically critical tokens while fusing less important but similar neighbors. Empirical validation on ImageNet-1K using ViT and DeiT architectures demonstrates that RAPID establishes a superior accuracy-compression Pareto frontier compared to plug-and-play baselines such as ToMe and ToFu. RAPID is particularly robust in aggressive compression regimes, achieving up to 4.29% higher accuracy than ToMe at extreme reduction rates. Our framework provides a training-free template for optimizing vision models by aligning reduction strategies with hierarchical feature evolution.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Token pruning and merging for ViTs, one of 12 directions in Efficient LLM inference and serving. Builds on 14 of its 21 references that are in the map. First posted on arXiv 6 Jun 2026.
Builds on
14 of 21 references.
Closest ancestors · 6
Same direction, newest first
- 2025Similarity-Aware Token Pruning: Your VLM but Faster14 Mar
- 2023Token Fusion: Bridging the Gap between Token Pruning and Token Merging2 Dec
- 2023Joint Token Pruning and Squeezing Towards More Aggressive Compression of Vision Transformers21 Apr
- 2022Token Merging: Your ViT But Faster17 Oct
- 2021SPViT: Enabling Faster Vision Transformers via Soft Token Pruning27 Dec
- 2021Learned Token Pruning for Transformers2 Jul
Other references · 8
- 2017Attention Is All You Needcited by 59,932
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
- 2021MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformercited by 636
- 2021AdaViT: Adaptive Tokens for Efficient Vision Transformercited by 360
- 2023EfficientViT: Memory Efficient Vision Transformer with Cascaded Group Attentioncited by 172
- 2024Beyond Text-Visual Attention: Exploiting Visual Cues for Effective Token Pruning in VLMscited by 153
- 2021Token Pooling in Vision Transformerscited by 68
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2023PPT: Token Pruning and Pooling for Efficient Vision Transformers
- 2025Where Do Tokens Go? Understanding Pruning Behaviors in STEP at High Resolutions
- 2026When Token Compression Breaks: Structural Pruning vs. Token Reduction for Robust ViT Segmentation under High Compression
- 2026R'enyi Entropy: A New Token Pruning Metric for Vision Transformers
- 2022Beyond Attentive Tokens: Incorporating Token Importance and Diversity for Efficient Vision Transformers
- 2024Learning to Merge Tokens via Decoupled Embedding for Efficient Vision Transformers
- 2023Multi-Scale And Token Mergence: Make Your ViT More Efficient
- 2026Later Is Better: Token Reduction for ViTs Under Distribution Shift