arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.06767cs.CVcs.LG
收藏

Wukong:一个一亿级大规模中文跨模态预训练基准

Wukong: A 100 Million Large-scale Chinese Cross-modal Pre-training Benchmark

Huawei Noah’s Ark Lab(华为诺亚方舟实验室);Sun Yat-sen University(中山大学)

Jiaxi Gu, Xiaojun Meng, Guansong Lu, Lu Hou, Minzhe Niu, Xiaodan Liang, Lewei Yao, Runhui Huang, Wei Zhang, Xin Jiang, Chunjing Xu, Hang Xu

摘要:

视觉语言预训练(VLP)模型在各种下游任务上表现出卓越的性能。它们的成功在很大程度上依赖于预先训练的跨模态数据集的规模。然而,由于缺乏大规模的中文数据集和基准,阻碍了中文VLP模型的发展和更广泛的多语言应用。在这项工作中,我们发布了一个名为Wukong的大规模中文跨模态数据集,其中包含从网络上收集的1亿个中文图像-文本对。Wukong旨在对不同的多模式预培训方法进行基准测试,以促进VLP研究和社区发展。此外,我们还发布了一组使用各种图像编码器(ViT-B/ViT-L/SwinT)进行预训练的模型,并将高级预训练技术应用于VLP,例如锁定图像文本调整,对比学习中的令牌相似性以及减少令牌交互。还提供了广泛的实验和不同下游任务的基准测试,包括一个新的最大的人类验证的图像-文本测试数据集。实验表明,Wukong可以作为一个有前途的中文预训练数据集和不同的跨模态学习方法的基准。对于10个数据集上的zero-shot图像分类任务,$Wukong_{ViT-L}$实现了73.03%的平均准确率。在图文检索任务中,AIC-ICC的平均召回率达到71.6%,比文澜2.0提高了12.9%。此外,我们的悟空模型在下游任务上与多个数据集上的其他变体进行了基准测试,例如,Flickr 8 K-CN,Flickr-30 K-CN,COCO-CN,et al.更多信息可参考:https://wukong-dataset.github.io/wukong-dataset/.

英文摘要:

Vision-Language Pre-training (VLP) models have shown remarkable performance on various downstream tasks. Their success heavily relies on the scale of pre-trained cross-modal datasets. However, the lack of large-scale datasets and benchmarks in Chinese hinders the development of Chinese VLP models and broader multilingual applications. In this work, we release a large-scale Chinese cross-modal dataset named Wukong, which contains 100 million Chinese image-text pairs collected from the web. Wukong aims to benchmark different multi-modal pre-training methods to facilitate the VLP research and community development. Furthermore, we release a group of models pre-trained with various image encoders (ViT-B/ViT-L/SwinT) and also apply advanced pre-training techniques into VLP such as locked-image text tuning, token-wise similarity in contrastive learning, and reduced-token interaction. Extensive experiments and a benchmarking of different downstream tasks including a new largest human-verified image-text test dataset are also provided. Experiments show that Wukong can serve as a promising Chinese pre-training dataset and benchmark for different cross-modal learning methods. For the zero-shot image classification task on 10 datasets, $Wukong_{ViT-L}$ achieves an average accuracy of 73.03%. For the image-text retrieval task, it achieves a mean recall of 71.6% on AIC-ICC which is 12.9% higher than WenLan 2.0. Also, our Wukong models are benchmarked on downstream tasks with other variants on multiple datasets, e.g., Flickr8K-CN, Flickr-30K-CN, COCO-CN, et al. More information can be referred to: https://wukong-dataset.github.io/wukong-dataset/.

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

↑