基于目标检测提议框信息的知识蒸馏模式

A KNOWLEDGE DISTILLATION MODEL BASED ON PROPOSAL BOXES INFORMATION OF OBJECT DETECTION

  • 摘要: 针对目标检测的蒸馏方法主要存在两个问题:第一,大都侧重于特征提取部分的蒸馏;第二,大都依赖GroundTruth(GT)确定蒸馏区域。针对上述问题,提出置信度排序蒸馏,对区域生成网络(RegionProposalNetwork,RPN)产生的提议框创建基于排序的自适应蒸馏模式;提出置信度引导特征蒸馏,探索一种利用教师网络所产生的提议框来指导蒸馏区域的新范式。实验表明,相较于基线模型,该算法在PASCALVOC数据集上提升了8.5百分点。

     

    Abstract: There are two main problems with current distillation methods for object detection: first, most focus on the distillation of the feature extraction part; second, most rely on ground truth (GT) to determine the distillation region. To address the above problems, we proposed confidence ranking distillation, and created an adaptive distillation model based on ranking for the proposal boxes generated by the region proposal network (RPN); proposed confidence-guided feature distillation, exploring a new paradigm of using the proposal boxes generated by the teacher network to guide distillation regions. Experiments show that the algorithm improves by 8.5 percentage points on the PASCAL VOC dataset compared with the baseline model.

     

/

返回文章
返回