先选图再生成指令,大幅降低视觉指令微调的数据成本
Filter Images First, Generate Instructions Later: Pre-Instruction Data Selection for Visual Instruction Tuning
- 先根据任务重要性分配采样预算,再聚类选代表性图像
- 仅对15%图像生成指令,性能接近全量数据训练
- 适合资源有限者构建定制化视觉指令数据集
大型视觉语言模型的视觉指令微调(VIT)需要大量图像-指令对数据集,成本高昂。现有方法虽尝试筛选高质量子集以减少训练时间,但常忽略从无标签图像生成指令的高成本问题。多数数据集依赖人工标注或付费API,限制了资源受限用户的使用。为此,我们提出预指令数据选择(PreSel),一种更实用的数据选择范式:先直接挑选最具价值的无标签图像,再仅对选定图像生成指令。PreSel首先估计VIT数据集中各视觉任务的重要性,分配任务级采样预算;随后在每类任务内对图像特征聚类,按预算选取最具代表性的样本。该方法显著降低指令生成与模型微调的计算开销。实验显示,仅对15%图像生成指令,PreSel在LLaVA-1.5和Vision-Flan数据集上的表现即可媲美全量数据训练。
原文摘要 · Abstract (English)
Visual instruction tuning (VIT) for large vision-language models (LVLMs) requires training on expansive datasets of image-instruction pairs, which can be costly. Recent efforts in VIT data selection aim to select a small subset of high-quality image-instruction pairs, reducing VIT runtime while maintaining performance comparable to full-scale training. However, a major challenge often overlooked is that generating instructions from unlabeled images for VIT is highly expensive. Most existing VIT datasets rely heavily on human annotations or paid services like the GPT API, which limits users with constrained resources from creating VIT datasets for custom applications. To address this, we introduce Pre-Instruction Data Selection (PreSel), a more practical data selection paradigm that directly selects the most beneficial unlabeled images and generates instructions only for the selected images. PreSel first estimates the relative importance of each vision task within VIT datasets to derive task-wise sampling budgets. It then clusters image features within each task, selecting the most representative images with the budget. This approach reduces computational overhead for both instruction generation during VIT data formation and LVLM fine-tuning. By generating instructions for only 15% of the images, PreSel achieves performance comparable to full-data VIT on the LLaVA-1.5 and Vision-Flan datasets. The link to our project page: https://bardisafa.github.io/PreSel
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。