• Electronics Optics & Control
  • Vol. 30, Issue 11, -1 (2023)
YE Yingjie and DOU Jie
Author Affiliations
  • [in Chinese]
  • show less
    DOI: 10.3969/j.issn.1671-637x.2023.11.007 Cite this Article
    YE Yingjie, DOU Jie. Consistency Loss Between Classification and Localization Based on Cosine Similarity[J]. Electronics Optics & Control, 2023, 30(11): -1 Copy Citation Text show less

    Abstract

    Object detection is decoupled into two subtasks, namely, classification and localization, in mainstream detectors. Each task possesses a separate detection subnetwork and is trained with an independent loss function. In this way, the correlation between classification and localization is disregarded, and thus the classification score predicted by the model is not capable of representing the localization quality of the prediction box. Consequently, predictions of high localization quality may be suppressed by their poorly localized counterparts in the procedure of Non-Maximum Suppression (NMS), inducing precision degradation. To tackle this problem, a consistency loss is proposed to constrain the rank similarity between the classification score predicted by the model and the localization quality in training process to reinforce about their consistency. Based on FCOS-ResNet50 model and PASCAL VOC dataset, the proposed loss function brings about 1.3 percentage points of mAP0.5, 4.3 percentage points of mAP75, and 5.4 percentage points of mAP90 improvements.
    YE Yingjie, DOU Jie. Consistency Loss Between Classification and Localization Based on Cosine Similarity[J]. Electronics Optics & Control, 2023, 30(11): -1
    Download Citation