Differentiable Patch Selection for Image Recognition
Not on the map. Why?
Abstract (arXiv)
Neural Networks require large amounts of memory and compute to process high resolution images, even when only a small part of the image is actually informative for the task at hand. We propose a method based on a differentiable Top-K operator to select the most relevant parts of the input to efficiently process high resolution images. Our method may be interfaced with any downstream neural network, is able to aggregate information from different patches in a flexible way, and allows the whole model to be trained end-to-end using backpropagation. We show results for traffic sign recognition, inter-patch relationship reasoning, and fine-grained recognition without using object/part bounding box annotations during training.
Key facts. Cited by 68 papers in the map (data through 7 Oct 2026). Builds on 37 of its 50 references that are in the map. First posted on arXiv 7 Apr 2021.
Builds on
37 of 50 references.
References · 37
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2015Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networkscited by 9,617
- 2017Mask R-CNNcited by 7,259
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2017Squeeze-and-Excitation Networkscited by 4,763
- 2015Fast R-CNNcited by 2,930
- 2015Spatial Transformer Networkscited by 1,423
- 2015Show, Attend and Tell: Neural Image Caption Generation with Visual Attentioncited by 1,421
- 2017Bottom-Up and Top-Down Attention for Image Captioning and Visual Question Answeringcited by 1,310
25 more
- 2018Attention-based Deep Multiple Instance Learningcited by 1,077
- 2017Residual Attention Network for Image Classificationcited by 385
- 2014Recurrent Models of Visual Attentioncited by 344
- 2019On the Relationship between Self-Attention and Convolutional Layerscited by 299
- 2015Stacked Attention Networks for Image Question Answeringcited by 273
- 2015Bilinear CNNs for Fine-grained Visual Recognitioncited by 228
- 2016Attend, Infer, Repeat: Fast Scene Understanding with Generative Modelscited by 209
- 2020Fast Differentiable Sorting and Rankingcited by 196
- 2015Attention to Scale: Scale-aware Semantic Image Segmentationcited by 169
- 2018Tell Me Where to Look: Guided Attention Inference Networkcited by 168
- 2019Differentiable Ranks and Sorting using Optimal Transportcited by 113
- 2017Learning non-maximum suppressioncited by 112
- 2018Learning to Navigate for Fine-grained Classificationcited by 103
- 2020Learning with Differentiable Perturbed Optimizerscited by 81
- 2019Reparameterizable Subset Sampling via Continuous Relaxationscited by 69
- 2015Action Recognition using Visual Attentioncited by 49
- 2019Saccader: Improving Accuracy of Hard Attention Models for Visioncited by 35
- 2019Processing Megapixel Images with Deep Attention-Sampling Modelscited by 29
- 2017A-Lamp: Adaptive Layout-Aware Multi-Patch Deep Convolutional Neural Network for Photo Aesthetic Assessmentcited by 28
- 2017Dynamic Computational Time for Visual Attentioncited by 23
- 2020Differentiable Top-k Operator with Optimal Transportcited by 17
- 2019Patch Transformer for Multi-tagging Whole Slide Histopathology Imagescited by 13
- 2016Recurrent Attentional Networks for Saliency Detectioncited by 11
- 2016Fully Convolutional Attention Networks for Fine-Grained Recognitioncited by 8
- 2018MIST: Multiple Instance Spatial Transformer Networkcited by 3
Built on by
Citing papers per year: 2021: 7; 2022: 21; 2023: 12; 2024: 11; 2025: 10; 2026: 7.
68 papers.
Sort:
By area:
- 2023Self-Chained Image-Language Model for Video Localization and Question Answeringcited by 200
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunitiescited by 129
- 2022TS2-Net: Token Shift and Selection Transformer for Text-Video Retrievalcited by 104
- 2022CLIP-TSA: CLIP-Assisted Temporal Self-Attention for Weakly-Supervised Video Anomaly Detectioncited by 67
- 2021Efficient Video Transformers with Spatial-Temporal Token Selectioncited by 58
- 2022Differentiable Zooming for Multiple Instance Learning on Whole-Slide Imagescited by 43
- 2021Towards mental time travel: a hierarchical memory for reinforcement learning agentscited by 38
- 2024KVQ: Kwai Video Quality Assessment for Short-form Videoscited by 38
- 2022Vision Transformer with Attentive Pooling for Robust Facial Expression Recognitioncited by 35
- 2024Adaptive Slot Attention: Object Discovery with Dynamic Slot Numbercited by 34
- 2023Fast, Differentiable and Sparse Top-k: a Convex Analysis Perspectivecited by 26
- 2022Monotonic Differentiable Sorting Networkscited by 25
38 more
- 2022Token Turing Machinescited by 25
- 2023SI-MIL: Taming Deep MIL for Self-Interpretability in Gigapixel Histopathologycited by 25
- 2022ScoreNet: Learning Non-Uniform Attention and Augmentation for Transformer-Based Histopathological Image Classificationcited by 21
- 2022Differentiable Top-k Classification Learningcited by 21
- 2021Efficient Classification of Very Large Images with Tiny Objectscited by 17
- 2023Differentiable Clustering with Perturbed Spanning Forestscited by 14
- 2022Iterative Patch Selection for High-Resolution Image Recognitioncited by 13
- 2023Learning by Sorting: Self-supervised Learning with Group Ordering Constraintscited by 13
- 2025Positional Encoding in Transformer-Based Time Series Models: A Surveycited by 13
- 2021A Probabilistic Hard Attention Model For Sequentially Observed Scenescited by 12
- 2022EmbryosFormer: Deformable Transformer and Collaborative Encoding-Decoding for Embryos Stage Development Classificationcited by 12
- 2022Consistency driven Sequential Transformers Attention Model for Partially Observable Scenescited by 11
- 2022Knowledge Mining with Scene Text for Fine-Grained Recognitioncited by 10
- 2022Task-adaptive Spatial-Temporal Video Sampler for Few-shot Action Recognitioncited by 9
- 2024StimuVAR: Spatiotemporal Stimuli-aware Video Affective Reasoning with Multimodal Large Language Modelscited by 8
- 2025Label-free Concept Based Multiple Instance Learning for Gigapixel Histopathologycited by 8
- 2021Leveraging Recursive Gumbel-Max Trick for Approximate Inference in Combinatorial Spacescited by 7
- 2023Few-shot Action Recognition via Intra- and Inter-Video Information Maximizationcited by 7
- 2024PATHS: A Hierarchical Transformer for Efficient Whole Slide Image Analysiscited by 7
- 2024Learning to Rank Patches for Unbiased Image Redundancy Reductioncited by 6
- 2022Dynamics-aware Adversarial Attack of Adaptive Neural Networkscited by 5
- 2022Mirror Sinkhorn: Fast Online Optimization on Transport Polytopescited by 5
- 2025No More Sliding Window: Efficient 3D Medical Image Segmentation with Differentiable Top-k Patch Samplingcited by 5
- 2025LookWhere? Efficient Visual Recognition by Learning Where to Look and What to See from Self-Supervisioncited by 5
- 2023Token Sparsification for Faster Medical Image Segmentationcited by 4
- 2025VEGAS: Towards Visually Explainable and Grounded Artificial Social Intelligencecited by 4
- 2024Soft Condorcet Optimization for Ranking of General Agentscited by 3
- 2021Recurrent Attention Models with Object-centric Capsule Representation for Multi-object Recognitioncited by 2
- 2022Weakly-Supervised End-to-End CAD Retrieval to Scan Objectscited by 2
- 2022NUNet: Deep Learning for Non-Uniform Super-Resolution of Turbulent Flowscited by 2
- 2023Self-Ordering Point Cloudscited by 2
- 2023Set Learning for Accurate and Calibrated Modelscited by 2
- 2024MST: Adaptive Multi-Scale Tokens Guided Interactive Segmentationcited by 2
- 2024Newton Losses: Using Curvature Information for Learning with Differentiable Algorithmscited by 2
- 2025Energy-Efficient Transformer Inference: Optimization Strategies for Time Series Classificationcited by 2
- 2025REEF: Relevance-Aware and Efficient LLM Adapter for Video Understandingcited by 2
- 2022Problem-dependent attention and effort in neural networks with applications to image resolution and model selectioncited by 1
- 2023Learning to Summarize Videos by Contrasting Clipscited by 1
50 most cited of 68 All 68 (JSON)
- 2026LookThere! Sparse Vision by Reinforced Selection4 Sep 2026
- 2026Solving the Needle-in-a-Haystack Problem in Mammography Vision-Language Model with Differentiable Subset Sampling2 Sep 2026
- 2026A2: Smaller Self-Supervised ViTs Localize Better than Larger Ones2 Jun 2026
- 2026TT-Sparse: Learning Sparse Rule Models with Differentiable Truth Tables8 Mar 2026
- 2026Transformer-Based Inpainting for Real-Time 3D Streaming in Sparse Multi-Camera Setups5 Mar 2026
- 2026ORGAN: Object-Centric Representation Learning using Cycle Consistent Generative Adversarial Networks2 Mar 2026
- 2026WaveFormer: Wavelet Embedding Transformer for Biomedical Signals12 Feb 2026
- 2025EvoPS: Evolutionary Patch Selection for Whole Slide Image Analysis in Computational Pathology10 Nov 2025
- 2025DyWPE: Signal-Aware Dynamic Wavelet Positional Encoding for Time Series Transformers18 Sep 2025
- 2025LookWhere? Efficient Visual Recognition by Learning Where to Look and What to See from Self-Supervision23 May 2025
- 2025REEF: Relevance-Aware and Efficient LLM Adapter for Video Understanding7 Apr 2025
- 2025VEGAS: Towards Visually Explainable and Grounded Artificial Social Intelligence3 Apr 2025
38 more
- 2025Decision-aware training of spatiotemporal forecasting models to select a top K subset of sites for intervention7 Mar 2025
- 2025Energy-Efficient Transformer Inference: Optimization Strategies for Time Series Classification23 Feb 2025
- 2025Positional Encoding in Transformer-Based Time Series Models: A Survey17 Feb 2025
- 2025No More Sliding Window: Efficient 3D Medical Image Segmentation with Differentiable Top-k Patch Sampling18 Jan 2025
- 2025Label-free Concept Based Multiple Instance Learning for Gigapixel Histopathology6 Jan 2025
- 2024On the Generalizability of Iterative Patch Selection for Memory-Efficient High-Resolution Image Classification15 Dec 2024
- 2024PATHS: A Hierarchical Transformer for Efficient Whole Slide Image Analysis27 Nov 2024
- 2024Soft Condorcet Optimization for Ranking of General Agents31 Oct 2024
- 2024Newton Losses: Using Curvature Information for Learning with Differentiable Algorithms24 Oct 2024
- 2024StimuVAR: Spatiotemporal Stimuli-aware Video Affective Reasoning with Multimodal Large Language Models31 Aug 2024
- 2024SITransformer: Shared Information-Guided Transformer for Extreme Multimodal Summarization28 Aug 2024
- 2024Adaptive Slot Attention: Object Discovery with Dynamic Slot Number13 Jun 2024
- 2024Resource Efficient Perception for Vision Systems12 May 2024
- 2024Learning to Rank Patches for Unbiased Image Redundancy Reduction31 Mar 2024
- 2024KVQ: Kwai Video Quality Assessment for Short-form Videos11 Feb 2024
- 2024MST: Adaptive Multi-Scale Tokens Guided Interactive Segmentation9 Jan 2024
- 2023SI-MIL: Taming Deep MIL for Self-Interpretability in Gigapixel Histopathology22 Dec 2023
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunities25 Jul 2023
- 2023Set Learning for Accurate and Calibrated Models5 Jul 2023
- 2023Differentiable Clustering with Perturbed Spanning Forests25 May 2023
- 2023Self-Chained Image-Language Model for Video Localization and Question Answering11 May 2023
- 2023Few-shot Action Recognition via Intra- and Inter-Video Information Maximization10 May 2023
- 2023Self-Ordering Point Clouds3 Apr 2023
- 2023Token Sparsification for Faster Medical Image Segmentation11 Mar 2023
- 2023Fast, Differentiable and Sparse Top-k: a Convex Analysis Perspective2 Feb 2023
- 2023Patch Gradient Descent: Training Neural Networks on Very Large Images31 Jan 2023
- 2023Learning to Summarize Videos by Contrasting Clips12 Jan 2023
- 2023Learning by Sorting: Self-supervised Learning with Group Ordering Constraints5 Jan 2023
- 2022Vision Transformer with Attentive Pooling for Robust Facial Expression Recognition11 Dec 2022
- 2022CLIP-TSA: CLIP-Assisted Temporal Self-Attention for Weakly-Supervised Video Anomaly Detection9 Dec 2022
- 2022Mirror Sinkhorn: Fast Online Optimization on Transport Polytopes18 Nov 2022
- 2022Token Turing Machines16 Nov 2022
- 2022Iterative Patch Selection for High-Resolution Image Recognition24 Oct 2022
- 2022Dynamics-aware Adversarial Attack of Adaptive Neural Networks15 Oct 2022
- 2022EmbryosFormer: Deformable Transformer and Collaborative Encoding-Decoding for Embryos Stage Development Classification7 Oct 2022
- 2022Task-adaptive Spatial-Temporal Video Sampler for Few-shot Action Recognition20 Jul 2022
- 2022TS2-Net: Token Shift and Selection Transformer for Text-Video Retrieval16 Jul 2022
- 2022SCAI: A Spectral data Classification framework with Adaptive Inference for the IoT platform24 Jun 2022
50 newest of 68 All 68 (JSON)
- 2023Token Sparsification for Faster Medical Image Segmentation11 Mar 2023
- 2023Self-Chained Image-Language Model for Video Localization and Question Answeringcited by 200
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunitiescited by 129
- 2022TS2-Net: Token Shift and Selection Transformer for Text-Video Retrievalcited by 104
- 2022CLIP-TSA: CLIP-Assisted Temporal Self-Attention for Weakly-Supervised Video Anomaly Detectioncited by 67
- 2021Efficient Video Transformers with Spatial-Temporal Token Selectioncited by 58
- 2022Differentiable Zooming for Multiple Instance Learning on Whole-Slide Imagescited by 43
- 2021Towards mental time travel: a hierarchical memory for reinforcement learning agentscited by 38
- 2024KVQ: Kwai Video Quality Assessment for Short-form Videoscited by 38
- 2022Vision Transformer with Attentive Pooling for Robust Facial Expression Recognitioncited by 35
- 2024Adaptive Slot Attention: Object Discovery with Dynamic Slot Numbercited by 34
- 2023Fast, Differentiable and Sparse Top-k: a Convex Analysis Perspectivecited by 26
- 2022Monotonic Differentiable Sorting Networkscited by 25
12 most cited of 64 All 64 (JSON)
- 2026LookThere! Sparse Vision by Reinforced Selection4 Sep 2026
- 2026A2: Smaller Self-Supervised ViTs Localize Better than Larger Ones2 Jun 2026
- 2026TT-Sparse: Learning Sparse Rule Models with Differentiable Truth Tables8 Mar 2026
- 2026Transformer-Based Inpainting for Real-Time 3D Streaming in Sparse Multi-Camera Setups5 Mar 2026
- 2026ORGAN: Object-Centric Representation Learning using Cycle Consistent Generative Adversarial Networks2 Mar 2026
- 2026WaveFormer: Wavelet Embedding Transformer for Biomedical Signals12 Feb 2026
- 2025EvoPS: Evolutionary Patch Selection for Whole Slide Image Analysis in Computational Pathology10 Nov 2025
- 2025DyWPE: Signal-Aware Dynamic Wavelet Positional Encoding for Time Series Transformers18 Sep 2025
- 2025LookWhere? Efficient Visual Recognition by Learning Where to Look and What to See from Self-Supervision23 May 2025
- 2025VEGAS: Towards Visually Explainable and Grounded Artificial Social Intelligence3 Apr 2025
- 2025Decision-aware training of spatiotemporal forecasting models to select a top K subset of sites for intervention7 Mar 2025
- 2025Energy-Efficient Transformer Inference: Optimization Strategies for Time Series Classification23 Feb 2025
12 newest of 64 All 64 (JSON)