arXiv:2501.14828cs.CVcs.AI2025-01被引 26

用注意力机制的集成模型提升图像描述生成质量

An Ensemble Model with Attention Based Mechanism for Image Captioning

  • 采用变压器编码器-解码器结构,结合卷积神经网络提取图像特征
  • 通过投票机制选择高BLEU分数的生成结果,在Flickr8K和Flickr30k上领先
  • 集成学习显著提升描述丰富性,适合多领域图像理解应用

图像字幕生成通过建立词语与图像内容之间的关联,为输入图像生成描述性文本。近年来,基于变压器的深度学习模型在自动图像字幕生成中表现最佳。本文深入研究变压器模型,强调注意力机制的关键作用。所提模型采用变压器编码器-解码器架构生成文本字幕,并利用深度卷积神经网络提取图像特征。为生成更丰富的字幕,提出一种新型集成学习框架,基于投票机制选取具有最高双语评估理解(BLEU)得分的字幕。在公开数据集上进行评估:在Flickr8K数据集上,该模型取得BLEU-[1-3]分别为0.728、0.495、0.323的最高得分;在Flickr30k数据集上,BLEU-[1-4]得分分别为0.798、0.561、0.387、0.269,优于最新方法。同时,该模型在语义命题图像字幕评估(SPICE)指标上分别达到Flickr8K的0.164和Flickr30k的0.387。实验证明,集成学习显著推动了图像字幕生成的发展,可在多个领域广泛应用。

原文摘要 · Abstract (English)

Image captioning creates informative text from an input image by creating a relationship between the words and the actual content of an image. Recently, deep learning models that utilize transformers have been the most successful in automatically generating image captions. The capabilities of transformer networks have led to notable progress in several activities related to vision. In this paper, we thoroughly examine transformer models, emphasizing the critical role that attention mechanisms play. The proposed model uses a transformer encoder-decoder architecture to create textual captions and a deep learning convolutional neural network to extract features from the images. To create the captions, we present a novel ensemble learning framework that improves the richness of the generated captions by utilizing several deep neural network architectures based on a voting mechanism that chooses the caption with the highest bilingual evaluation understudy (BLEU) score. The proposed model was evaluated using publicly available datasets. Using the Flickr8K dataset, the proposed model achieved the highest BLEU-[1-3] scores with rates of 0.728, 0.495, and 0.323, respectively. The suggested model outperformed the latest methods in Flickr30k datasets, determined by BLEU-[1-4] scores with rates of 0.798, 0.561, 0.387, and 0.269, respectively. The model efficacy was also obtained by the Semantic propositional image caption evaluation (SPICE) metric with a scoring rate of 0.164 for the Flicker8k dataset and 0.387 for the Flicker30k. Finally, ensemble learning significantly advances the process of image captioning and, hence, can be leveraged in various applications across different domains.

图像描述注意力机制集成学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。