Electronics Optics & Control, Volume. 30, Issue 11, -1(2023)

Consistency Loss Between Classification and Localization Based on Cosine Similarity

YE Yingjie and DOU Jie
Author Affiliations
  • [in Chinese]
  • show less

    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.

    Tools

    Get Citation

    Copy Citation Text

    YE Yingjie, DOU Jie. Consistency Loss Between Classification and Localization Based on Cosine Similarity[J]. Electronics Optics & Control, 2023, 30(11): -1

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: Nov. 2, 2022

    Accepted: --

    Published Online: Jan. 20, 2024

    The Author Email:

    DOI:10.3969/j.issn.1671-637x.2023.11.007

    Topics