计算机应用 ›› 2021, Vol. 41 ›› Issue (9): 2496-2503.DOI: 10.11772/j.issn.1001-9081.2020111829

所属专题: 人工智能

• 人工智能 • 上一篇    下一篇

图像特征注意力与自适应注意力融合的图像内容中文描述

赵宏, 孔东一   

  1. 兰州理工大学 计算机与通信学院, 兰州 730050
  • 收稿日期:2020-11-23 修回日期:2021-03-11 出版日期:2021-09-10 发布日期:2021-05-12
  • 通讯作者: 孔东一
  • 作者简介:赵宏(1971-),男,甘肃西和人,教授,博士,CCF会员,主要研究方向:并行与分布式处理、自然语言处理、深度学习;孔东一(1995-),男,河北石家庄人,硕士研究生,主要研究方向:图像内容描述、深度学习。
  • 基金资助:
    国家自然科学基金资助项目(51668043,61262016)。

Chinese description of image content based on fusion of image feature attention and adaptive attention

ZHAO Hong, KONG Dongyi   

  1. School of Computer and Communication, Lanzhou University of Technology, Lanzhou Gansu 730050, China
  • Received:2020-11-23 Revised:2021-03-11 Online:2021-09-10 Published:2021-05-12
  • Supported by:
    This work is partially supported by the National Natural Science Foundation of China (51668043, 61262016).

摘要: 针对现有基于注意力机制的图像内容中文描述模型无法在关注信息不减弱和无缺失的条件下对重点内容进行注意力加强关注的问题,提出一种图像特征注意力与自适应注意力融合的图像内容中文描述模型。模型使用编解码结构,首先在编码器网络中提取图像特征,并通过图像特征注意力提取图像全部特征区域的注意力信息;然后使用解码器网络将带有注意力权重的图像特征解码生成隐藏信息,以保证关注信息不减弱、无缺失;最后利用自适应注意力的视觉哨兵模块对图像特征中的重点内容进行再次加强关注,从而更加精准地提取图像的主体内容。使用多种评价指标(BLEU、METEOR、ROUGEL和CIDEr)进行模型验证,将所提模型与单一基于自适应注意力和基于图像特征注意力的图像描述模型进行对比实验,该模型的CIDEr评价指标值分别提高了10.1%和7.8%;同时与基线模型NIC(Neural Image Caption )以及基于自底向上和自顶向下(BUTD)注意力的图像描述模型相比,该模型的CIDEr评价指标值分别提高了10.9%和12.1%。实验结果表明,所提模型的图像理解能力得到了有效提升,其各项评价指标得分均优于对比模型。

关键词: 图像内容中文描述, 注意力机制, 深度学习, 卷积神经网络, 循环神经网络

Abstract: Aiming at the problem that the existing Chinese description models of image content based on attention mechanism cannot focus on the key content without weakening or missing attention information, a Chinese description model of image content based on fusion of image feature attention and adaptive attention was proposed. An encode-decode structure was used in this model. Firstly, the image features were extracted in the encoder network, and the attention information of all feature regions of the image was extracted by the image feature attention. Then, the decoder network was used to decode the image features with attention weights to generate hidden information, so as to ensure that the attention information was not weakened or missed. Finally, the visual sentry module of self-adaptive attention was used to focus on the key content in the image features again, so that the main content of the image was able to be extracted more accurately. Several evaluation indices including BLEU, METEOR, ROUGEL and CIDEr were used to verify the models, the proposed model was compared with the image description models based on self-adaptive attention or image feature attention only, and the proposed model had the evaluation value of CIDEr improved by 10.1% and 7.8% respectively. Meanwhile, compared with the baseline model Neural Image Caption (NIC) and the Bottom-Up and Top-Down (BUTD) attention based image description model, the proposed model had the evaluation index value of CIDEr increased by 10.9% and 12.1% respectively. Experimental results show that the image understanding ability of the proposed model is effectively improved, and the score of each evaluation index of the model is better than those of the comparison models.

Key words: Chinese description of image content, attention mechanism, deep learning, Convolutional Neural Network (CNN), Recurrent Neural Network (RNN)

中图分类号: