arXiv:2412.18870cs.CV2024-12被引 1

通过联合采样提升交通场景数据质量,显著改善3D目标检测性能。

TSceneJAL: Joint Active Learning of Traffic Scenes for 3D Object Detection

  • 基于类别熵、相似性和不确定性三重策略,智能筛选高质量场景。
  • 在KITTI等4个数据集上实现最高12%的检测精度提升。
  • 适合自动驾驶数据高效构建与模型训练优化的研究者使用。

大多数自动驾驶数据集采集和标注成本高昂,导致大量低质量冗余数据,影响模型性能与效率。本文提出交通场景联合主动学习框架TSceneJAL,可从已标注与未标注数据中高效采样平衡、多样且复杂的交通场景。该框架创新性地融合三种采样机制:基于类别熵的场景采样,识别多类物体共存场景以缓解类别不平衡;基于有向图表示与边际化核算法的相似性采样,选取稀疏且多样场景;基于混合密度网络的不确定性采样,挑选回归结果最模糊或最复杂的样本。三者结合形成联合选择策略,生成最优子数据集。在KITTI、Lyft、nuScenes和SUScape数据集上的实验表明,本方法在3D目标检测任务上优于现有先进方法,最高提升达12%。

原文摘要 · Abstract (English)

Most autonomous driving (AD) datasets incur substantial costs for collection and labeling, inevitably yielding a plethora of low-quality and redundant data instances, thereby compromising performance and efficiency. Many applications in AD systems necessitate high-quality training datasets using both existing datasets and newly collected data. In this paper, we propose a traffic scene joint active learning (TSceneJAL) framework that can efficiently sample the balanced, diverse, and complex traffic scenes from both labeled and unlabeled data. The novelty of this framework is threefold: 1) a scene sampling scheme based on a category entropy, to identify scenes containing multiple object classes, thus mitigating class imbalance for the active learner; 2) a similarity sampling scheme, estimated through the directed graph representation and a marginalize kernel algorithm, to pick sparse and diverse scenes; 3) an uncertainty sampling scheme, predicted by a mixture density network, to select instances with the most unclear or complex regression outcomes for the learner. Finally, the integration of these three schemes in a joint selection strategy yields an optimal and valuable subdataset. Experiments on the KITTI, Lyft, nuScenes and SUScape datasets demonstrate that our approach outperforms existing state-of-the-art methods on 3D object detection tasks with up to 12% improvements.

3D检测主动学习自动驾驶数据采样

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