Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

ASAP: Attention Sink Anchored Pruning

About

Vision Transformers (ViTs) face severe computational bottlenecks due to the quadratic complexity of self-attention at high resolutions. Existing token reduction methods rely on local metrics - such as single-layer attention scores - that are inherently vulnerable to the attention sink phenomenon, where uninformative tokens are paradoxically preserved over salient foreground objects. We propose ASAP (Attention Sink Anchored Pruning), a training-free framework that recasts this sink as a feature. Modeling ViT information flow as a Lazy Random Walk, ASAP identifies the sink as a dominant accumulator of probability mass. By computing the diffusion distance to the sink within the cumulative transition matrix, ASAP partitions tokens via Radial Diffusion Clustering and compresses background redundancy through Transition Weight Pooling in a single shot. Extensive experiments across image, video, and vision-language tasks demonstrate ASAP outperforms state-of-the-art methods, accelerating throughput by up to 48% while maintaining - or even exceeding - baseline accuracy.

Jaehyuk Lee, Hanyoung Kim, Yanggee Kim, Donghun Lee• 2026

Related benchmarks

TaskDatasetResultRank
Object Hallucination EvaluationPOPE--
2019
Visual Question AnsweringVQA v2
Accuracy75.9
1429
Image ClassificationImageNet-1K
Accuracy (Base)83.35
23
Multimodal UnderstandingMMBench MMB EN
Score65.8
22
Multimodal Visual Question AnsweringLLaVA Evaluation Suite (GQA, MME, POPE, SQA-Img, VizWiz, VQAv2, MMB-En) 1.5
GQA60.4
16
Multimodal UnderstandingMME
MME Score1.85e+3
10
Visual Question AnsweringGQA
GQA Score62.3
10
Object Hallucination EvaluationPOPE
Inference (ms)64
9
Showing 8 of 8 rows

Other info

Follow for update