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

GiT:通过通用语言接口迈向通用视觉Transformer

GiT: Towards Generalist Vision Transformer through Universal Language Interface

Peking University(北京大学);Max Planck Institute for Informatics(马克斯·普朗克信息学研究所);The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳));ETH Zurich(苏黎世联邦理工学院)

Haiyang Wang, Hao Tang, Li Jiang, Shaoshuai Shi, Muhammad Ferjad Naeem, Hongsheng Li, Bernt Schiele, Liwei Wang

AI总结:

本文提出GiT,一种仅用普通ViT和通用语言接口统一多种视觉任务的通用视觉Transformer,在五个基准上联合训练并取得领先性能,缩小了视觉与语言的架构差距。

摘要:

本文提出一个简单而有效的框架,称为GiT,仅使用普通的ViT即可同时适用于各种视觉任务。受大型语言模型(LLM)中广泛使用的多层Transformer架构(如GPT)的通用性启发,我们寻求拓宽其范围,使其作为强大的视觉基础模型(VFM)。然而,与语言建模不同,视觉任务通常需要特定模块,如用于检测的边界框头和用于分割的像素解码器,这极大地阻碍了强大的多层Transformer在视觉领域的应用。为解决此问题,我们设计了一个通用语言接口,使成功的自回归解码能够熟练地统一各种视觉任务,从图像级理解(如描述)到稀疏感知(如检测),再到密集预测(如分割)。基于上述设计,整个模型仅由ViT组成,无需任何特定附加组件,提供了显著的架构简化。GiT是一个多任务视觉模型,在五个代表性基准上联合训练,无需任务特定的微调。有趣的是,我们的GiT在通用性能上建立了新的基准,并促进了任务间的相互增强,与单独训练相比带来了显著改进。这反映了在LLM中观察到的类似影响。进一步用27个数据集丰富训练后,GiT在各种任务上取得了强大的零样本结果。由于其简单设计,这一范式有望缩小视觉与语言之间的架构差距。代码和模型将在\url{https://github.com/Haiyang-W/GiT}提供。

英文摘要:

This paper proposes a simple, yet effective framework, called GiT, simultaneously applicable for various vision tasks only with a vanilla ViT. Motivated by the universality of the Multi-layer Transformer architecture (e.g, GPT) widely used in large language models (LLMs), we seek to broaden its scope to serve as a powerful vision foundation model (VFM). However, unlike language modeling, visual tasks typically require specific modules, such as bounding box heads for detection and pixel decoders for segmentation, greatly hindering the application of powerful multi-layer transformers in the vision domain. To solve this, we design a universal language interface that empowers the successful auto-regressive decoding to adeptly unify various visual tasks, from image-level understanding (e.g., captioning), over sparse perception (e.g., detection), to dense prediction (e.g., segmentation). Based on the above designs, the entire model is composed solely of a ViT, without any specific additions, offering a remarkable architectural simplification. GiT is a multi-task visual model, jointly trained across five representative benchmarks without task-specific fine-tuning. Interestingly, our GiT builds a new benchmark in generalist performance, and fosters mutual enhancement across tasks, leading to significant improvements compared to isolated training. This reflects a similar impact observed in LLMs. Further enriching training with 27 datasets, GiT achieves strong zero-shot results over various tasks. Due to its simple design, this paradigm holds promise for narrowing the architectural gap between vision and language. Code and models will be available at \url{https://github.com/Haiyang-W/GiT}.

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

↑