From Patching to Pruning Visual Computation in Vision Language Models
cs.CV, cs.LG
Submitted: 2026-10-02
Updated: 2026-10-02
License: http://creativecommons.org/licenses/by-sa/4.0/
The gist: Vision language models (VLMs) incur substantial inference cost because every visual token is processed by the attention and MLP projections of every decoder layer, even when token-specific visual
Terminology
Abstract
Vision language models (VLMs) incur substantial inference cost because every visual token is processed by the attention and MLP projections of every decoder layer, even when token-specific visual computation is unnecessary at many depths. We introduce Patch-to-Prune (P2P), inspired by Mechanistic Interpretability, a training-free framework that converts activation patching from a diagnostic tool into an inference-time computation bypass. P2P performs validation-guided forward and backward layer sweeps to identify decoder regions whose visual-token projection outputs can be replaced by fixed neutral proxy activation vectors within a user-specified accuracy tolerance. Unlike conventional token-pruning methods, P2P preserves the sequence length, token order, positional information, attention mask, and residual pathways, thereby pruning computation without removing tokens or modifying the pretrained model weights. We evaluate P2P on four VLMs from the Qwen2.5-VL and LLaVA families across seven multi-modal benchmarks using mutually disjoint calibration, validation, and test partitions. P2P at a 3% tolerance retains around 94% of dense accuracy while reducing FLOPs by 55%. Beyond these efficiency gains, our layer-wise analysis suggests that visual processing in VLMs is non-uniformly distributed across decoder depth: early and late layers often require little token-specific visual computation, whereas intermediate layers appear to perform most task-relevant visual integration, enabling later reasoning to rely largely on visual information already embedded in shared residual and textual representations. This makes P2P both an efficient inference framework and a causal lens into visual information processing in VLMs.
Sources
- Qwen-VL: A Versatile Vision-Language Model for Understanding, Localization, Text Reading, and Beyond
- How to use and interpret activation patching
- Token Reduction Should Go Beyond Efficiency in Generative Models -- From Vision, Language to Multimodality
- AtP*: An efficient and scalable method for localizing LLM behaviour to components
- Pruning-as-Search: Efficient Neural Architecture Search via Channel Pruning and Structural Reparameterization
- Effective MoE-based LLM Compression by Exploiting Heterogeneous Inter-Group Experts Routing Frequency and Information Density
- DraftAttention: Fast Video Diffusion via Low-Resolution Attention Guidance
- Efficient Reasoning with Hidden Thinking
- Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolution
- PyramidDrop: Accelerating Your Large Vision-Language Models via Pyramid Visual Redundancy Reduction
- A Survey of Advancing Audio Super-Resolution and Bandwidth Extension from Discriminative to Generative Models
- SparseVLM: Visual Token Sparsification for Efficient Vision-Language Model Inference
- Taming Diffusion for Dataset Distillation with High Representativeness
- HIERAMP: Coarse-to-Fine Autoregressive Amplification for Generative Dataset Distillation
- OmniMem: Scalable and Adaptive Memory Retrieval for Long Video Generation
- 7B Fully Open Source Moxin-LLM/VLM -- From Pretraining to GRPO-based Reinforcement Learning Enhancement
- Open-Source Multimodal Moxin Models with Moxin-VLM and Moxin-VLA
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models