• Home
  • Search
  • End-to-End Transformer Based Model for Image Captioning
  • Cite Icon151
  • https://doi.org/10.1609/aaai.v36i3.20160Copy DOI Icon

End-to-End Transformer Based Model for Image Captioning

  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

CNN-LSTM based architectures have played an important role in image captioning, but limited by the training efficiency and expression ability, researchers began to explore the CNN-Transformer based models and achieved great success. Meanwhile, almost all recent works adopt Faster R-CNN as the backbone encoder to extract region-level features from given images. However, Faster R-CNN needs a pre-training on an additional dataset, which divides the image captioning task into two stages and limits its potential applications. In this paper, we build a pure Transformer-based model, which integrates image captioning into one stage and realizes end-to-end training. Firstly, we adopt SwinTransformer to replace Faster R-CNN as the backbone encoder to extract grid-level features from given images; Then, referring to Transformer, we build a refining encoder and a decoder. The refining encoder refines the grid features by capturing the intra-relationship between them, and the decoder decodes the refined features into captions word by word. Furthermore, in order to increase the interaction between multi-modal (vision and language) features to enhance the modeling capability, we calculate the mean pooling of grid features as the global feature, then introduce it into refining encoder to refine with grid features together, and add a pre-fusion process of refined global feature and generated words in decoder. To validate the effectiveness of our proposed model, we conduct experiments on MSCOCO dataset. The experimental results compared to existing published works demonstrate that our model achieves new state-of-the-art performances of 138.2% (single model) and 141.0% (ensemble of 4 models) CIDEr scores on 'Karpathy' offline test split and 136.0% (c5) and 138.3% (c40) CIDEr scores on the official online test server. Trained models and source code will be released.

Similar Papers
  • Research Article
  • Citations14

External knowledge-assisted Transformer for image captioning

  • Nov 07, 2023
  • Image and Vision Computing
  • Zhixin Li +2
  • Research Article
  • Citations67

Visuals to Text: A Comprehensive Review on Automatic Image Captioning

  • Aug 01, 2022
  • IEEE/CAA Journal of Automatica Sinica
  • Yue Ming +5
  • Research Article

Multi‐Gated Dual‐Stream Visual Feature Fusion for Image Captioning

  • Feb 28, 2026
  • Computer Graphics Forum
  • Yuzhe Lu +7
  • Conference Article
  • Citations2

CITE: Compact Interactive TransformEr for Multilingual Image Captioning

  • Jan 06, 2023
  • Yueyuan Xu +4
  • PDF
  • Research Article
  • Citations14

Boosting Memory with a Persistent Memory Mechanism for Remote Sensing Image Captioning

  • Jun 09, 2020
  • Remote Sensing
  • Kun Fu +4
  • PDF
  • Research Article
  • Citations28

Multi-Gate Attention Network for Image Captioning

  • Jan 01, 2021
  • IEEE Access
  • Weitao Jiang +4
  • PDF
  • Research Article
  • Citations3

Semantic-Enhanced Cross-Modal Fusion for Improved Unsupervised Image Captioning

  • Aug 22, 2023
  • Electronics
  • Nan Xiang +4
  • Conference Article
  • Citations39

DeeCap: Dynamic Early Exiting for Efficient Image Captioning

  • Jun 01, 2022
  • Zhengcong Fei +3
  • Research Article
  • Citations1

A review on image captioning system from artificial intelligence, machine learning and deep learning techniques

  • Jan 01, 2022
  • i-manager’s Journal on Image Processing
  • B S Revathi +1
  • Research Article
  • Citations93

High-Order Interaction Learning for Image Captioning

  • Jul 01, 2022
  • IEEE Transactions on Circuits and Systems for Video Technology
  • Yanhui Wang +4
  • Research Article
  • Citations44

A Hindi Image Caption Generation Framework Using Deep Learning

  • Mar 15, 2021
  • ACM Transactions on Asian and Low-Resource Language Information Processing
  • Santosh Kumar Mishra +3
  • Research Article
  • Citations11

A Benchmark for Feature-injection Architectures in Image Captioning

  • Dec 06, 2021
  • European Journal of Science and Technology
  • Rumeysa Keski̇n +4
  • Research Article
  • Citations31

A Deep Attention based Framework for Image Caption Generation in Hindi Language

  • Oct 07, 2019
  • Computación y Sistemas
  • Rijul Dhir +3
  • Conference Article
  • Citations719

X-Linear Attention Networks for Image Captioning

  • Jun 01, 2020
  • Yingwei Pan +3
  • Research Article
  • Citations79

Spatio-Temporal Memory Attention for Image Captioning

  • Jan 01, 2020
  • IEEE Transactions on Image Processing
  • Junzhong Ji +4
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.