Unified Open-Set Recognition and Novel Class Discovery via Prototype-Guided Representation
Jiuqing Dong, Sicheng Wang, Jianxin Xue, Siwen Zhang, Zixin Li, Heng Zhou · Applied Sciences · 2025
The existing research on open-set recognition (OSR) and novel class discovery (NCD) has largely treated these tasks as independent fields. OSR aims to identify samples that do not belong to the training set classes, while NCD seeks to further classify such unseen, unlabeled samples into novel classes. However, there is a lack of a unified framework to automate both tasks systematically. In this paper, we propose a unified training framework to identify and categorize unseen samples. Specifically, we conduct a comprehensive evaluation of existing post hoc OSR methods and observe that their performance is highly sensitive to the temperature scaling factor. To address this, we introduce a distance-based evaluation method for OSR, which not only outperforms existing post hoc approaches but also integrates seamlessly with them to deliver enhanced performance. Furthermore, we developed a prototype-based classification head leveraging this distance metric, which facilitates compact feature representations for known classes and guides the clustering of unknown classes, thereby significantly enhancing the classification accuracy for novel classes. On the CUB-200-2011 dataset, our unified framework achieves a 0.95–6.12% improvement in AUROC scores on OSR benchmarks and a 3.19% increase in classification accuracy for novel classes. Extensive experiments and visualizations validate the effectiveness of the proposed approach. We believe that this unified framework will pave the way for automating the integration of OSR and NCD, offering a more efficient and systematic approach to addressing these tasks.