arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.01473cs.CLcs.CVcs.LG

XGPT: Cross-modal Generative Pre-Training for Image Captioning

MOE Key Laboratory of Computational Linguistics, Peking University(北京大学计算语言学教育部重点实验室);Microsoft Research Asia(微软亚洲研究院);Microsoft(微软)

Qiaolin Xia, Haoyang Huang, Nan Duan, Dongdong Zhang, Lei Ji, Zhifang Sui, Edward Cui, Taroon Bharti, Xin Liu, Ming Zhou

英文摘要:

While many BERT-based cross-modal pre-trained models produce excellent results on downstream understanding tasks like image-text retrieval and VQA, they cannot be applied to generation tasks directly. In this paper, we propose XGPT, a new method of Cross-modal Generative Pre-Training for Image Captioning that is designed to pre-train text-to-image caption generators through three novel generation tasks, including Image-conditioned Masked Language Modeling (IMLM), Image-conditioned Denoising Autoencoding (IDA), and Text-conditioned Image Feature Generation (TIFG). As a result, the pre-trained XGPT can be fine-tuned without any task-specific architecture modifications to create state-of-the-art models for image captioning. Experiments show that XGPT obtains new state-of-the-art results on the benchmark datasets, including COCO Captions and Flickr30k Captions. We also use XGPT to generate new image captions as data augmentation for the image retrieval task and achieve significant improvement on all recall metrics.

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

↑