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

视觉-语言预训练的正确分类方法

Classification Done Right for Vision-Language Pre-Training

ByteDance Research(字节跳动研究院)

Zilong Huang, Qinghao Ye, Bingyi Kang, Jiashi Feng, Haoqi Fan

AI总结:

提出SuperClass分类方法,直接用原始文本标签替代对比学习,无需文本编码器和大批次,在视觉-语言下游任务上表现优于CLIP。

摘要:

我们引入了SuperClass,一种用于图像-文本数据视觉-语言预训练的超级简单的分类方法。与对比学习对应方法CLIP(通过文本编码器进行对比)不同,SuperClass直接利用分词后的原始文本作为监督分类标签,无需额外的文本过滤或选择。由于缺少作为对比目标的文本编码,SuperClass不需要文本编码器,也不像CLIP那样需要维持大批次大小。SuperClass在各种下游任务上展示了优越的性能,包括经典计算机视觉基准和视觉-语言下游任务。我们进一步探索了SuperClass在模型规模、训练长度或数据规模上的扩展行为,并报告了令人鼓舞的结果以及与CLIP的比较。代码:https://github.com/x-cls/superclass

英文摘要:

We introduce SuperClass, a super simple classification method for vision-language pre-training on image-text data. Unlike its contrastive counterpart CLIP who contrast with a text encoder, SuperClass directly utilizes tokenized raw text as supervised classification labels, without the need for additional text filtering or selection. Due to the absence of the text encoding as contrastive target, SuperClass does not require a text encoder and does not need to maintain a large batch size as CLIP does. SuperClass demonstrated superior performance on various downstream tasks, including classic computer vision benchmarks and vision language downstream tasks. We further explored the scaling behavior of SuperClass on model size, training length, or data size, and reported encouraging results and comparisons to CLIP. https://github.com/x-cls/superclass

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

↑