Concept-skill Transferability-based Data Selection for Large Vision-Language Models

Jaewoo Lee, Boyang Li, Sung Ju Hwang
6/16/2024
cs.CVcs.LG

No official code found. Best implementation: G-JWLee/COINCIDE_code (23 stars, last commit 2024-11-04).

Abstract

Instruction tuning, or supervised finetuning on extensive task-specific data, is necessary for Large Vision-Language Models (LVLMs) to generalize well across a broad range of vision-language (VL) tasks. However, training on large VL datasets can become prohibitively expensive. In this work, we introduce COINCIDE, an effective and scalable data selection technique that uses a small model as a reference model to select visual instruction tuning data for efficient finetuning of a target LVLM, focusing on diversity and transferability. Specifically, we cluster the training data using internal activations from a small model, which identifies VL concept-skill compositions needed by a target LVLM. We then sample data from these diverse clusters by considering their density and transferability, or the ability to transfer well to other concept-skill compositions. This approach ensures the diversity of these compositions, which is vital for LVLM generalization. Extensive experiments demonstrate that COINCIDE achieves superior performance and data selection efficiency against 8 strong baselines on two distinct datasets: LLaVA-1.5 and Vision-Flan. Using only 20% of the LLaVA-1.5 dataset, COINCIDE achieves performance comparable to the LVLM finetuned on the whole dataset, with 70% reduction of the wall-clock running time. On the Vision-Flan dataset, our method achieves superior results with only 16.7% of the training data.

View on arXivView PDF

Code Implementations(1)

230Shell, PythonAug 15, 20241 years ago

Similar Papers

Cite this paper

@article{lee2024conceptskill,
  title  = {Concept-skill Transferability-based Data Selection for Large Vision-Language Models},
  author = {Jaewoo Lee and Boyang Li and Sung Ju Hwang},
  year   = {2024},
  eprint = {2406.10995},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2406.10995v2}
}

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/2406.10995.svg)](https://codeofpaper.com/paper/2406.10995)

Discussion