《计算机应用》唯一官方网站

• •    下一篇

课程学习指导下的半监督目标检测框架

张英俊1,李牛牛2,谢斌红3,张睿1,陆望东1   

  1. 1. 太原科技大学
    2. 山西省太原市万柏林区和平街道太原科技大学
    3. 太原科技大学 计算机科学与技术学院,太原 030024
  • 收稿日期:2023-08-07 修回日期:2023-10-10 发布日期:2023-12-18 出版日期:2023-12-18
  • 通讯作者: 李牛牛
  • 基金资助:
    基于轻量卷积网络自适应压缩与加速的不锈钢焊缝缺陷超声检测方法研究;煤矿智慧安全管理监测平台研究

Semi-supervised object detection framework guided by curriculum learning

  • Received:2023-08-07 Revised:2023-10-10 Online:2023-12-18 Published:2023-12-18

摘要: 摘 要: 为了提高伪标签的质量,解决半监督目标检测中的确认偏差问题,并针对现有算法中忽视无标注数据复杂性导致错误伪标签的难点,提出了一种课程学习指导下的半监督目标检测框架。该框架主要由ICSD(Iou-Confidence-Standard-Deviation)难度测量器和BP(Batch-Package)训练调度器两个模块组成。其中,ICSD难度测量器综合考虑了伪边界框之间的IOU、置信度、类别标签等信息,并引入C_IOU(Checkpoint_IOU)方法来评估无标注数据的可靠性。BP训练调度器则设计了两种高效调度策略,分别从Batch和Package角度出发,优先选择可靠性指标高的无标记数据,实现以课程学习的方式对整个无标记数据集的充分利用。在PASCAL VOC和MS-COCO数据集上的广泛对比实验表明,该框架不仅适用于现有的半监督目标检测算法,而且在检测精度和稳定性方面都得到了显著提升。

关键词: 半监督学习, 目标检测, 课程学习, 训练策略, 难度测量器, 训练调度器

Abstract: Abstract: In order to enhance the quality of pseudo labels, address the issue of confirmation bias in semi-supervised object detection, and tackle the challenge of ignoring complexities in unlabeled data leading to erroneous pseudo labels in existing algorithms, a semi-supervised object detection framework guided by curriculum learning is proposed. The framework consists of two modules: the ICSD (Iou-Confidence-Standard-Deviation) difficulty measurer and the BP (Batch-Package) trainingscheduler. The ICSD difficulty measurer comprehensively considers information such as IOU, confidence, and class labels between pseudo-bounding boxes,and introduces the C_IOU (Checkpoint_IOU) method to evaluate the reliability of unlabeled data. The BP training scheduler has designed two efficient scheduling strategies, starting from the perspective of Batch and Package respectively, giving priority to unlabeled data with high reliability indicators to achieve full utilization of the entire unlabeled data set in the form of course learning. Extensive comparative experiments on the PASCAL VOC and MS-COCO datasets demonstrate that this framework applies to existing semi-supervised object detection algorithms and exhibits significant improvements in detection accuracy and stability.

Key words: semi-supervised learning, object detection, curriculum learning, training strategy, difficulty measurer, training scheduler

中图分类号: