Capsule Graph Neural Network for Multi-Label Image Recognition (Student Abstract)
Xiangping Zheng, Xun Liang, Bo Wu
[AAAI-22] Student Abstract and Poster Program
Abstract:
This paper studies the problem of learning complex relationships between multi-labels for image recognition. Its challenges come from the rich and diverse semantic information in images. However, current methods cannot fully explore the mutual interactions among labels and do not explicitly model the label co-occurrence. To overcome these shortcomings, we innovatively propose CGML that consists of two crucial modules: 1) an image representation learning module that aims to complete the feature extraction of an image whose features are expressed in the form of primary capsules; 2) a label adaptive graph convolutional network module that leverages the popular graph convolutional networks with an adaptive label correlation graph to model label dependencies. Experiments show that our approach obviously outperforms the existing state-of-the-art methods.
Sessions where this paper appears
-
Poster Session 5
Sat, February 26 12:45 AM - 2:30 AM (+00:00)
Red 1
-
Poster Session 9
Sun, February 27 8:45 AM - 10:30 AM (+00:00)
Red 1