vix.ing · top · new · best · stats · spec

Where to Look: A Unified Attention Model for Visual Recognition with Reinforcement Learning

2021/11/13 by Gang Chen, Chen, Gang
Computer Science · #68T01 #Advanced Neural Network Applications #Computer Vision and Pattern Recognition (cs.CV) #Domain Adaptation and Few-Shot Learning #FOS: Computer and information sciences #I.2.9 #Visual Attention and Saliency Detection

paper · pdf · doi:10.48550/arxiv.2111.07169

openalex publication_date 2021/11/13 · openalex created_date 2021/11/22 · openalex updated_date 2026/07/28

Abstract

The idea of using the recurrent neural network for visual attention has gained popularity in computer vision community. Although the recurrent attention model (RAM) leverages the glimpses with more large patch size to increasing its scope, it may result in high variance and instability. For example, we need the Gaussian policy with high variance to explore object of interests in a large image, which may cause randomized search and unstable learning. In this paper, we propose to unify the top-down and bottom-up attention together for recurrent visual attention. Our model exploits the image pyramids and Q-learning to select regions of interests in the top-down attention mechanism, which in turn to guide the policy search in the bottom-up approach. In addition, we add another two constraints over the bottom-up recurrent neural networks for better exploration. We train our model in an end-to-end reinforcement learning framework, and evaluate our method on visual classification tasks. The experimental results outperform convolutional neural networks (CNNs) baseline and the bottom-up recurrent attention models on visual classification tasks.

Citations

Related