arXiv:2409.15486cs.CVcs.AI2024-09被引 8

用视觉语言模型挖掘长尾数据,提升罕见样本识别效果。

VLMine: Long-Tail Data Mining with Vision Language Models

  • 利用大模型提取图像关键词,按频率识别稀有样本。
  • 在多个基准上实现10%至50%的性能提升。
  • 方法简单通用,适用于2D分类与3D检测任务。

确保机器学习在长尾样本上的稳健表现是自动驾驶等实际应用中的关键问题。本文聚焦于从无标签数据中识别罕见样本。提出一种简单且可扩展的数据挖掘方法,利用大型视觉语言模型(VLM)将图像内容总结为关键词,并基于关键词频率识别长尾样本。实验发现,相较于基于模型不确定性的传统方法,VLM能提供更有效的稀有样本信号。因此,提出一种整合多种挖掘算法信号的通用策略。在两个不同任务上进行评估:2D图像分类(以类间差异为主)和3D物体检测(以类内差异为主)。此外,通过检测任务证明了从2D图像中提取的知识可迁移至3D领域。实验结果一致显示,在ImageNet-LT、Places-LT和Waymo Open Dataset等多个代表性基准上,性能相比基线技术提升10%至50%。

原文摘要 · Abstract (English)

Ensuring robust performance on long-tail examples is an important problem for many real-world applications of machine learning, such as autonomous driving. This work focuses on the problem of identifying rare examples within a corpus of unlabeled data. We propose a simple and scalable data mining approach that leverages the knowledge contained within a large vision language model (VLM). Our approach utilizes a VLM to summarize the content of an image into a set of keywords, and we identify rare examples based on keyword frequency. We find that the VLM offers a distinct signal for identifying long-tail examples when compared to conventional methods based on model uncertainty. Therefore, we propose a simple and general approach for integrating signals from multiple mining algorithms. We evaluate the proposed method on two diverse tasks: 2D image classification, in which inter-class variation is the primary source of data diversity, and on 3D object detection, where intra-class variation is the main concern. Furthermore, through the detection task, we demonstrate that the knowledge extracted from 2D images is transferable to the 3D domain. Our experiments consistently show large improvements (between 10\% and 50\%) over the baseline techniques on several representative benchmarks: ImageNet-LT, Places-LT, and the Waymo Open Dataset.

长尾数据视觉语言模型数据挖掘知识迁移

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。