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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multi-Label Classification | PASCAL VOC 2007 (test) | mAP95 | 125 | |
| Multi-Label Classification | MS-COCO 2014 (test) | mAP83.8 | 81 | |
| Multi-label Image Classification | VOC 2012 (test) | mAP95.4 | 72 | |
| Multi-label recognition | MS-COCO | Overall F1 Score (OF1)77.2 | 66 | |
| Multi-label recognition | PASCAL VOC 2007 | Avg OF189.4 | 66 | |
| Multi-label recognition | VG-200 | Avg OF140.1 | 66 | |
| Multi-label image recognition | VOC 2007 (test) | mAP95 | 61 | |
| Multi-Label Classification | VOC 07 | mAP89.5 | 61 | |
| Multi-label image recognition | MS-COCO 2014 (val) | mAP83.8 | 51 | |
| Multi-Label Classification | MS-COCO (val) | mAP83.8 | 47 |