Supervised Fine-tuning in turn Improves Visual Foundation Models

Xiaohu Jiang, Yixiao Ge, Yuying Ge, Dachuan Shi, Chun Yuan +1 more
1/18/2024
cs.CVcs.AI

No official code found. Best implementation: TencentARC/ViSFT (38 stars, Apache-2.0, last commit 2024-01-20).

Abstract

Image-text training like CLIP has dominated the pretraining of vision foundation models in recent years. Subsequent efforts have been made to introduce region-level visual learning into CLIP's pretraining but face scalability challenges due to the lack of large-scale region-level datasets. Drawing inspiration from supervised fine-tuning (SFT) in natural language processing such as instruction tuning, we explore the potential of fine-grained SFT in enhancing the generation of vision foundation models after their pretraining. Thus a two-stage method ViSFT (Vision SFT) is proposed to unleash the fine-grained knowledge of vision foundation models. In ViSFT, the vision foundation model is enhanced by performing visual joint learning on some in-domain tasks and then tested on out-of-domain benchmarks. With updating using ViSFT on 8 V100 GPUs in less than 2 days, a vision transformer with over 4.4B parameters shows improvements across various out-of-domain benchmarks including vision and vision-linguistic scenarios.

View on arXivView PDF

Code Implementations(1)

TencentARC/ViSFTCommunity100%
Apache-2.0
381C, C++Jan 16, 20242 years agoApache-2.0

Similar Papers

Cite this paper

@article{jiang2024supervised,
  title  = {Supervised Fine-tuning in turn Improves Visual Foundation Models},
  author = {Xiaohu Jiang and Yixiao Ge and Yuying Ge and Dachuan Shi and Chun Yuan and Ying Shan},
  year   = {2024},
  eprint = {2401.10222},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2401.10222v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2401.10222.svg)](https://codeofpaper.com/paper/2401.10222)

Discussion