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

Prompt-based Learning for Unpaired Image Captioning

Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳));Peng Cheng Laboratory(鹏城实验室);Anhui University(安徽大学);Beijing Institute of Technology(北京理工大学);Sun Yat-Sen University(中山大学);The Hong Kong Polytechnic University(香港理工大学)

Peipei Zhu, Xiao Wang, Lin Zhu, Zhenglong Sun, Weishi Zheng, Yaowei Wang, Changwen Chen

英文摘要:

Unpaired Image Captioning (UIC) has been developed to learn image descriptions from unaligned vision-language sample pairs. Existing works usually tackle this task using adversarial learning and visual concept reward based on reinforcement learning. However, these existing works were only able to learn limited cross-domain information in vision and language domains, which restrains the captioning performance of UIC. Inspired by the success of Vision-Language Pre-Trained Models (VL-PTMs) in this research, we attempt to infer the cross-domain cue information about a given image from the large VL-PTMs for the UIC task. This research is also motivated by recent successes of prompt learning in many downstream multi-modal tasks, including image-text retrieval and vision question answering. In this work, a semantic prompt is introduced and aggregated with visual features for more accurate caption prediction under the adversarial learning framework. In addition, a metric prompt is designed to select high-quality pseudo image-caption samples obtained from the basic captioning model and refine the model in an iterative manner. Extensive experiments on the COCO and Flickr30K datasets validate the promising captioning ability of the proposed model. We expect that the proposed prompt-based UIC model will stimulate a new line of research for the VL-PTMs based captioning.

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

↑