VIVO: Surpassing Human Performance in Novel Object Captioning with Visual Vocabulary Pre-Training

By Xiaowei Hu, Xi Yin, Kevin Qinghong Lin, Lijuan Wang, L. Zhang, Jianfeng Gao, Zicheng Liu. 2020

58 citations recorded by the source corpus.

It is highly desirable yet challenging to generate image captions that can describe novel objects which are unseen in caption-labeled training data, a capability that is evaluated in the novel object captioning challenge (nocaps). In this challenge, no additional image-caption training data, other than COCO Captions, is allowed for model training. Thus, conventional Vision-Language Pre-training (VLP) methods cannot be applied. This paper presents VIsual VOcabulary pre-training (VIVO) that performs pre-training in the absence of caption annotations. By breaking the dependency of paired image-caption training data in VLP, VIVO can leverage large amounts of paired image-tag data to learn a visual vocabulary. This is done by pre-training a multi-layer Transformer model that learns to align image-level tags with their corresponding image region features. To address the unordered nature of image tags, VIVO uses a Hungarian matching loss with masked tag prediction to conduct pre-training. We validate the effectiveness of VIVO by fine-tuning the pre-trained model for image captioning. In addition, we perform an analysis of the visual-text alignment inferred by our model. The results show that our model can not only generate fluent image captions that describe novel objects, but also identify the locations of these objects. Our single model has achieved new state-of-the-art results on nocaps and surpassed the human CIDEr score.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Multimodal Learning

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Zhe Gan
  2. Jianfeng Gao
  3. Lijuan Wang
  4. Zicheng Liu
  5. Jianfeng Wang
  6. Linjie Li
  7. Xiaowei Hu
  8. Mohit Bansal

In Computer Vision and Pattern Recognition

  1. Jianfeng Gao
  2. Zhe Gan
  3. Luc Van Gool
  4. Dacheng Tao
  5. Lijuan Wang
  6. Zicheng Liu
  7. Cordelia Schmid
  8. Trevor Darrell

In Multimodal Learning

  1. Zhe Gan
  2. Lijuan Wang
  3. Zicheng Liu
  4. Jianfeng Wang
  5. Xiaowei Hu
  6. Jianfeng Gao
  7. Linjie Li
  8. Zhengyuan Yang

Citations in the corpus