GiT: Towards Generalist Vision Transformer through Universal Language Interface

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Haiyang, Tang, Hao, Jiang, Li, Shi, Shaoshuai, Naeem, Muhammad Ferjad, Li, Hongsheng, Schiele, Bernt, Wang, Liwei
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909136924442624
author Wang, Haiyang
Tang, Hao
Jiang, Li
Shi, Shaoshuai
Naeem, Muhammad Ferjad
Li, Hongsheng
Schiele, Bernt
Wang, Liwei
author_facet Wang, Haiyang
Tang, Hao
Jiang, Li
Shi, Shaoshuai
Naeem, Muhammad Ferjad
Li, Hongsheng
Schiele, Bernt
Wang, Liwei
contents 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}.
format Preprint
id arxiv_https___arxiv_org_abs_2403_09394
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle GiT: Towards Generalist Vision Transformer through Universal Language Interface
Wang, Haiyang
Tang, Hao
Jiang, Li
Shi, Shaoshuai
Naeem, Muhammad Ferjad
Li, Hongsheng
Schiele, Bernt
Wang, Liwei
Computer Vision and Pattern Recognition
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}.
title GiT: Towards Generalist Vision Transformer through Universal Language Interface
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.09394