Attentive Recurrent Neural Network for Weak-supervised Multi-label Image Classification

By Liang Li, Shuhui Wang, Shuqiang Jiang, Qingming Huang. 2018

39 citations recorded by the source corpus.

Multi-label image classification is a fundamental and challenging task in computer vision, and recently achieved significant progress by exploiting semantic relations among labels. However, the spatial positions of labels for multi-labels images are usually not provided in real scenarios, which brings insuperable barrier to conventional models. In this paper, we propose an end-to-end attentive recurrent neural network for multi-label image classification under only image-level supervision, which learns the discriminative feature representations and models the label relations simultaneously. First, inspired by attention mechanism, we propose a recurrent highlight network (RHN) which focuses on the most related regions in the image to learn the discriminative feature representations for different objects in an iterative manner. Second, we develop a gated recurrent relation extractor (GRRE) to model the label relations using multiplicative gates in a recurrent fashion, which learns to decide how multiple labels of the image influence the relation extraction. Extensive experiments on three benchmark datasets show that our model outperforms the state-of-the-arts, and performs better on small-object categories and under the scenario with large number of labels.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Michal Drozdzal
  2. Adriana Romero
  3. Zhengjun Zha
  4. Li Su
  5. Qingming Huang
  6. Shaofei Cai
  7. Luis Villaseñor Pineda
  8. Amaia Salvador

In Computer Vision and Pattern Recognition

  1. Michal Drozdzal
  2. Luis Villaseñor Pineda
  3. Adriana Romero
  4. Qingming Huang
  5. Amaia Salvador
  6. Zhengjun Zha
  7. Li Su
  8. Shaofei Cai

Citations in the corpus