ERASE: Eliminating Redundant Visual Tokens via Adaptive Two-Stage Token Pruning
About
Recent advancements in Vision-Language Models (VLMs) enable large language models (LLMs) to process high-resolution images, significantly improving real-world multimodal understanding. However, this capability introduces a large number of vision tokens, resulting in substantial computational overhead. To mitigate this issue, various vision token pruning methods have been proposed. Nevertheless, existing approaches predominantly rely on learned semantic features within the model to capture visual redundancy. Moreover, they lack adaptive mechanisms to adjust pruning strategies according to the complexity of the input image. In this paper, we propose ERASE, a two-stage vision token pruning framework that identifies and retains salient tokens through pruning strategies adaptive to image complexity. Experiment results demonstrate that ERASE significantly reduces vision tokens while preserving accuracy. For Qwen2.5-VL-7B, at a token pruning ratio of 85\%, ERASE retains 89.46% of the original model accuracy, whereas the best prior method retains only 78.1%. Our code is available at https://github.com/Tuna-Luna/ERASE.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Visual Question Answering | TextVQA | Accuracy82.12 | 1453 | |
| Visual Question Answering | ChartQA | Accuracy81.44 | 519 | |
| Optical Character Recognition | OCRBench | Score800 | 433 | |
| Visual Question Answering | TextVQA | TextVQA Accuracy78.62 | 210 | |
| Visual Question Answering | DocVQA | Accuracy91.32 | 205 | |
| Visual Question Answering | InfoVQA | Accuracy73.34 | 195 | |
| Information Visual Question Answering | InfoVQA | Accuracy82.45 | 110 | |
| Mathematical Visual Question Answering | MathVista | Accuracy56.4 | 87 | |
| Visual Question Answering | OCRBench | Score784 | 53 | |
| Visual Grounding | HRBench8K | Accuracy75.63 | 51 |