PU Learning for RPN Generalization

Tengfei Hou,Ya Zhang,Yanfeng Wang
DOI: https://doi.org/10.1145/3447450.3447451
2020-01-01
Abstract:Region proposal is an important part of the two-stage object detection. RPNs pay more attention to the object within dataset, which has poor generalization and cannot generate objects that is unseen in the dataset. The traditional method does not rely on data but the performance is poor. Our work use the two-step method of PU learning to improve the generalization of RPN. In the first step, we identify reliable negative samples to train the basic RPN network. Then we consider rotation to extract more efficient pseudo-labels. Finally we use co-training to improve robustness about noise from pseudo-labeled. On Lvis dataset as an extension of COCO with more categories, the AR100 increase by 4%, and AR1000 can increase by 3%, which proves the reliability of the algorithm.
What problem does this paper attempt to address?