arXiv:2410.16512cs.CV2024-10ICLR被引 40

TIPS模型通过增强空间感知,让图文预训练同时胜任密集与全局视觉任务。

TIPS: Text-Image Pretraining with Spatial awareness

  • 用合成文本替代噪声网页标题,提升空间表征学习效果
  • 结合对比学习与掩码图像建模,显著增强空间一致性
  • 无需微调即可在16个数据集上实现强泛化性能,适合多任务场景

尽管图像-文本表征学习近年广受欢迎,现有模型普遍缺乏空间感知能力,难以直接用于密集理解任务。因此,许多密集视觉应用(如深度估计、语义分割)仍依赖无监督的纯图像预训练,即使缺乏显式监督信号。本文提出一种通用图像-文本模型TIPS,填补图文与自监督学习之间的差距。方法上:首先发现用合成文本替换噪声网络标题可显著提升密集理解性能,因合成描述提供更丰富的空间信号;我们设计了一种融合噪声与合成文本的训练策略,在密集与全局任务中均取得提升。其次,提出将对比图像-文本学习与自监督掩码图像建模结合,以增强空间一致性,大幅提高下游应用表现。基于Transformer架构,在精选公共图像数据集上训练模型,实验覆盖8项任务、16个数据集,验证了其在图像-文本与仅图像任务中均具备出色的即插即用性能。代码与模型已开源。

原文摘要 · Abstract (English)

While image-text representation learning has become very popular in recent years, existing models tend to lack spatial awareness and have limited direct applicability for dense understanding tasks. For this reason, self-supervised image-only pretraining is still the go-to method for many dense vision applications (e.g. depth estimation, semantic segmentation), despite the lack of explicit supervisory signals. In this paper, we close this gap between image-text and self-supervised learning, by proposing a novel general-purpose image-text model, which can be effectively used off the shelf for dense and global vision tasks. Our method, which we refer to as Text-Image Pretraining with Spatial awareness (TIPS), leverages two simple and effective insights. First, on textual supervision: we reveal that replacing noisy web image captions by synthetically generated textual descriptions boosts dense understanding performance significantly, due to a much richer signal for learning spatially aware representations. We propose an adapted training method that combines noisy and synthetic captions, resulting in improvements across both dense and global understanding tasks. Second, on the learning technique: we propose to combine contrastive image-text learning with self-supervised masked image modeling, to encourage spatial coherence, unlocking substantial enhancements for downstream applications. Building on these two ideas, we scale our model using the transformer architecture, trained on a curated set of public images. Our experiments are conducted on 8 tasks involving 16 datasets in total, demonstrating strong off-the-shelf performance on both dense and global understanding, for several image-only and image-text tasks. Code and models are released at https://github.com/google-deepmind/tips.

图文预训练空间感知自监督学习通用模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。