Semi-Supervised Object Detection in the Open World
Allabadi, Garvita, Lucic, Ana, Pao-Huang, Peter, Wang, Yu-Xiong, Adve, Vikram
–arXiv.org Artificial Intelligence
Existing approaches for semi-supervised object detection assume a fixed set of classes present in training and unlabeled datasets, i.e., in-distribution (ID) data. The performance of these techniques significantly degrades when these techniques are deployed in the open-world, due to the fact that the unlabeled and test data may contain objects that were not seen during training, i.e., out-of-distribution (OOD) data. The two key questions that we explore in this paper are: can we detect these OOD samples and if so, can we learn from them? With these considerations in mind, we propose the Open World Semi-supervised Detection framework (OWSSD) that effectively detects OOD data along with a semi-supervised learning pipeline that learns from both ID and OOD data. We introduce an ensemble based OOD detector consisting of lightweight auto-encoder networks trained only on ID data. Through extensive evalulation, we demonstrate that our method performs competitively against state-of-the-art OOD detection algorithms and also significantly boosts the semi-supervised learning performance in open-world scenarios.
arXiv.org Artificial Intelligence
Jul-28-2023
- Country:
- Asia > Middle East
- Israel (0.14)
- North America > United States (0.28)
- Asia > Middle East
- Genre:
- Research Report (0.82)
- Technology: