Multi-label zero-shot learning with graph convolutional networks

Guangjin Ou, Guoxian Yu, Carlotta Domeniconi, Xuequan Lu, Xiangliang Zhang

Research output: Contribution to journalArticlepeer-review

32 Scopus citations

Abstract

The goal of zero-shot learning (ZSL) is to build a classifier that recognizes novel categories with no corresponding annotated training data. The typical routine is to transfer knowledge from seen classes to unseen ones by learning a visual-semantic embedding. Existing multi-label zero-shot learning approaches either ignore correlations among labels, suffer from large label combinations, or learn the embedding using only local or global visual features. In this paper, we propose a Graph Convolution Networks based Multi-label Zero-Shot Learning model, abbreviated as MZSL-GCN. Our model first constructs a label relation graph using label co-occurrences and compensates the absence of unseen labels in the training phase by semantic similarity. It then takes the graph and the word embedding of each seen (unseen) label as inputs to the GCN to learn the label semantic embedding, and to obtain a set of inter-dependent object classifiers. MZSL-GCN simultaneously trains another attention network to learn compatible local and global visual features of objects with respect to the classifiers, and thus makes the whole network end-to-end trainable. In addition, the use of unlabeled training data can reduce the bias toward seen labels and boost the generalization ability. Experimental results on benchmark datasets show that our MZSL-GCN competes with state-of-the-art approaches.
Original languageEnglish (US)
Pages (from-to)333-341
Number of pages9
JournalNeural Networks
Volume132
DOIs
StatePublished - Sep 21 2020

Fingerprint

Dive into the research topics of 'Multi-label zero-shot learning with graph convolutional networks'. Together they form a unique fingerprint.

Cite this