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

Masked Non-Autoregressive Image Captioning

Peking University Shenzhen Graduate School(北京大学深圳研究生院);Nanjing University(南京大学);Peking University(北京大学);Peng Cheng Laboratory(鹏城实验室);City University of Hong Kong(香港城市大学)

Junlong Gao, Xi Meng, Shiqi Wang, Xia Li, Shanshe Wang, Siwei Ma, Wen Gao

英文摘要:

Existing captioning models often adopt the encoder-decoder architecture, where the decoder uses autoregressive decoding to generate captions, such that each token is generated sequentially given the preceding generated tokens. However, autoregressive decoding results in issues such as sequential error accumulation, slow generation, improper semantics and lack of diversity. Non-autoregressive decoding has been proposed to tackle slow generation for neural machine translation but suffers from multimodality problem due to the indirect modeling of the target distribution. In this paper, we propose masked non-autoregressive decoding to tackle the issues of both autoregressive decoding and non-autoregressive decoding. In masked non-autoregressive decoding, we mask several kinds of ratios of the input sequences during training, and generate captions parallelly in several stages from a totally masked sequence to a totally non-masked sequence in a compositional manner during inference. Experimentally our proposed model can preserve semantic content more effectively and can generate more diverse captions.

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

↑