An image caption model based on attention mechanism and deep reinforcement learning

Frontiers in Neuroscience

Neuroscience

Sen Zhou¹  Ya Du²  Jiasai Luo³  Yu Pang³  Tong Bai³  Huiqian Wang³ 

[1] Chongqing Academy of Metrology and Quality Inspection, Chongqing, China;Department of Peripheral Vascular (Wound Repair), Chongqing Hospital of Traditional Chinese Medicine, Chongqing, China;School of Optoelectronic Engineering, Chongqing University of Posts and Telecommunications, Chongqing, China;

关键词: image caption;    encoder-decoder architecture;    deep neural networks;    attention mechanism;    deep reinforcement learning;   

DOI  :  10.3389/fnins.2023.1270850

 received in 2023-08-01, accepted in 2023-09-04,  发布年份 2023

来源: Frontiers

【 摘 要 】

Image caption technology aims to convert visual features of images, extracted by computers, into meaningful semantic information. Therefore, the computers can generate text descriptions that resemble human perception, enabling tasks such as image classification, retrieval, and analysis. In recent years, the performance of image caption has been significantly enhanced with the introduction of encoder-decoder architecture in machine translation and the utilization of deep neural networks. However, several challenges still persist in this domain. Therefore, this paper proposes a novel method to address the issue of visual information loss and non-dynamic adjustment of input images during decoding. We introduce a guided decoding network that establishes a connection between the encoding and decoding parts. Through this connection, encoding information can provide guidance to the decoding process, facilitating automatic adjustment of the decoding information. In addition, Dense Convolutional Network (DenseNet) and Multiple Instance Learning (MIL) are adopted in the image encoder, and Nested Long Short-Term Memory (NLSTM) is utilized as the decoder to enhance the extraction and parsing capability of image information during the encoding and decoding process. In order to further improve the performance of our image caption model, this study incorporates an attention mechanism to focus details and constructs a double-layer decoding structure, which facilitates the enhancement of the model in terms of providing more detailed descriptions and enriched semantic information. Furthermore, the Deep Reinforcement Learning (DRL) method is employed to train the model by directly optimizing the identical set of evaluation indexes, which solves the problem of inconsistent training and evaluation standards. Finally, the model is trained and tested on MS COCO and Flickr 30 k datasets, and the results show that the model has improved compared with commonly used models in the evaluation indicators such as BLEU, METEOR and CIDEr.

【 授权许可】

Unknown   
Copyright © 2023 Bai, Zhou, Pang, Luo, Wang and Du.