journal article·2024Open access

Enhancing Image Description Generation through Deep Reinforcement Learning: Fusing Multiple Visual Features and Reward Mechanisms

Yan Li, Qiyuan Wang, Kaidi Jia

Computers, Materials & Continua· Volume 78· Issue 2

Abstract

Image description task is the intersection of computer vision and natural language processing, and it has important prospects, including helping computers understand images and obtaining information for the visually impaired. This study presents an innovative approach employing deep reinforcement learning to enhance the accuracy of natural language descriptions of images. Our method focuses on refining the reward function in deep reinforcement learning, facilitating the generation of precise descriptions by aligning visual and textual features more closely. Our approach comprises three key architectures. Firstly, it utilizes Residual Network 101 (ResNet-101) and Faster Region-based Convolutional Neural Network (Faster R-CNN) to extract average and local image features, respectively, followed by the implementation of a dual attention mechanism for intricate feature fusion. Secondly, the Transformer model is engaged to derive contextual semantic features from textual data. Finally, the generation of descriptive text is executed through a two-layer long short-term memory network (LSTM), directed by the value and reward functions. Compared with the image description method that relies on deep learning, the score of Bilingual Evaluation Understudy (BLEU-1) is 0.762, which is 1.6% higher, and the score of BLEU-4 is 0.299. Consensus-based Image Description Evaluation (CIDEr) scored 0.998, Recall-Oriented Understudy for Gisting Evaluation (ROUGE) scored 0.552, the latter improved by 0.36%. These results not only attest to the viability of our approach but also highlight its superiority in the realm of image description. Future research can explore the integration of our method with other artificial intelligence (AI) domains, such as emotional AI, to create more nuanced and context-aware systems.
Source: openalex_inverted_index · Quality: baseline

Research context

OpenAlex topics: Multimodal Machine Learning Applications · Domain Adaptation and Few-Shot Learning · Advanced Image and Video Retrieval Techniques

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

References (30)

Show source-provided references
1. An overview of image caption generation methods
2. Every picture tells a story: Generating sentences from images
3. Understanding and generating simple image descriptions10.1109/TPAMI.2012.162 ↗
4. Single-image dehazing based on two-stream convolutional neural network
5. A hybrid CNN for image denoising
6. Framing image description as a ranking task: Data, models and evaluation metrics10.1613/jair.3994 ↗
7. J. Mao, W. Xu, Y. Yang, J. Wang, Z. Huang and A. Yuille, “Deep captioning with multimodal recurrent neural networks (m-RNN),” arXiv preprint arXiv:1412.6632, 2014.
8. Image-denoising algorithm based on improved K-singular value decomposition and atom optimization10.1049/cit2.12044 ↗
9. Show and tell: A neural image caption generator
10. Show, attend and tell: Neural image caption generation with visual attention
11. Deep image retrieval using artificial neural network interpolation and indexing based on similarity measurement10.1049/cit2.12083 ↗
12. Show, control and tell: A framework for generating controllable and grounded captions
13. An image retrieval framework design analysis using saliency structure and color difference histogram10.3390/su141610357 ↗
14. Knowing when to look: Adaptive attention via a visual sentinel for image captioning
15. Stacked cross attention for image-text matching
16. A deep learning-based image captioning method to automatically generate comprehensive explanations of bridge damage10.1111/mice.12793 ↗
17. Deep residual learning for image recognition: A survey10.3390/app12188972 ↗
18. Survey on image classification algorithms based on deep residual network
19. Bottom-up and top-down attention for image captioning and visual question answering
20. Deep convolutional neural networks [Lecture Notes]10.1109/MSP.2018.2842646 ↗
21. Faster R-CNN: Towards real-time object detection with region proposal networks
22. Millimeter wave image object detection based on convolutional neural network
23. Guiding the long-short term memory model for image caption generation
24. X-linear attention networks for image captioning
25. D. P. Kingma, and J. Ba, “Adam: A method for stochastic optimization,” arXiv preprint arXiv:1412.6980, 2014.
26. Image captioning: Transforming objects into words
27. GSAN: Graph self-attention network for interaction measurement in autonomous driving
28. Attention is all you need
29. Image captioning with semantic attention
30. W. Liu, S. Chen, L. Guo, X. Zhu, and J. Liu, “CPTR: Full transformer network for image captioning,” arXiv preprint arXiv:2101.10804, 2021.