通过跨模态注意力轨迹筛选数据,显著减少视觉语言模型训练冗余。
Data Selection for Fine-tuning Vision Language Models via Cross Modal Alignment Trajectories
- 基于注意力矩阵轨迹聚类,识别信息相似的训练样本
- 可删减50%~85%训练数据,性能不降且训练提速20%
- 适合大规模视觉语言模型高效微调,尤其适用于资源受限场景
数据高效学习旨在通过仅使用最具信息量的子集来消除大规模训练数据中的冗余。尽管该方向在视觉模型和大语言模型中已有广泛研究,但对大型视觉-语言模型(LVLM)仍缺乏有效方法,现有方法甚至无法超越随机选择。本文提出首个针对LVLM指令微调的数据高效方法——XMAS。我们证明:在指令微调过程中,具有相似跨模态注意力矩阵的样本具有相近梯度,因此对模型参数的影响方式相同,传递的信息一致。基于此,XMAS通过微调小型代理模型获取注意力矩阵的前几个奇异值轨迹,对样本进行聚类,并从各簇中均衡采样,从而有效去除大规模LVLM训练数据的冗余。大量实验表明,XMAS可删除LLaVA-665k数据集50%、Vision-Flan数据集85%的样本,同时保持LLaVA-1.5-7B在10个下游基准上的性能不变,并使训练速度提升1.2倍,较最佳基线多减少30%数据。项目主页见:https://bigml-cs-ucla.github.io/XMAS-project-page/
原文摘要 · Abstract (English)
Data-efficient learning aims to eliminate redundancy in large training datasets by training models on smaller subsets of the most informative examples. While data selection has been extensively explored for vision models and large language models (LLMs), it remains underexplored for Large Vision-Language Models (LVLMs). Notably, none of existing methods can outperform random selection at different subset sizes. In this work, we propose the first principled method for data-efficient instruction tuning of LVLMs. We prove that examples with similar cross-modal attention matrices during instruction tuning have similar gradients. Thus, they influence model parameters in a similar manner and convey the same information to the model during training. Building on this insight, we propose XMAS, which clusters examples based on the trajectories of the top singular values of their attention matrices obtained from fine-tuning a small proxy LVLM. By sampling a balanced subset from these clusters, XMAS effectively removes redundancy in large-scale LVLM training data. Extensive experiments show that XMAS can discard 50% of the LLaVA-665k dataset and 85% of the Vision-Flan dataset while fully preserving performance of LLaVA-1.5-7B on 10 downstream benchmarks and speeding up its training by 1.2x. This is 30% more data reduction compared to the best baseline for LLaVA-665k. The project's website can be found at https://bigml-cs-ucla.github.io/XMAS-project-page/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。