Learning to Zoom: a Saliency-Based Sampling Layer for Neural Networks

Abstract

We introduce a saliency-based distortion layer for convolutional neural networks that helps to improve the spatial sampling of input data for a given task. Our differentiable layer can be added as a preprocessing block to existing task networks and trained altogether in an end-to-end fashion. The effect of the layer is to efficiently estimate how to sample from the original data in order to boost task performance. For example, for an image classification task in which the original data might range in size up to several megapixels, but where the desired input images to the task network are much smaller, our layer learns how best to sample from the underlying high resolution data in a manner which preserves task-relevant information better than uniform downsampling. This has the effect of creating distorted, caricature-like intermediate images, in which idiosyncratic elements of the image that improve task performance are zoomed and exaggerated. Unlike alternative approaches such as spatial transformer networks, our proposed layer is inspired by image saliency, computed efficiently from uniformly downsampled data, and degrades gracefully to a uniform sampling strategy under uncertainty. We apply our layer to improve existing networks for the tasks of human gaze estimation and fine-grained object classification. Code for our method is available in: http://github.com/recasens/Saliency-Sampler

arXiv commentEuropean Conference on Computer Vision, 2018, 14 pages, 6 figures

References in corpus (11)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size
• Recurrent Models of Visual Attention
• Multiple Object Recognition with Visual Attention
• Deformable Convolutional Networks
• Attend, Infer, Repeat: Fast Scene Understanding with Generative Models
• Eye Tracking for Everyone
• Dynamic Computational Time for Visual Attention
• Dense Transformer Networks
• The iNaturalist Species Classification and Detection Dataset
• Visual Concept Recognition and Localization via Iterative Introspection

Cited by in corpus (6)

• Streaming convolutional neural networks for end-to-end learning with multi-megapixel images
• Dynamic Neural Networks: A Survey
• Looking for the Devil in the Details: Learning Trilinear Attention Sampling Network for Fine-grained Image Recognition
• SegBlocks: Block-Based Dynamic Resolution Networks for Real-Time Segmentation
• Rapid Elastic Architecture Search under Specialized Classes and Resource Constraints
• Cost-Aware Fine-Grained Recognition for IoTs Based on Sequential Fixations