PMAL: Open Set Recognition via Robust Prototype Mining

Authors

  • Jing Lu Hikvision Research Institute
  • Yunlu Xu Hikvision Research Institute
  • Hao Li Hikvision Research Institute
  • Zhanzhan Cheng Zhejiang University & Hikvision Research Institute
  • Yi Niu Hikvision Research Institute

DOI:

https://doi.org/10.1609/aaai.v36i2.20081

Keywords:

Computer Vision (CV), Machine Learning (ML)

Abstract

Open Set Recognition (OSR) has been an emerging topic. Besides recognizing predefined classes, the system needs to reject the unknowns. Prototype learning is a potential manner to handle the problem, as its ability to improve intra-class compactness of representations is much needed in discrimination between the known and the unknowns. In this work, we propose a novel Prototype Mining And Learning (PMAL) framework. It has a prototype mining mechanism before the phase of optimizing embedding space, explicitly considering two crucial properties, namely high-quality and diversity of the prototype set. Concretely, a set of high-quality candidates are firstly extracted from training samples based on data uncertainty learning, avoiding the interference from unexpected noise. Considering the multifarious appearance of objects even in a single category, a diversity-based strategy for prototype set filtering is proposed. Accordingly, the embedding space can be better optimized to discriminate therein the predefined classes and between known and unknowns. Extensive experiments verify the two good characteristics (i.e., high-quality and diversity) embraced in prototype mining, and show the remarkable performance of the proposed framework compared to state-of-the-arts.

Downloads

Published

2022-06-28

How to Cite

Lu, J., Xu, Y., Li, H., Cheng, Z., & Niu, Y. (2022). PMAL: Open Set Recognition via Robust Prototype Mining. Proceedings of the AAAI Conference on Artificial Intelligence, 36(2), 1872-1880. https://doi.org/10.1609/aaai.v36i2.20081

Issue

Section

AAAI Technical Track on Computer Vision II