GM-MLIC: Graph Matching based Multi-Label Image Classification

04/30/2021
by   Yanan Wu, et al.
0

Multi-Label Image Classification (MLIC) aims to predict a set of labels that present in an image. The key to deal with such problem is to mine the associations between image contents and labels, and further obtain the correct assignments between images and their labels. In this paper, we treat each image as a bag of instances, and reformulate the task of MLIC as an instance-label matching selection problem. To model such problem, we propose a novel deep learning framework named Graph Matching based Multi-Label Image Classification (GM-MLIC), where Graph Matching (GM) scheme is introduced owing to its excellent capability of excavating the instance and label relationship. Specifically, we first construct an instance spatial graph and a label semantic graph respectively, and then incorporate them into a constructed assignment graph by connecting each instance to all labels. Subsequently, the graph network block is adopted to aggregate and update all nodes and edges state on the assignment graph to form structured representations for each instance and label. Our network finally derives a prediction score for each instance-label correspondence and optimizes such correspondence with a weighted cross-entropy loss. Extensive experiments conducted on various image datasets demonstrate the superiority of our proposed method.

READ FULL TEXT

page 3

page 6

01/10/2019

GM-PLL: Graph Matching based Partial Label Learning

Partial Label Learning (PLL) aims to learn from the data where each trai...
06/26/2017

Multi-Label Learning with Label Enhancement

Multi-label learning deals with training instances associated with multi...
12/17/2019

Cross-Modality Attention with Semantic Graph Embedding for Multi-Label Classification

Multi-label image and video classification are fundamental yet challengi...
03/31/2018

Multi-label Learning with Missing Labels using Mixed Dependency Graphs

This work focuses on the problem of multi-label learning with missing la...
06/30/2015

Learning to Detect Blue-white Structures in Dermoscopy Images with Weak Supervision

We propose a novel approach to identify one of the most significant derm...
04/05/2019

A Regularization Approach for Instance-Based Superset Label Learning

Different from the traditional supervised learning in which each trainin...
10/30/2017

Prototype Matching Networks for Large-Scale Multi-label Genomic Sequence Classification

One of the fundamental tasks in understanding genomics is the problem of...