Enhancing image caption generation through context-aware attention mechanism

Ahatesham Bhuiyan, Eftekhar Hossain, Mohammed Moshiul Hoque, M. Ali Akber Dewan

Research output: Contribution to journalJournal Articlepeer-review

1 Citation (Scopus)

Abstract

Image captioning, the process of generating natural language descriptions based on image content, has garnered attention in AI research for its implications in scene understanding and human-computer interaction. While much prior research has focused on caption generation for English, addressing low-resource languages like Bengali presents challenges, particularly in producing coherent captions linking visual objects with corresponding words. This paper proposes a context-aware attention mechanism over semantic attention to accurately diagnose objects for image captioning in Bengali. The proposed architecture consists of an encoder and a decoder block. We chose ResNet-50 over the other pre-trained models for encoding the image features due to its ability to solve the vanishing gradient problem and recognize complex object features. For decoding generated captions, a bidirectional Gated Recurrent Unit (GRU) architecture combined with an attention mechanism captures contextual dependencies in both directions, resulting in more accurate captions. The paper also highlights the challenge of transferring knowledge between domains, especially with culturally specific images. Evaluation of three Bengali benchmark datasets, namely BAN-Cap, BanglaLekhaImageCaption, and Bornon, demonstrates significant performance improvement in METEOR score over existing methods by approximately 30%, 18%, and 45%, respectively. The proposed context-aware, attention-based image captioning system significantly outperforms current state-of-the-art models in Bengali caption generation despite limitations in reference captions on certain datasets.

Original languageEnglish
Article numbere36272
JournalHeliyon
Volume10
Issue number17
DOIs
Publication statusPublished - 15 Sep. 2024

Keywords

  • Attention mechanism
  • Computer vision
  • Cross-domain transfer
  • Encoder-decoder
  • Image captioning
  • Natural language processing

Fingerprint

Dive into the research topics of 'Enhancing image caption generation through context-aware attention mechanism'. Together they form a unique fingerprint.

Cite this