Improving object detection quality with structural constraints

PLoS One. 2022 May 18;17(5):e0267863. doi: 10.1371/journal.pone.0267863. eCollection 2022.

Abstract

Recent researches revealed object detection networks using the simple "classification loss + localization loss" training objective are not effectively optimized in many cases, while providing additional constraints on network features could effectively improve object detection quality. Specifically, some works used constraints on training sample relations to successfully learn discriminative network features. Based on these observations, we propose Structural Constraint for improving object detection quality. Structural constraint supervises feature learning in classification and localization network branches with Fisher Loss and Equi-proportion Loss respectively, by requiring feature similarities of training sample pairs to be consistent with corresponding ground truth label similarities. Structural constraint could be applied to all object detection network architectures with the assist of our Proxy Feature design. Our experiment results showed that structural constraint mechanism is able to optimize object class instances' distribution in network feature space, and consequently detection results. Evaluations on MSCOCO2017 and KITTI datasets showed that our structural constraint mechanism is able to assist baseline networks to outperform modern counterpart detectors in terms of object detection quality.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Learning*

Grants and funding

This study was funded by the National Natural Science Foundation of China (https://www.nsfc.gov.cn) in the form of a grant [62172022] and by the Beijing Natural Science Foundation in the form of funds to DK. This study was also funded by the National Natural Science Foundation of China in the form of grants to BY [U1811463, U19B2039]. The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.