HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

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

Chen Tianshui ; Xu Muxin ; Hui Xiaolu ; Wu Hefeng ; Lin Liang

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

Abstract

Recognizing multiple labels of images is a practical and challenging task,and significant progress has been made by searching semantic-aware regions andmodeling label dependency. However, current methods cannot locate the semanticregions accurately due to the lack of part-level supervision or semanticguidance. Moreover, they cannot fully explore the mutual interactions among thesemantic regions and do not explicitly model the label co-occurrence. Toaddress these issues, we propose a Semantic-Specific Graph RepresentationLearning (SSGRL) framework that consists of two crucial modules: 1) a semanticdecoupling module that incorporates category semantics to guide learningsemantic-specific representations and 2) a semantic interaction module thatcorrelates these representations with a graph built on the statistical labelco-occurrence and explores their interactions via a graph propagationmechanism. Extensive experiments on public benchmarks show that our SSGRLframework 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 VOC2007 & 2012, Microsoft-COCO and Visual Genome benchmarks, respectively. Ourcodes and models are available at https://github.com/HCPLab-SYSU/SSGRL.

Code Repositories

HCPLab-SYSU/SSGRL
Official
pytorch
Mentioned in GitHub
ZFT-CQU/DSDL
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
multi-label-classification-on-pascal-voc-2007SSGRL (pretrain from MS-COCO)
mAP: 95.0
multi-label-classification-on-pascal-voc-2007SSGRL (pretrain from ImageNet)
mAP: 93.4

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp