Reconcile Prediction Consistency for Balanced Object Detection
Keyang Wang, Lei Zhang
Abstract
Classification and regression are two pillars of object detectors. In most CNN-based detectors, these two pillars are optimized independently. Without direct interactions be-tween them, the classification loss and the regression loss can not be optimized synchronously toward the optimal direction in the training phase. This clearly leads to lots of inconsistent predictions with high classification score but low localization accuracy or low classification score but high localization accuracy in the inference phase, especially for the objects of irregular shape and occlusion, which severely hurts the detection performance of existing detectors after N-MS. To reconcile prediction consistency for balanced object detection, we propose a Harmonic loss to harmonize the optimization of classification branch and localization branch. The Harmonic loss enables these two branches to supervise and promote each other during training, thereby producing consistent predictions with high co-occurrence of top classification and localization in the inference phase. Furthermore, in order to prevent the localization loss from being dominated by outliers during training phase, a Harmonic IoU loss is proposed to harmonize the weight of the localization loss of different IoU-level samples. Comprehensive experiments on benchmarks PASCAL VOC and MS COCO demonstrate the generality and effectiveness of our model for facilitating existing object detectors to state-of-the-art accuracy.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers3
- Localization Distillation for Dense Object DetectionZhaohui Zheng, Rongguang Ye, Ping Wang, Dongwei Ren et al.CVPR 2022 · 177 citations
- Ranking-Based Siamese Visual TrackingFeng Tang, Qiang LingCVPR 2022 · 87 citations
- Harmonious Teacher for Cross-Domain Object DetectionJinhong Deng, Dongli Xu, Wen Li, Lixin DuanCVPR 2023
Builds on5
- FCOS: Fully Convolutional One-Stage Object DetectionZhi Tian, Chunhua Shen, Hao Chen, Tong HeICCV 2019 · 6,042 citations
- CenterNet: Keypoint Triplets for Object DetectionKaiwen Duan, Song Bai, Lingxi Xie, Honggang Qi et al.ICCV 2019 · 3,348 citations
- Scale-Aware Trident Networks for Object DetectionYanghao Li, Yuntao Chen, Naiyan Wang, Zhaoxiang ZhangICCV 2019 · 1,031 citations
- Single-Shot Two-Pronged Detector with Rectified IoU LossKeyang Wang, Lei ZhangACM MM 2020 · 29 citations
- Prime Sample Attention in Object DetectionYuhang Cao, Kai Chen, Chen Change Loy, Dahua LinCVPR 2020
Related papers
- Correlation Loss: Enforcing Correlation between Classification and LocalizationFehmi Kahraman, Kemal Oksuz, Sinan Kalkan, Emre AkbasAAAI 2023 · 10 citations
- RankDetNet: Delving Into Ranking Constraints for Object DetectionJi Liu, Dong Li, Rongzhang Zheng, Lu Tian et al.CVPR 2021
- SCALoss: Side and Corner Aligned Loss for Bounding Box RegressionTu Zheng, Shuai Zhao, Yang Liu, Zili Liu et al.AAAI 2022 · 14 citations
- A Ranking-based, Balanced Loss Function Unifying Classification and Localisation in Object DetectionKemal Oksuz, Baris Can Cam, Emre Akbas, Sinan KalkanNeurIPS 2020 · 48 citations
- Generalized Focal Loss: Learning Qualified and Distributed Bounding Boxes for Dense Object DetectionXiang Li, Wenhai Wang, Lijun Wu, Shuo Chen et al.NeurIPS 2020 · 2,118 citations
