Can reinforcement-learned token selection reach extreme sparsity in vision transformers?

LookThere uses reinforcement learning to jointly train a token selector and feature extractor, reaching 0.2% input with accuracy that rivals heuristic token...

Direct answer

Vision transformers normally process every patch, but most recognition tasks need only a small fraction. LookThere reframes token selection as a reinforcement learning problem, jointly training a shallow selector and a deep extractor so the model learns which patches matter for the task rather than inheriting attention heuristics [1]. Across ImageNet, ADE20K, zero-shot classification, counting, and high-resolution traffic and billiards benchmarks, it pushes the accuracy-compute Pareto frontier past LookWhere, the prior selection-extraction state of the art [1][5]. The gain is largest at extreme sparsity, where attention-distilled selectors degrade and reinforcement-learned selection stays competitive [1][3].

8sources cited

This article was generated with WisPaper-powered search and paper analysis.

Why attention-based token selection struggles at extreme sparsity

Earlier adaptive computation work established that ViTs can skip tokens, but most methods either prune or merge after processing all tokens in early layers, or distill a teacher's attention to decide where to look [3][5]. LookWhere showed that a selector-extractor factorization distilled from DINOv2 attention reaches strong accuracy-efficiency trade-offs on ImageNet and ADE20K, and generalizes to high-resolution traffic, birds, and billiards benchmarks [5]. SPOT extended this line by aggregating attention dynamics across layers and heads, improving ImageNet accuracy over DeiT baselines while reducing GFLOPS [3]. Both approaches, however, tie selection to attention maps that reflect visual salience rather than task relevance, and the LookThere authors show DINOv2 attention scattering onto foliage around birds or cars instead of signs [1].

The practical consequence is that attention-distilled selectors need more tokens than necessary to preserve accuracy, and their advantage shrinks as sparsity increases. LookThere's own comparison shows LookWhere losing substantial accuracy at 92% sparsity on SigLIP 2 zero-shot classification, dropping from 71.3% dense to 44.6%, while LookThere retains 61.7% [1]. That gap is the clearest evidence that the heuristic itself, not just the selector architecture, becomes the bottleneck under aggressive token budgets.

Joint reinforcement learning replaces the heuristic with task feedback

LookThere factorizes inference into a shallow selector that sees a low-resolution view and a high-capacity extractor that processes only selected high-resolution patches [1]. The selector outputs a hierarchical Bernoulli policy over patches, and the extractor's task loss becomes the reward signal optimized with Group Relative Policy Optimization, so the selector learns which patches actually help prediction rather than which patches a teacher attends to [1]. This differs from PRANCE, which also uses PPO for token and channel decisions but optimizes a combinatorial architecture-and-token space with a Result-to-Go mechanism [8]. LookThere keeps the backbone fixed and learns only spatial selection, which simplifies credit assignment and lets it scale to high-resolution inputs where PRANCE-style joint optimization has not been demonstrated.

The reinforcement learning formulation also connects to work outside vision. Sparsity Forcing applies GRPO with a joint efficiency-performance reward to prune visual tokens in multimodal LLMs, and reports that dynamic top-p sampling outperforms fixed top-k or threshold policies [4]. Video-KTR uses token-level attribution to shape RL updates in video reasoning, showing that selective token supervision improves accuracy on Video-Holmes and related benchmarks [6]. LookThere's contribution is to bring this RL-for-sparsity logic into spatial vision transformer selection, where the action space is a patch mask rather than a token budget in a language model.

Where the Pareto frontier actually moves

On ImageNet classification with DINOv2 initialization, LookThere matches LookWhere at 10% of patches, and under SigLIP, MAE, and EVA initializations it surpasses LookWhere, reaching 81.6%, 78.1%, and 83.0% top-1 respectively against LookWhere's 68.2%, 67.7%, and 48.3% [1]. The EVA result is the most striking: LookWhere collapses to 48.3% while LookThere retains 83.0%, suggesting that attention distillation depends heavily on the teacher having task-relevant attention, whereas RL selection does not [1]. On ADE20K segmentation, LookThere improves mIoU over LookWhere across all four backbones, reaching 42.0% with DINOv2 at 15% of input versus LookWhere's 34.9% [1].

The high-resolution results are where the method is most differentiated. On Traffic Signs, LookThere nearly matches DINOv2 accuracy at 5% of patches and remains competitive at 0.2%, while on Billiards it reaches 98.8% of its DINOv2 teacher's accuracy at a fraction of the patches and 50 GFLOPs versus LookWhere's 261 GFLOPs [1]. On class-agnostic counting with CACViT, LookThere achieves lower MAE than LookWhere at equal sparsity despite lacking explicit selection supervision [1]. These are the settings where selection before deep computation matters most, because processing all high-resolution tokens is prohibitive.

