基于余弦相似度的分类定位一致性损失

作者：叶英杰; 窦杰^*

来源：电光与控制, 2023, 30(11): 41-48.

• 目标检测
• 损失函数
• 非极大值抑制
• 分类定位一致性
• 余弦相似度

主流的目标检测模型将检测分为分类和定位两个子任务，分类和定位各自具有独立的子网络，且在训练过程中采用互相独立的损失函数。这种模型结构和训练方式忽略了分类和定位之间的相互联系，使得模型预测的类别得分无法体现预测框的定位质量，进一步导致高定位质量的预测在非极大值抑制(NMS)阶段被低定位质量的预测抑制，损害了模型的检测精度。针对该问题，提出了一种一致性损失的概念，该损失通过在训练过程中约束模型预测的类别得分和定位质量的排名相似度，提升了二者的一致程度。基于FCOS-ResNet50模型与PASCAL VOC数据集，所提的损失函数能够提升约1.3个百分点的mAP0.5、4.3个百分点的mAP75和5.4个百分点的mAP90。

• 单位
  瞬态物理国家重点实验室; 南京理工大学