arXiv:2410.04880cs.CV2024-10被引 1

用不确定性采样减少标注数据,提升鱼类检测精度。

Improved detection of discarded fish species through BoxAL active learning

  • 基于Faster R-CNN的置信度评估,选择最不确定图像进行标注
  • 仅需400张更少标注样本即达随机采样性能,最后迭代时平均AP达39.0±1.6
  • 适合渔业自动化监测、资源调查等需要高效标注的场景

近年来,基于数据驱动的深度学习技术被广泛应用于自动捕捞记录。然而,这些方法依赖大量标注数据,而数据标注耗时、费力、成本高且需专业知识。本文提出一种名为BoxAL的主动学习方法,结合Faster R-CNN目标检测模型的认知不确定性估计,从未标注数据池中选取最不确定的图像进行标注与训练。我们使用为拖网捕捞深海物种设计的专用成像系统获取的开源图像数据集进行评估。结果表明,本方法在仅使用400张更少标注图像的情况下,即可达到与随机采样相当的目标检测性能;在1100张训练图像的最后迭代中,基于置信度采样的平均AP为39.0±1.6,显著高于随机采样的34.8±1.8。此外,认知不确定性可有效识别当前模型难以处理的图像,所选新数据比剩余未标注数据更具训练价值。相关软件已公开于https://github.com/pieterblok/boxal。

原文摘要 · Abstract (English)

In recent years, powerful data-driven deep-learning techniques have been developed and applied for automated catch registration. However, these methods are dependent on the labelled data, which is time-consuming, labour-intensive, expensive to collect and need expert knowledge. In this study, we present an active learning technique, named BoxAL, which includes estimation of epistemic certainty of the Faster R-CNN object-detection model. The method allows selecting the most uncertain training images from an unlabeled pool, which are then used to train the object-detection model. To evaluate the method, we used an open-source image dataset obtained with a dedicated image-acquisition system developed for commercial trawlers targeting demersal species. We demonstrated, that our approach allows reaching the same object-detection performance as with the random sampling using 400 fewer labelled images. Besides, mean AP score was significantly higher at the last training iteration with 1100 training images, specifically, 39.0±1.6 and 34.8±1.8 for certainty-based sampling and random sampling, respectively. Additionally, we showed that epistemic certainty is a suitable method to sample images that the current iteration of the model cannot deal with yet. Our study additionally showed that the sampled new data is more valuable for training than the remaining unlabeled data. Our software is available on https://github.com/pieterblok/boxal.

目标检测主动学习渔业监测

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