Vision Transformers (ViTs) have demonstrated state-ofthe-art performance in several benchmarks, yet their high computational costs hinders their practical deployment. Patch Pruning offers significant savings, but existing approaches restrict token reduction to deeper layers, leaving early-stage compression unexplored. This limits their potential for holistic efficiency. In this work, we present a novel Video Patch Pruning framework (VPP) that integrates temporal prior knowledge to enable efficient sparsity within early ViT layers. Our approach is motivated by the observation that prior features extracted from deeper layers exhibit strong foreground selectivity. Therefore we propose a fully differentiable module for temporal mapping to accurately select the most relevant patches in early network stages. Notably, the proposed method enables a patch reduction of up to 60% in dense prediction tasks, exceeding the capabilities of conventional image-based patch pruning, which typically operate around a 30% patch sparsity. VPP excels the high-sparsity regime, sustaining remarkable performance even when patch usage is reduced below 55%. Specifically, it preserves stable results with a maximal performance drop of 0.6% on the Youtube-VIS 2021 dataset.
Figures & tables
Figure 1: Pruning masks for image-based (SViT) and video-based (VPP) pruning at 60% avg. patch reduction. For instance segmentation, only the highlighted objects must be segmented, while the background can be ignored. VPP uses temporal information for early-stage pruning, effectively preserving the foreground.
Method
pruning before layer 3
pruning- stages
additional prune param
dynamic pruning
DynamicViT [ 24 ]
✗
3
0.723 M
✗
TPS [ 36 ]
✗
3
0.723 M
✗
SViT [ 20 ]
✗
9
0.341 M
✓
VPP (ours)
✓
4
0.189 M
✓
Table 1: Categorical overview Patch Pruning methods. VPP leverages video context for early feature reduction, ensuring minimal computational overhead. The method also enables adaptive mask sizing to better adapt the mask to varying instance sizes.
Figure 2: Foreground Selectivity (FGS) across layers. This plot shows the ability of feature x to identify patches belonging to an object instance. Initial features (idx: 0) and early blocks (idx: 1-3) significantly lack in Foreground Selectivity.
Figure 3: Mapping-Selective Module (Map-SM). Using the features from preceding frame ( x1t−1 ), Map-SM calculates an association matrix A to establish patch correspondences in order to map high-level features x6t−1 onto the current frame. Mask M6t−1 ensures that only the remaining patches from the preceding frame are considered for the mask selection process.
Figure 4: Performance Loss in AP vs. Pruned Patches on Youtube-VIS 2019 and 2021, using model size small.
model
tiny
small
dataset
goal PKR (%)
method
PKR (%) ↓
mAPbbox↑
mAPsegm↑
AP↑
ΔAP
PKR (%) ↓
mAPbbox↑
mAPsegm↑
AP↑
Δ AP
YTVIS 21
100
ViT-Adapter [ 3 ]
100
49.1
44.8
42.2
-
100
57.3
52.3
50.4
-
55
DynViT [ 24 ]
54.4
44.1
40.4
36.9
-5.3
54.4
53.8
48.8
45.2
-5.2
TPS [ 36 ]
54.4
46.9
42.8
38.9
-3,3
54.4
54.9
50.2
46.2
-4.2
SViT [ 20 ]
54.0
46.7
42.9
39.2
-3.0
54.2
55.8
51.1
46.9
-3.5
VPP (ours)
52.7
49.9
45.7
42.3
+0.1
51.5
56.1
51.3
49.8
-0.6
Table 2: Average Precision on YTVIS 19/21 Dataset. The results are grouped into similar sparsity levels, measured by PKR. Instance segmentation metrics mAP (bbox and segm) are reported, independent of tracking ID association. The best scores are highlighted as bold.
PKr (%)
method
IoI
IoIS
IoIM
IoIL
55%
SViT
73.7%
77.2%
68.8%
65.8%
VPP (ours)
82.3%
87.3%
76.2%
72.2%
40%
SViT
62.7%
67.5%
56.0%
52.1%
VPP (ours)
73.4%
79.5%
65.9%
60.7%
Table 3: Intersection over Instance (IoI) of ViT-Small on YoutubeVIS 2021. Scores are separated by instance size: S mall ( ≤10% ), M edium (10%−20%) , and L arge (>20%) .
No.
Map-SM index i
SM idx index i
PKR (%)
AP
Note
1
-
0,3,6,9
55.9
40.1
2
-
1,3,6,9
52.4
40.3
3
0
3,6,9
51.6
41.5
4
1
-
54.2
40.6
const Kr
5
1
3,6,9
52.7
42.3
(default)
Table 4: Positioning ablation of Map-SM and SM on dataset YoutubeVIS 2021 at PKR=55%. Early patch initialization with Map-SM (no. 5) improves AP by +2.0% , compared to image-based patch selection (no. 1+2).
Figure 5: Patch density per layer. VPP removes patches after layer 1, while maintaining high patch density within the deepest layers (7-12). In contrast, image-based SViT requires dense initial layers, leaving the deepest layers with insufficient patch density.
Figure 6: Mask adaptation of VPP at scene switches. Patch activity is visualized through shading: less activated patches are darker, and fully black patches are skipped. White, textureless inputs ( t≤0 ) led to a centrally sampled masks. After the video is enabled at t=1 , VPP is able to reallocate the foreground mask within a single timestep ( t=2 ). The last column shows the GT-mask as reference.
method
AP ↑
FLOPS ↓
FPSBB↑
FPSM↑
ViT-Ada [ 3 ]
50.4
88.0 G
17.14
6.58
SViT [ 20 ]
46.9
59.6 G
24.87
8.19
VPP (ours)
49.8
59.9 G
25.94
8.13
Table 5: FLOPS and Throughput, measured in Frames per Second (FPS). All experiments utilize the YTVIS-2021 dataset and involve pruning the ViT-Ada-small model with a 55% PKR.
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
name hyperparameter
value
pos idx Map-SM
1
ref-layer Map-SM
6
pos idx SM
3,6,9
τ
10
scale LspMap
10
scale LspSM
40
Appendix
Table 6: Hyperparameters used in Video Patch Pruning
κinit
PKR (%)
AP ↑
0.3
51.4
41.8
0.5
52.7
42.3
0.7
54.1
39.5
Appendix
Table 7: Evaluation of model performance across different settings for κinit . ViT-Adapter Tiny is trained on Youtube-VIS 2021 at a goal-PKR of 55%. κinit=0.5 shows the best performance.
Figure 7: Examples for the initial pruning mask M1t , generated by Mapping Selective Module (Map-SM). This mask removes 40% of patches after the first transformer block. All masks demonstrate that patches from foreground objects are sampled more densely, while background regions are effectively pruned during the initial mask generation.
PKR (%)
dataset Youtube-VIS
method
IoI
IoIS
IoIM
IoIL
55%
2021
SViT
73.7%
77.2%
68.8%
65.8%
VPP (ours)
82.3%
87.3%
76.2%
72.2%
2019
SViT
63.1%
67.9%
61.0%
58.5%
VPP (ours)
77.3%
84.4%
75.0%
71.4%
40%
2021
SViT
62.7%
67.5%
56.0%
52.1%
VPP (ours)
73.4%
79.5%
65.9%
60.7%
Appendix
Table 8: Intersection over Instance (IoI) over datasets Youtube-VIS 2019 and 2021. VPP shows superior patch coverage every setting, improving the IoI scores by >10% compared to SViT.
Figure 8: Comparison of token usage ( 55% PKR) for image-based ( SViT ) and video-based ( VPP ) Pruning. Unlike image-based patch pruning, VPP computes foreground features almost fully dense, while sparsely activating background patches for context preservation. For Video Instance Segmentation, this is essential to ensure the model focuses on the instances to be segmented, thereby minimizing redundant background computation.
Figure 9: Exemplary pruning masks for videos 1–4 at a 40% Patch Keep Ratio (PKR). Results are shown for timesteps t∈[1,8] . Patch activity is represented via grayscale shading, where darker patches are activated less. Fully black patches are removed by Map-SM after the first layer.
Figure 10: Exemplary pruning masks for videos 5-8 at a 40% Patch Keep Ratio (PKR). Results are shown for timesteps t∈[1,8] . Patch activity is represented via grayscale shading, where darker patches are activated less. Fully black patches are removed by Map-SM after the first layer.
Multimodal Large Language Models (MLLMs) have demonstrated strong performance in video understanding, yet efficiently processing long, high-resolution videos remains challenging. Such videos often contain substantial spatiotemporal redundancy, and processing redundant visual tokens can incur avoidable computational overhead. Many existing methods prune visual tokens during or after vision transformer (ViT) encoding, leaving much of the encoding cost unaddressed. Some approaches prune patches before encoding but rely on learned auxiliary networks for patch selection, incurring additional training and inference overhead. To address these limitations, we propose FlashGaze, a training-free method that reduces spatiotemporal redundancy before ViT encoding without introducing auxiliary networks. FlashGaze uses pixel-space differences as a proxy for information loss and employs Quadtree Dynamic Programming to jointly optimize patch dropping, merging, and keeping under a fixed budget. Experiments on two MLLM backbones across multiple benchmarks demonstrate substantial efficiency gains while largely preserving accuracy. On Qwen3-VL-8B, FlashGaze retains 98% of the full-input baseline accuracy on LongVideoBench while achieving up to 5.4x and 17x speedups in ViT encoding and MLLM prefill, respectively, and reducing peak GPU memory usage by a factor of 1.8. These efficiency gains enable the model to process videos with more frames and higher resolutions on the same GPU hardware, unlocking video understanding at scales previously out of reach.
Ziye Zhu, Yanghao Zhou, Lixing Tan +3
Peking University · Canva Research · Sun Yat-sen University
Token pruning is essential for enhancing the computational efficiency of vision-language models (VLMs), particularly for video-based tasks where temporal redundancy is prevalent. Prior approaches typically prune tokens either (1) within the vision transformer (ViT) exclusively for unimodal perception tasks such as action recognition and object segmentation, without adapting to downstream vision-language tasks; or (2) only within the LLM while leaving the ViT output intact, often requiring complex text-conditioned token selection mechanisms. In this paper, we introduce Spatio-Temporal Token Scoring (STTS), a simple and lightweight module that prunes vision tokens across both the ViT and the LLM without text conditioning or token merging, and is fully compatible with end-to-end training. By learning how to score temporally via an auxiliary loss and spatially via LLM downstream gradients, aided by our efficient packing algorithm, STTS prunes 50% of vision tokens throughout the entire architecture, resulting in a 62% improvement in efficiency during both training and inference with only a 0.7% drop in average performance across 13 short and long video QA tasks. Efficiency gains increase with more sampled frames per video. Applying test-time scaling for long-video QA further yields performance gains of 0.5-1% compared to the baseline. Overall, STTS represents a novel, simple yet effective technique for unified, architecture-wide vision token pruning.
Jianrui Zhang, Yue Yang, Rohun Tripathi +5
University of Wisconsin-Madison · Allen Institute for AI
Video Large Language Models (VLLMs) excel in video understanding, but their excessive visual tokens pose a significant computational challenge for real-world applications. Current methods aim to enhance inference efficiency by visual token pruning. However, they do not consider the dynamic characteristics and temporal dependencies of video frames, as they perceive video understanding as a multi-frame task. To address these challenges, we propose MMG-Vid, a novel training-free visual token pruning framework that removes redundancy by Maximizing Marginal Gains at both segment-level and token-level. Specifically, we first divide the video into segments based on frame similarity, and then dynamically allocate the token budget for each segment to maximize the marginal gain of each segment. Subsequently, we propose a temporal-guided DPC algorithm that jointly models inter-frame uniqueness and intra-frame diversity, thereby maximizing the marginal gain of each token. By combining both stages, MMG-Vid can maximize the utilization of the limited token budget, significantly improving efficiency while maintaining strong performance. Extensive experiments demonstrate that MMG-Vid can maintain over 99.5% of the original performance, while effectively reducing 75% visual tokens and accelerating the prefilling stage by 3.9x on LLaVA-OneVision-7B. Code is available at https://github.com/ylnxxts/MMG-Vid.
Junpeng Ma, Qizhe Zhang, Ming Lu +4
State Key Laboratory of Multimedia Information Processing, School of Computer Science, Peking University · Fudan University · Taobao & Tmall Group of Alibaba