arrow
Return

Modeling graph-structured contexts for image captioning

delete2023-01-01
delete15
PRE
AI
李志新 cover
李志新 (Zhixin Li) *
J
Jiahui Wei
F
Feicheng Huang
马慧芳 (Huifang Ma)
DOI:10.1016/j.imavis.2022.104591delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
The performance of image captioning has been significantly improved recently through deep neural network ar-chitectures combining with attention mechanisms and reinforcement learning optimization. Exploring visual re-lationships and interactions between different objects appearing in the image, however, is far from being investigated. In this paper, we present a novel approach that combines scene graphs with Transformer, which we call SGT, to explicitly encode available visual relationships between detected objects. Specifically, we pretrain an scene graph generation model to predict graph representations for images. After that, for each graph node, a Graph Convolutional Network (GCN) is employed to acquire relationship knowledge by aggregating the informa-tion of its local neighbors. As we train the captioning model, we feed the potential relation-aware information into the Transformer to generate descriptive sentence. Experiments on the MSCOCO dataset and the Flickr30k dataset validate the superiority of our SGT model, which can realize state-of-the-art results in terms of all the standard evaluation metrics.(c) 2022 Elsevier B.V. All rights reserved.
Keywords:
Image captioning
Transformer
Scene graph
Reinforcement learning
Attention mechanism

Journal

Image and Vision Computing cover
Image and Vision Computing
IF:
4.2
Papers:
4.0K
Citations:
6.7K

Organization

G
Guangxi Normal University
Scholars:
7.7K
Papers: 4.9K
Citations: 5.1K
N
northwest normal university - china
Scholars:
7.8K
Papers: 4.8K
Citations: 4