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

• •    下一篇

自步学习指导下的半监督目标检测框架

谢斌红,剌颖坤,张英俊,张睿   

  1. 太原科技大学
  • 收稿日期:2024-08-06 修回日期:2024-10-21 发布日期:2024-11-19 出版日期:2024-11-19
  • 通讯作者: 剌颖坤
  • 基金资助:
    基于轻量卷积网络自适应压缩与加速的不锈钢焊缝缺陷超声检测方法研究;煤矿智慧安全管理监测平台研究

Semi-supervised object detection framework guided by self-paced learning#br#

  • Received:2024-08-06 Revised:2024-10-21 Online:2024-11-19 Published:2024-11-19

摘要: 为了提高伪标签质量并解决半监督目标检测中的确认偏差问题,提出了一种基于动态参数的自步学习指导下的半监督目标检测框架。该框架设计了动态自步参数和连续权重变量,以优化半监督目标检测的效果。动态自步参数根据模型在训练过程中的实时表现评估样本的难易程度;连续权重变量则通过比较样本损失与动态自步参数的关系,精确评估每个样本在训练中的重要性和可靠性,并对样本中每个物体都进行了精细化权重设计。此外,该框架采用单一模型进行迭代训练,并引入一致性正则化策略评估模型预测的一致性。这种设计不仅能为模型提供更有针对性的权重信息,并且通过权重信息的动态调整使模型自适应地优化训练过程。在PASCAL VOC和MS-COCO数据集上的广泛对比实验显示,该框架不仅显著提升了模型的检测精度,同时验证了其广泛的通用性和高效的收敛性能。

关键词: 半监督目标检测, 自步学习, 一致性正则化, 动态自步参数, 连续权重变量

Abstract: In order to improve the quality of pseudo-labels and solve the problem of confirmation bias in semi-supervised object detection,a semi-supervised object detection framework based on dynamic parameters under the guidance of self-step learning was proposed.The framework was designed with dynamic self-stepping parameters and continuous weight variables to optimize the effect of semi-supervised object detection.The dynamic self-stepping parameter evaluates the difficulty of the sample based on the real-time performance of the model during the training process.The continuous weight variables accurately evaluated the importance and reliability of each sample in training by comparing the relationship between sample loss and dynamic self-stepping parameters,and refined weight design was carried out for each object in the sample.In addition, the framework uses a single model for iterative training, and introduces a consistency regularization strategy to evaluate the consistency of model predictions.This design not only provides more targeted weight information for the model, but also adaptively optimizes the training process through the dynamic adjustment of the weight information.Extensive comparative experiments on PASCAL VOC and MS-COCO datasets show that the framework not only significantly improves the detection accuracy of the model, but also verifies its wide versatility and efficient convergence performance.

Key words: Semi-Supervised Object Detection, Self-Paced Learning, Consistency Regularization, Dynamic Self-Paced Parameter, Continuous Weight Variable

中图分类号: