Abstract

The task of multi-label image recognition is to predict a set of object labels that present in an image. As objects normally co-occur in an image, it is desirable to model the label dependencies to improve the recognition performance. To capture and explore such important dependencies, we propose a multi-label classification model based on Graph Convolutional Network (GCN). The model builds a directed graph over the object labels, where each node (label) is represented by word embeddings of a label, and GCN is learned to map this label graph into a set of inter-dependent object classifiers. These classifiers are applied to the image descriptors extracted by another sub-net, enabling the whole network to be end-to-end trainable. Furthermore, we propose a novel re-weighted scheme to create an effective label correlation matrix to guide information propagation among the nodes in GCN. Experiments on two multi-label image recognition datasets show that our approach obviously outperforms other existing state-of-the-art methods. In addition, visualization analyses reveal that the classifiers learned by our model maintain meaningful semantic topology.

Keywords

Computer sciencePattern recognition (psychology)Artificial intelligenceGraphMulti-label classificationVisualizationCognitive neuroscience of visual object recognitionConvolutional neural networkSet (abstract data type)Contextual image classificationImage (mathematics)Feature extractionTheoretical computer science

Affiliated Institutions

Related Publications

Publication Info

Year
2019
Type
article
Pages
5172-5181
Citations
1170
Access
Closed

External Links

Social Impact

Social media, news, blog, policy document mentions

Citation Metrics

1170
OpenAlex

Cite This

Zhao-Min Chen, Xiu-Shen Wei, Peng Wang et al. (2019). Multi-Label Image Recognition With Graph Convolutional Networks. , 5172-5181. https://doi.org/10.1109/cvpr.2019.00532

Identifiers

DOI
10.1109/cvpr.2019.00532