通过置信度优先策略,让模型从易到难学习飞鸟目标检测。
Self-Paced Learning Strategy with Easy Sample Prior Based on Confidence for the Flying Bird Object Detection Model Training
- 基于置信度设计新损失函数,更适配单类目标检测。
- 先用简单样本训练,再结合难易样本提升检测性能,AP50提升2.1%。
- 适合监控视频中飞鸟等小目标的检测任务,可提升早期学习效率。
为避免难样本对飞鸟目标检测模型(FBOD model)训练过程的影响,本文提出一种基于置信度的易样本优先自步学习策略(SPL-ESP-BC)。首先改进自步学习中的基于损失的最小化函数,提出基于置信度的最小化函数,更适用于单类目标检测任务;其次,提出易样本优先(ESP)策略,使模型在训练初期即可判断样本难易程度。采用先以简单样本训练、再以全样本使用SPL策略的方式训练FBOD模型。结合置信度最小化函数与易样本优先策略,形成SPL-ESP-BC训练方法。实验表明,相比不区分难易样本的标准训练策略,该方法使FBOD模型的AP50提升2.1%;相较其他基于损失的SPL策略,SPL-ESP-BC在综合检测性能上表现最佳。
原文摘要 · Abstract (English)
In order to avoid the impact of hard samples on the training process of the Flying Bird Object Detection model (FBOD model, in our previous work, we designed the FBOD model according to the characteristics of flying bird objects in surveillance video), the Self-Paced Learning strategy with Easy Sample Prior Based on Confidence (SPL-ESP-BC), a new model training strategy, is proposed. Firstly, the loss-based Minimizer Function in Self-Paced Learning (SPL) is improved, and the confidence-based Minimizer Function is proposed, which makes it more suitable for one-class object detection tasks. Secondly, to give the model the ability to judge easy and hard samples at the early stage of training by using the SPL strategy, an SPL strategy with Easy Sample Prior (ESP) is proposed. The FBOD model is trained using the standard training strategy with easy samples first, then the SPL strategy with all samples is used to train it. Combining the strategy of the ESP and the Minimizer Function based on confidence, the SPL-ESP-BC model training strategy is proposed. Using this strategy to train the FBOD model can make it to learn the characteristics of the flying bird object in the surveillance video better, from easy to hard. The experimental results show that compared with the standard training strategy that does not distinguish between easy and hard samples, the AP50 of the FBOD model trained by the SPL-ESP-BC is increased by 2.1%, and compared with other loss-based SPL strategies, the FBOD model trained with SPL-ESP-BC strategy has the best comprehensive detection performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。