arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1908.06954cs.CV

Attention on Attention for Image Captioning

Peking University(北京大学);Macau University of Science and Technology(澳门科技大学);Peng Cheng Laboratory(鹏城实验室)

Lun Huang, Wenmin Wang, Jie Chen, Xiao-Yong Wei

英文摘要:

Attention mechanisms are widely used in current encoder/decoder frameworks of image captioning, where a weighted average on encoded vectors is generated at each time step to guide the caption decoding process. However, the decoder has little idea of whether or how well the attended vector and the given attention query are related, which could make the decoder give misled results. In this paper, we propose an Attention on Attention (AoA) module, which extends the conventional attention mechanisms to determine the relevance between attention results and queries. AoA first generates an information vector and an attention gate using the attention result and the current context, then adds another attention by applying element-wise multiplication to them and finally obtains the attended information, the expected useful knowledge. We apply AoA to both the encoder and the decoder of our image captioning model, which we name as AoA Network (AoANet). Experiments show that AoANet outperforms all previously published methods and achieves a new state-of-the-art performance of 129.8 CIDEr-D score on MS COCO Karpathy offline test split and 129.6 CIDEr-D (C40) score on the official online testing server. Code is available at https://github.com/husthuaan/AoANet.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