计算机与现代化Issue(3):52-59,8.DOI:10.3969/j.issn.1006-2475.2025.03.008
重利用不可靠伪标签的单阶段半监督目标检测
One-stage Semi-supervised Object Detection by Reusing Unreliable Pseudo-labels
摘要
Abstract
The key to semi-supervised object detection methods is to assign pseudo labels to the targets of unlabeled data.To guarantee the quality of pseudo-labels,the semi-supervised object detection methods usually use a confidence threshold to filter low-quality pseudo-labels,which will cause most pseudo-labels to be removed due to their low confidence.Contrastive learning is used to reuse most of low-confidence unreliable pseudo labels for boosting the performance of semi-supervised object detection method.Specifically,the pseudo-labels are divided into reliable and unreliable ones according to the prediction confidence.Be-sides the reliable pseudo-labels,the unreliable pseudo-labels are exploited as negative samples for model training of contrast learning.To balance the number of unreliable pseudo-labels between different classes,a memory module is designed to store the unreliable pseudo-labels of different batches in the training process.The experimental results show that the mAP of the improved semi-supervised method on COCO data set is 13.6%,23.0%,and 27.5%with the labeling ratio of 1%,5%,and 10%,which is better than the existing semi-supervised learning methods.On the COCO-additional data set,the mAP of the improved semi-supervised method reaches 44.7%,which is 4.5 percentage points higher than supervised learning.关键词
半监督学习/目标检测/对比学习/重利用不可靠伪标签/端到端训练Key words
semi-supervised learning/object detection/contrastive learning/reusing unreliable pseudo-labels/end-to-end training分类
信息技术与安全科学引用本文复制引用
邵叶秦,王海权,周昆阳,郭于荻,施佺..重利用不可靠伪标签的单阶段半监督目标检测[J].计算机与现代化,2025,(3):52-59,8.基金项目
国家自然科学基金面上项目(61671255) (61671255)