Attentive Linear Transformation for Image Captioning

作者:Ye, Senmao; Han, Junwei*; Liu, Nian
来源:IEEE Transactions on Image Processing, 2018, 27(11): 5514-5524.
DOI:10.1109/TIP.2018.2855406

摘要

We propose a novel attention framework called attentive linear transformation (ALT) for automatic generation of image captions. Instead of learning the spatial or channel-wise attention in existing models, ALT learns to attend to the high-dimensional transformation matrix from the image feature space to the context vector space. Thus ALT can learn various relevant feature abstractions, including spatial attention, channel-wise attention, and visual dependence. Besides, we propose a soft threshold regression to predict the spatial attention probabilities. It preserves more relevant local regions than popular softmax regression. Extensive experiments on the MS COCO and the Flickr30k data sets all demonstrate the superiority of our model compared with other state-of-the-art models.