Graph-based Visual-Semantic Entanglement Network for Zero-shot Image Recognition

TL;DR

This paper introduces a graph-based neural network that enhances zero-shot image recognition by modeling semantic relationships through a knowledge graph, improving the connection between visual features and semantic attributes.

Contribution

It proposes a novel multi-path entangled network combining CNN and GCN with attribute word vectors for improved semantic modeling in zero-shot learning.

Findings

Outperforms state-of-the-art on AwA2, CUB, and SUN datasets.

Enhances semantic linkage modeling of visual features.

Improves zero-shot recognition accuracy.

Abstract

Zero-shot learning uses semantic attributes to connect the search space of unseen objects. In recent years, although the deep convolutional network brings powerful visual modeling capabilities to the ZSL task, its visual features have severe pattern inertia and lack of representation of semantic relationships, which leads to severe bias and ambiguity. In response to this, we propose the Graph-based Visual-Semantic Entanglement Network to conduct graph modeling of visual features, which is mapped to semantic attributes by using a knowledge graph, it contains several novel designs: 1. it establishes a multi-path entangled network with the convolutional neural network (CNN) and the graph convolutional network (GCN), which input the visual features from CNN to GCN to model the implicit semantic relations, then GCN feedback the graph modeled information to CNN features; 2. it uses attribute word vectors as the target for the graph semantic modeling of GCN, which forms a self-consistent regression for graph modeling and supervise GCN to learn more personalized attribute relations; 3. it fuses and supplements the hierarchical visual-semantic features refined by graph modeling into visual embedding. Our method outperforms state-of-the-art approaches on multiple representative ZSL datasets: AwA2, CUB, and SUN by promoting the semantic linkage modelling of visual features.