Back to Search Start Over

Progressive Hard-case Mining across Pyramid Levels for Object Detection

Authors :
Wu, Binghong
Yang, Yehui
Yang, Dalu
Wu, Junde
Wang, Xiaorong
Huang, Haifeng
Wang, Lei
Xu, Yanwu
Wu, Binghong
Yang, Yehui
Yang, Dalu
Wu, Junde
Wang, Xiaorong
Huang, Haifeng
Wang, Lei
Xu, Yanwu
Publication Year :
2021

Abstract

In object detection, multi-level prediction (e.g., FPN) and reweighting skills (e.g., focal loss) have drastically improved one-stage detector performance. However, the synergy between these two techniques is not fully explored in a unified framework. We find that, during training, the one-stage detector's optimization is not only restricted to the static hard-case mining loss (gradient drift) but also suffered from the diverse positive samples' proportions split by different pyramid levels (level discrepancy). Under this concern, we propose Hierarchical Progressive Focus (HPF) consisting of two key designs: 1) progressive focus, a more flexible hard-case mining setting calculated adaptive to the convergence progress, 2) hierarchical sampling, automatically generating a set of progressive focus for level-specific target optimization. Based on focal loss with ATSS-R50, our approach achieves 40.5 AP, surpassing the state-of-the-art QFL (Quality Focal Loss, 39.9 AP) and VFL (Varifocal Loss, 40.1 AP). Our best model achieves 55.1 AP on COCO test-dev, obtaining excellent results with only a typical training setting. Moreover, as a plug-and-play scheme, HPF can cooperate well with recent advances, providing a stable performance improvement on nine mainstream detectors.

Details

Database :
OAIster
Publication Type :
Electronic Resource
Accession number :
edsoai.on1269576286
Document Type :
Electronic Resource