Automated caption generation for digital images is one of the fundamental problems in artificial intelligence. Most of the existing works use LSTM (Long Short-Term Memory) as a recurrent neural network cell to solve this task. After training, their deep neural models can generate an image caption. But there is an issue, the next predicted word of the caption depends mainly on the last predicted word, rather than the image content. In this paper, we present model that can automatically generate an image description and is based on a recurrent neural network with modified LSTM cell that has an additional gate responsible for image features. This modification results in generation of more accurate captions. We have trained and tested our model on MSCOCO image dataset by using only images and their captions.
oai:noad.sci.am:135815
agasy18@gmail.com ; hakop@ipia.sci.am
Institute for Informatics and Automation Problems
11th International Conference on Computer Science and Information Technologies CSIT 2017
Mar 3, 2021
Jul 17, 2020
17
https://noad.sci.am/publication/149342
Edition name | Date |
---|---|
Aghasi, Poghosyan, Long Short-Term Memory with Read-only Unit in NeuralImage Caption Generator | Mar 3, 2021 |
Poghosyan Aghasi
Poghosyan Aghasi Sarukhanyan Hakob
Poghosyan Aghasi Hakob Sarukhanyan