Vision Model Pre-training on Interleaved Image-Text Data via Latent Compression Learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Yang, Chenyu, Zhu, Xizhou, Zhu, Jinguo, Su, Weijie, Wang, Junjie, Dong, Xuan, Wang, Wenhai, Lu, Lewei, Li, Bin, Zhou, Jie, Qiao, Yu, Dai, Jifeng
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909436283453440
author Yang, Chenyu
Zhu, Xizhou
Zhu, Jinguo
Su, Weijie
Wang, Junjie
Dong, Xuan
Wang, Wenhai
Lu, Lewei
Li, Bin
Zhou, Jie
Qiao, Yu
Dai, Jifeng
author_facet Yang, Chenyu
Zhu, Xizhou
Zhu, Jinguo
Su, Weijie
Wang, Junjie
Dong, Xuan
Wang, Wenhai
Lu, Lewei
Li, Bin
Zhou, Jie
Qiao, Yu
Dai, Jifeng
contents Recently, vision model pre-training has evolved from relying on manually annotated datasets to leveraging large-scale, web-crawled image-text data. Despite these advances, there is no pre-training method that effectively exploits the interleaved image-text data, which is very prevalent on the Internet. Inspired by the recent success of compression learning in natural language processing, we propose a novel vision model pre-training method called Latent Compression Learning (LCL) for interleaved image-text data. This method performs latent compression learning by maximizing the mutual information between the inputs and outputs of a causal attention model. The training objective can be decomposed into two basic tasks: 1) contrastive learning between visual representation and preceding context, and 2) generating subsequent text based on visual representation. Our experiments demonstrate that our method not only matches the performance of CLIP on paired pre-training datasets (e.g., LAION), but can also leverage interleaved pre-training data (e.g., MMC4) to learn robust visual representation from scratch, showcasing the potential of vision model pre-training with interleaved image-text data. Code is released at https://github.com/OpenGVLab/LCL.
format Preprint
id arxiv_https___arxiv_org_abs_2406_07543
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Vision Model Pre-training on Interleaved Image-Text Data via Latent Compression Learning
Yang, Chenyu
Zhu, Xizhou
Zhu, Jinguo
Su, Weijie
Wang, Junjie
Dong, Xuan
Wang, Wenhai
Lu, Lewei
Li, Bin
Zhou, Jie
Qiao, Yu
Dai, Jifeng
Computer Vision and Pattern Recognition
Recently, vision model pre-training has evolved from relying on manually annotated datasets to leveraging large-scale, web-crawled image-text data. Despite these advances, there is no pre-training method that effectively exploits the interleaved image-text data, which is very prevalent on the Internet. Inspired by the recent success of compression learning in natural language processing, we propose a novel vision model pre-training method called Latent Compression Learning (LCL) for interleaved image-text data. This method performs latent compression learning by maximizing the mutual information between the inputs and outputs of a causal attention model. The training objective can be decomposed into two basic tasks: 1) contrastive learning between visual representation and preceding context, and 2) generating subsequent text based on visual representation. Our experiments demonstrate that our method not only matches the performance of CLIP on paired pre-training datasets (e.g., LAION), but can also leverage interleaved pre-training data (e.g., MMC4) to learn robust visual representation from scratch, showcasing the potential of vision model pre-training with interleaved image-text data. Code is released at https://github.com/OpenGVLab/LCL.
title Vision Model Pre-training on Interleaved Image-Text Data via Latent Compression Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.07543