ViLBERT: Pretraining Task-Agnostic Visiolinguistic Representations for Vision-and-Language Tasks
Written for the Research library (claude-opus-5-5) from the paper text, in our own words: the license "arXiv — non-exclusive licence to distribute" does not allow redistributing the text itself. Original — arxiv.org/pdf/1908.02265
Summary
Vision-and-language systems usually learn to connect words with image content only while training on a specific task. They start from separately pretrained vision and language models, so the grounding they learn is often narrow and generalizes poorly when paired data is scarce or biased. ViLBERT addresses this by pretraining visual grounding as a reusable capability. It extends BERT into a two-stream design: one transformer stream processes image region features and another processes text. The streams exchange information through co-attentional transformer layers, where each modality attends using the other's keys and values. The model is pretrained on about 3.3 million web image–caption pairs from Conceptual Captions with two self-supervised objectives: reconstructing masked words and region categories, and deciding whether a caption matches an image. It is then fine-tuned for four downstream tasks, mostly by adding a small task-specific head. It set new state-of-the-art results on all four, with gains of roughly 2–10 points over task-specific models. This marks a move toward general-purpose pretrained visiolinguistic representations.
Key points
- Architecture: separate visual and linguistic streams, which can have different depths, are linked by Co-TRM layers. In the authors' experiments this beat a single-stream BERT variant that treats image regions like tokens.
- Pretraining: Conceptual Captions (about 3.3M images with alt-text captions), using two objectives: masked multi-modal modelling and image–text alignment prediction.
- Transfer results (full pretraining): VQA test-dev 70.55; VCR Q→A 72.42, QA→R 74.47, Q→AR 54.04; RefCOCO+ val 72.34; Flickr30k image retrieval R@1 58.20.
- Zero-shot retrieval: without fine-tuning, Flickr30k R@1 is 31.86. The fine-tuned model reaches 58.20 and the prior SOTA was 48.60, so the pretrained alignment is already meaningful on its own.
- Data scaling: downstream scores rise steadily as more of Conceptual Captions is used for pretraining (0% → 25% → 50% → 100%). For example, VQA improves from 68.93 to 70.55.
Cited by
- Do As I Can, Not As I Say: Grounding Language in Robotic Affordances
- Flamingo: a Visual Language Model for Few-Shot Learning
- Experience Grounds Language
- On the Opportunities and Risks of Foundation Models
- UNITER: UNiversal Image-TExt Representation Learning
- Evaluating Large Language Models Trained on Code
- CodeBERT: A Pre-Trained Model for Programming and Natural Languages
- TaPas: Weakly Supervised Table Parsing via Pre-training
- CLIPScore: A Reference-free Evaluation Metric for Image Captioning
- LayoutLMv3: Pre-training for Document AI with Unified Text and Image Masking
- The Hateful Memes Challenge: Detecting Hate Speech in Multimodal Memes
- ViLT: Vision-and-Language Transformer Without Convolution or Region Supervision
- VisualBERT: A Simple and Performant Baseline for Vision and Language
- Oscar: Object-Semantics Aligned Pre-training for Vision-Language Tasks
- BLIP: Bootstrapping Language-Image Pre-training for Unified Vision-Language Understanding and Generation
- XGLUE: A New Benchmark Dataset for Cross-lingual Pre-training, Understanding and Generation
- Pre-train, Prompt, and Predict: A Systematic Survey of Prompting Methods in Natural Language Processing
- COMICORDA: Dialogue Act Recognition in Comic Books
- Text and Code Embeddings by Contrastive Pre-Training
- Learning Transferable Visual Models From Natural Language Supervision
- A-OKVQA: A Benchmark for Visual Question Answering using World Knowledge
- ALFWorld: Aligning Text and Embodied Environments for Interactive Learning
- VL-BERT: Pre-training of Generic Visual-Linguistic Representations
- LXMERT: Learning Cross-Modality Encoder Representations from Transformers
- Long Range Arena: A Benchmark for Efficient Transformers
- Training data-efficient image transformers & distillation through attention
- Multimodal Few-Shot Learning with Frozen Language Models
- SimVLM: Simple Visual Language Model Pretraining with Weak Supervision
- OFA: Unifying Architectures, Tasks, and Modalities Through a Simple Sequence-to-Sequence Learning Framework
- LayoutLMv2: Multi-modal Pre-training for Visually-rich Document Understanding
- The Dawn of LMMs: Preliminary Explorations with GPT-4V(ision)
- MM-Vet: Evaluating Large Multimodal Models for Integrated Capabilities
- MMMU: A Massive Multi-discipline Multimodal Understanding and Reasoning Benchmark for Expert AGI
- Socratic Models: Composing Zero-Shot Multimodal Reasoning with Language
Cite
@misc{lu2019vilbert,
title = {{ViLBERT: Pretraining Task-Agnostic Visiolinguistic Representations for Vision-and-Language Tasks}},
author = {Lu, Jiasen and Batra, Dhruv and Parikh, Devi and Lee, Stefan},
year = {2019},
doi = {10.48550/arxiv.1908.02265},
eprint = {1908.02265},
archivePrefix = {arXiv},
url = {https://doi.org/10.48550/arxiv.1908.02265}
}References
- Margaret A. Boden. Mind as Machine: A History of Cognitive Science. Oxford University Press, 2008. find
- Sahar Kazemzadeh, Vicente Ordonez, Mark Matten, and Tamara L. Berg. Referit game: Referring to objects in photographs of natural scenes. In EMNLP, 2014. find
- Stanislaw Antol, Aishwarya Agrawal, Jiasen Lu, Margaret Mitchell, Dhruv Batra, C. Lawrence Zitnick, and Devi Parikh. VQA: Visual question answering. In ICCV, 2015. find
- Abhishek Das, Satwik Kottur, Khushi Gupta, Avi Singh, Deshraj Yadav, Jose M. F. Moura, Devi Parikh, and Dhruv Batra. Visual dialog. In CVPR, 2017. find
- Xinlei Chen, Hao Fang, Tsung-Yi Lin, Ramakrishna Vedantam, Saurabh Gupta, Piotr Dollár, and C. Lawrence Zitnick. Microsoft COCO captions: Data collection and evaluation server. CoRR, abs/1504.00325, 2015. find our card
- Ravi Shekhar, Sandro Pezzelle, Yauhen Klimovich, Aurelie Herbelot, Moin Nabi, Enver Sangineto, and Raffaella Bernardi. "foil it! find one mismatch between image and language caption". In ACL, 2017. find
- Abhishek Das, Samyak Datta, Georgia Gkioxari, Stefan Lee, Devi Parikh, and Dhruv Batra. Embodied Question Answering. In CVPR, 2018. find
- Peter Anderson, Qi Wu, Damien Teney, Jake Bruce, Mark Johnson, Niko Sünderhauf, Ian Reid, Stephen Gould, and Anton van den Hengel. Vision-and-language navigation: Interpreting visually-grounded navigation instructions in real environments. In CVPR), 2018. find our card
- Aishwarya Agrawal, Dhruv Batra, Devi Parikh, and Aniruddha Kembhavi. Don’t just assume; look and answer: Overcoming priors for visual question answering. In CVPR, 2018. find
- Harsh Agrawal, Karan Desai, Xinlei Chen, Rishabh Jain, Dhruv Batra, Devi Parikh, Stefan Lee, and Peter Anderson. nocaps: novel object captioning at scale. arXiv preprint arXiv:1812.08658, 2018. — [arXiv:1812.08658](https://arxiv.org/abs/1812.08658) find
- Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. Deep residual learning for image recognition. In CVPR, 2016. find
- Jacob Devlin, Ming-Wei Chang, Kenton Lee, and Kristina Toutanova. Bert: Pre-training of deep bidirectional transformers for language understanding. arXiv preprint arXiv:1810.04805, 2018. find
- Matthew E. Peters, Mark Neumann, Mohit Iyyer, Matt Gardner, Christopher Clark, Kenton Lee, and Luke Zettlemoyer. Deep contextualized word representations. In NACCL, 2018. find
- Alec Radford, Karthik Narasimhan, Time Salimans, and Ilya Sutskever. Improving language understanding with unsupervised learning. Technical report, Technical report, OpenAI, 2018. find
- Olga Russakovsky, Jia Deng, Hao Su, Jonathan Krause, Sanjeev Satheesh, Sean Ma, Zhiheng Huang, Andrej Karpathy, Aditya Khosla, Michael Bernstein, Alexander C. Berg, and Li Fei-Fei. Imagenet large scale visual recognition challenge. IJCV, 2015. find our card
- Ranjay Krishna, Yuke Zhu, Oliver Groth, Justin Johnson, Kenji Hata, Joshua Kravitz, Stephanie Chen, Yannis Kalantidis, Li-Jia Li, David A Shamma, Michael Bernstein, and Li Fei-Fei. Visual genome: Connecting language and vision using crowdsourced dense image annotations. In arXiv, 2016. URL https://a… find our card
- Yukun Zhu, Ryan Kiros, Rich Zemel, Ruslan Salakhutdinov, Raquel Urtasun, Antonio Torralba, and Sanja Fidler. Aligning books and movies: Towards story-like visual explanations by watching movies and reading books. In ICCV, 2015. find our card
- English wikipedia, 2019. URL https://en.wikipedia.org/. ↗
- Ciprian Chelba, Tomas Mikolov, Mike Schuster, Qi Ge, Thorsten Brants, Phillipp Koehn, and Tony Robinson. One billion word benchmark for measuring progress in statistical language modeling. In arXiv, 2014. find our card
- Gustav Larsson, Michael Maire, and Gregory Shakhnarovich. Colorization as a proxy task for visual understanding. In CVPR, pages 6874–6883, 2017. find
- Dinesh Jayaraman, Ruohan Gao, and Kristen Grauman. Shapecodes: self-supervised feature learning by lifting views to viewgrids. In ECCV, pages 120–136, 2018. find
- Relja Arandjelovic and Andrew Zisserman. Look, listen and learn. In ICCV, pages 609–617, 2017. find
- Deepak Pathak, Ross Girshick, Piotr Dollár, Trevor Darrell, and Bharath Hariharan. Learning features by watching objects move. In CVPR, pages 2701–2710, 2017. find
- Piyush Sharma, Nan Ding, Sebastian Goodman, and Radu Soricut. Conceptual captions: A cleaned, hypernymed, image alt-text dataset for automatic image captioning. In ACL, 2018. find our card
- Rowan Zellers, Yonatan Bisk, Ali Farhadi, and Yejin Choi. From recognition to cognition: Visual commonsense reasoning. In CVPR, 2019. find
- Peter Young, Alice Lai, Micah Hodosh, and Julia Hockenmaier. From image descriptions to visual denotations: New similarity metrics for semantic inference over event descriptions. TACL, 2014. find
- Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N Gomez, Łukasz Kaiser, and Illia Polosukhin. Attention is all you need. In NeurIPS, 2017. find
- Yonghui Wu, Mike Schuster, Zhifeng Chen, Quoc V Le, Mohammad Norouzi, Wolfgang Macherey, Maxim Krikun, Yuan Cao, Qin Gao, Klaus Macherey, et al. Google’s neural machine translation system: Bridging the gap between human and machine translation. arXiv preprint arXiv:1609.08144, 2016. find
- Chen Sun, Austin Myers, Carl Vondrick, Kevin Murphy, and Cordelia Schmid. Videobert: A joint model for video and language representation learning. arXiv preprint arXiv:1904.01766, 2019. find our card
- Peter Anderson, Xiaodong He, Chris Buehler, Damien Teney, Mark Johnson, Stephen Gould, and Lei Zhang. Bottom-up and top-down attention for image captioning and visual question answering. In CVPR, pages 6077–6086, 2018. find
- Shaoqing Ren, Kaiming He, Ross Girshick, and Jian Sun. Faster r-cnn: Towards real-time object detection with region proposal networks. In NuerIPS, pages 91–99, 2015. find our card
- Sahar Kazemzadeh, Vicente Ordonez, Mark Matten, and Tamara Berg. Referitgame: Referring to objects in photographs of natural scenes. In EMNLP, 2014. find
- Licheng Yu, Zhe Lin, Xiaohui Shen, Jimei Yang, Xin Lu, Mohit Bansal, and Tamara L Berg. Mattnet: Modular attention network for referring expression comprehension. In CVPR, 2018. find
- Kaiming He, Georgia Gkioxari, Piotr Dollár, and Ross Girshick. Mask r-cnn. In ICCV, pages 2961–2969, 2017. find
- Kuang-Huei Lee, Xi Chen, Gang Hua, Houdong Hu, and Xiaodong He. Stacked cross attention for image-text matching. In ECCV, pages 201–216, 2018. find
- Gao Peng, Hongsheng Li, Haoxuan You, Zhengkai Jiang, Pan Lu, Steven Hoi, and Xiaogang Wang. Dynamic fusion with intra-and inter-modality attention flow for visual question answering. arXiv preprint arXiv:1812.05252, 2018. — [arXiv:1812.05252](https://arxiv.org/abs/1812.05252) find
- Alex Wang and Kyunghyun Cho. Bert has a mouth, and it must speak: Bert as a markov random field language model. arXiv preprint arXiv:1902.04094, 2019. find our card
- Carl Doersch, Abhinav Gupta, and Alexei A Efros. Unsupervised visual representation learning by context prediction. In ICCV, pages 1422–1430, 2015. find
- Richard Zhang, Phillip Isola, and Alexei A Efros. Colorful image colorization. In ECCV, pages 649–666. Springer, 2016. find
- Alexey Dosovitskiy, Philipp Fischer, Jost Tobias Springenberg, Martin Riedmiller, and Thomas Brox. Discriminative unsupervised feature learning with exemplar convolutional neural networks. IEEE PAMI, 38(9):1734–1747, 2015. find
- Deepak Pathak, Philipp Krahenbuhl, Jeff Donahue, Trevor Darrell, and Alexei A Efros. Context encoders: Feature learning by inpainting. In CVPR, pages 2536–2544, 2016. find
- Dinesh Jayaraman and Kristen Grauman. Learning image representations tied to ego-motion. In CVPR, pages 1413–1421, 2015. find
- Ishan Misra, C Lawrence Zitnick, and Martial Hebert. Shuffle and learn: unsupervised learning using temporal order verification. In ECCV, pages 527–544. Springer, 2016. find
- Guillaume Lample and Alexis Conneau. Cross-lingual language model pretraining. arXiv preprint arXiv:1901.07291, 2019. find
- Harm de Vries, Florian Strub, Jeremie Mary, Hugo Larochelle, Olivier Pietquin, and Aaron C. Courville. Modulating early visual processing by language. In NuerIPS, 2017. find