基于分类不确定性的伪标签目标检测算法

Pseudo-Label Object Detection Algorithm Based on Classification Uncertainty

伪标签目标检测算法利用大量未标注数据生成伪标签数据来增加训练数据规模,从而提高目标检测模型的性能。针对伪标签数据中存在大量错误标注数据且伪标签目标检测模型性能难以提升的问题,提出基于SoftTeacher-CUC的伪标签目标检测算法。SoftTeacher-CUC算法在SoftTeacher伪标签目标检测算法的基础上,利用分类不确定性方法计算模型生成的伪标签分类结果的不确定性来判断伪标签是否可靠,不确定性越低说明伪标签的分类结果越可靠。在此基础上,将计算得到的不确定性作为权重加入伪标签数据的分类损失函数中,进一步减少高不确定性伪标签为模型带来的负面影响。根据Teacher模型中不同模块的作用,采用不同权重的指数滑动平均方法更新Teacher模型,降低Teacher模型和Student模型参数之间的相似性,使一致性正则化方法发挥效用。实验结果表明,在标注数据分别占训练集1%、5%和10%的情况下,与SoftTeacher算法相比,SoftTeacher-CUC算法的平均精度均值分别提高了1.4、1.2和1.7个百分点,在标注数据较少的情况下,该算法具有更好的检测效果。

The pseudo-label object detection algorithm aims to increase the training data size by using a large amount of unlabeled data to generate pseudo-label data in order to improve the performance of the object detection model. To mitigate the problem of a large amount of incorrectly labeled data in the pseudo-label data and that the performance of the pseudo-label object detection model is difficult to improve,this paper proposes a pseudo-label object detection algorithm of SoftTeacher-CUC. The SoftTeacher-CUC algorithm is based on the SoftTeacher pseudo-label object detection algorithm.First,the classification uncertainty calculation method calculates the uncertainty of the pseudo-label classification results generated by the model to determine whether the pseudo-label is reliable.The lower the uncertainty,the more reliable the pseudo-label classification results. Then,the calculated uncertainty is added as a weight to the classification loss of the pseudo-label data to further reduce the negative impact of high-uncertainty pseudo-labels on the model. Finally,according to the role of the different modules of the Teacher model,an Exponential Moving Average(EMA) method with different weights is used to update the Teacher model. It reduces the similarity between the parameters of the Teacher model and those of the Student model to enable the consistency regularization method to work.The experimental results show that the labeled data accounts for 1%,5%,and 10% of the training set.Compared with the SoftTeacher algorithm,the mAP of the SoftTeacher-CUC algorithm is improved by 1.4,1.2,and 1.7 percentage points,respectively.Consequently,SoftTeacher-CUC algorithm has a better detection effect when there is less labeled data.