Generalization claims and their boundaries

LookThere reports generalization across tasks and architectures: ImageNet classification, ADE20K segmentation, Places365 scene recognition, MS-COCO instance segmentation, zero-shot classification by SigLIP 2 distillation, and counting regression [1]. On Places365 at 86% sparsity it holds 57.5% against LookWhere's 55.6%, and on MS-COCO instance segmentation at 25% of patches it reaches 0.444 mask AP versus LookWhere's 0.327 and a dense backbone's 0.473 [1]. These results support the claim that RL selection transfers across task types better than attention distillation, but they are all within the same broad family of ViT-based recognition and regression tasks.

The limitations section is explicit that the approach inherits RL's sample inefficiency and may overfit in low-data regimes, where the extractor can compensate for poor selection in ways that fail to generalize [1]. Patch-level credit assignment remains difficult because rewards are assigned to the full mask, and GRPO only partly alleviates this through grouped comparisons [1]. The paper also focuses on spatial selection; temporal and multimodal selection are left as future work [1]. Independent evidence from UAV detection shows that dynamic token pruning based on attention variance can work in edge deployment, but that is a different selection criterion and does not test LookThere's RL formulation [7]. Federated GI lesion classification with entropy-guided quantum-inspired pruning similarly demonstrates token pruning under privacy constraints, but again with a non-RL selection mechanism [2]. The boundary is therefore clear: LookThere's extreme-sparsity advantage is demonstrated on the reported benchmarks, not guaranteed across all vision tasks, model scales, or data regimes.

About These Sources

This research page is built on 8 studies (7 peer-reviewed, 1 preprint) — published from 2025 to 2026, 8 from 2024 or later — selected as the most relevant from 8 studies that passed quality screening, drawn from 75 papers retrieved from a database of over 500 million.

Sources used in this answer

1

LookThere! Sparse Vision by Reinforced Selection

LookThere introduces an end-to-end reinforcement learning framework that jointly trains a shallow token selector and a deep feature extractor, achieving accurate recognition with as little as 0.2% of input patches and surpassing LookWhere across classification, segmentation, zero-shot, counting, and high-resolution benchmarks [1].

2

Federated Gastrointestinal Lesion Classification with Clinical-Entropy Guided Quantum-Inspired Token Pruning in Vision Transformers.

This foundational federated learning study uses a Clinical-Entropy Guided Quantum Evolutionary Algorithm for adaptive token pruning in ViTs, achieving 92.33% micro accuracy and 99.23% specificity on gastrointestinal endoscopy classification across three clients [2].

3

SPOT: Sparsification with attention dynamics via token relevance in vision transformers

SPOT is a precursor token sparsification framework that aggregates token embeddings and attention dynamics across multiple ViT layers, improving ImageNet accuracy over DeiT baselines while reducing GFLOPS and transferring to CIFAR-100, Food-101, DTD, and EuroSAT [3].

4

Sparsity forcing: reinforcing token sparsity of mllms

Sparsity Forcing applies GRPO with a joint efficiency-performance reward to reinforce token sparsity in multimodal LLMs, showing that dynamic top-p sampling achieves a better efficiency-performance trade-off than fixed top-k or threshold-based pruning [4].

5

LookWhere? Efficient Visual Recognition by Learning Where to Look and What to See from Self-Supervision

LookWhere establishes the prior state of the art in selection-extraction by distilling self-supervised teacher attention, reaching strong accuracy-efficiency trade-offs on ImageNet, ADE20K, and high-resolution traffic, birds, and billiards benchmarks [5].

6

Video-ktr: Reinforcing video reasoning via key token attribution

Video-KTR demonstrates that token-level attribution signals combining visual, temporal, and uncertainty information can shape reinforcement learning updates in video multimodal models, improving reasoning accuracy on five benchmarks including Video-Holmes [6].

7

Real-time object detection for unmanned aerial vehicles based on vision transformer and edge computing.

This foundational UAV detection study adapts ViTs for edge deployment with hierarchical window attention and attention-variance-based dynamic token pruning, achieving 73.9% mAP@0.5:0.95 and 38.1 FPS in field trials on DJI Matrice 300 RTK [7].

8

PRANCE: Joint Token-Optimization and Structural Channel-Pruning for Adaptive ViT Inference.

PRANCE jointly optimizes token selection and structural channel pruning using PPO with a Result-to-Go training mechanism, reducing FLOPs by approximately 50% while retaining about 10% of tokens with lossless top-1 accuracy [8].