用注意力机制筛选高质量图文数据,20%数据达到顶尖效果
$Δ$-AttnMask: Attention-Guided Masked Hidden States for Efficient Data Selection and Augmentation
- 通过掩码高注意力区域计算损失差值Δ,无须标签或额外训练评估样本质量
- 仅用20%数据即达最优性能,训练速度提升5倍,准确率高出10.1%
- 不依赖具体模型或数据,适用于多种视觉语言模型和任务
视觉指令微调(VIF)对后训练视觉-语言模型(VLMs)至关重要。与仅需文本指令数据的单模态微调不同,VIF还需多模态数据以实现视觉与文本的联合理解,因此通常需要更多数据。这带来更严格的数据选择挑战:方法必须高效扩展以应对更大数据需求,同时保证图像、文本内容质量及其对齐。尽管对性能影响重大,现有研究仍忽视了VIF中的数据选择问题。本文提出Δ-AttnMask,一种数据高效的框架,通过注意力引导的隐藏状态掩码量化样本质量,无需领域标签、辅助模型或额外训练即可联合评估图像-文本对。通过计算原始状态与掩码高注意力区域状态间的损失差值(Δ),该方法内生地评估样本质量。在多个VLMs和数据集上的实验表明,Δ-AttnMask仅使用20%数据即达到当前最佳表现,训练加速5倍,整体准确率较全数据基线提升+10.1%。其模型无关和数据无关设计确保了在多种模态和架构间的广泛适用性。
原文摘要 · Abstract (English)
Visual Instruction Finetuning (VIF) is pivotal for post-training Vision-Language Models (VLMs). Unlike unimodal instruction finetuning in plain-text large language models, which mainly requires instruction datasets to enable model instruction-following ability, VIF also requires multimodal data to enable joint visual and textual understanding; therefore, it typically requires more data. Consequently, VIF imposes stricter data selection challenges: the method must scale efficiently to handle larger data demands while ensuring the quality of both visual and textual content, as well as their alignment. Despite its critical impact on performance, data selection for VIF remains an understudied area. In this paper, we propose $Δ$-AttnMask. This data-efficient framework quantifies sample quality through attention-guided masking of the model's hidden states, jointly evaluating image-text pairs without requiring domain labels, auxiliary models, or extra training. By computing loss differences ($Δ$) between the original states and states masked using high-attention regions, $Δ$-AttnMask intrinsically assesses sample quality. Experiments across multiple VLMs and datasets show that $Δ$-AttnMask achieves state-of-the-art performance with just 20% of data, accelerating training by 5x while surpassing full-dataset baselines by +10.1% in overall accuracy. Its model-agnostic and data-agnostic design ensures broad applicability across modalities and architectures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。