Short-term memory with read-only unit in neural image caption generator

Abstract

Automated caption generation for digital images is one of the fundamental problems in artificial intelligence. Most of the existing works use 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 with 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.