Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Learning Semantic-Specific Graph Representation for Multi-Label Image Recognition

About

Recognizing multiple labels of images is a practical and challenging task, and significant progress has been made by searching semantic-aware regions and modeling label dependency. However, current methods cannot locate the semantic regions accurately due to the lack of part-level supervision or semantic guidance. Moreover, they cannot fully explore the mutual interactions among the semantic regions and do not explicitly model the label co-occurrence. To address these issues, we propose a Semantic-Specific Graph Representation Learning (SSGRL) framework that consists of two crucial modules: 1) a semantic decoupling module that incorporates category semantics to guide learning semantic-specific representations and 2) a semantic interaction module that correlates these representations with a graph built on the statistical label co-occurrence and explores their interactions via a graph propagation mechanism. Extensive experiments on public benchmarks show that our SSGRL framework outperforms current state-of-the-art methods by a sizable margin, e.g. with an mAP improvement of 2.5%, 2.6%, 6.7%, and 3.1% on the PASCAL VOC 2007 & 2012, Microsoft-COCO and Visual Genome benchmarks, respectively. Our codes and models are available at https://github.com/HCPLab-SYSU/SSGRL.

Tianshui Chen, Muxin Xu, Xiaolu Hui, Hefeng Wu, Liang Lin• 2019

Related benchmarks

TaskDatasetResultRank
Multi-Label ClassificationPASCAL VOC 2007 (test)
mAP95
125
Multi-Label ClassificationMS-COCO 2014 (test)
mAP83.8
81
Multi-label Image ClassificationVOC 2012 (test)
mAP95.4
72
Multi-label recognitionMS-COCO
Overall F1 Score (OF1)77.2
66
Multi-label recognitionPASCAL VOC 2007
Avg OF189.4
66
Multi-label recognitionVG-200
Avg OF140.1
66
Multi-label image recognitionVOC 2007 (test)
mAP95
61
Multi-Label ClassificationVOC 07
mAP89.5
61
Multi-label image recognitionMS-COCO 2014 (val)
mAP83.8
51
Multi-Label ClassificationMS-COCO (val)
mAP83.8
47
Showing 10 of 35 rows

Other info

Code

Follow for update