论文标题

有条件概率生成模型的开放式识别

Open Set Recognition with Conditional Probabilistic Generative Models

论文作者

Sun, Xin, Zhang, Chi, Lin, Guosheng, Ling, Keck-Voon

论文摘要

深层神经网络在广泛的视觉理解任务中取得了突破。阻碍其现实世界应用的典型挑战是,在测试阶段,未知样本可能会被馈入系统,但是传统的深层神经网络会错误地将这些未知样本识别为已知类别之一。开放式识别(OSR)是克服此问题的潜在解决方案,在该解决方案中,开放式分类器应具有灵活性拒绝未知样本并同时保持已知类别的高分类精度。概率生成模型,例如变异自动编码器(VAE)和对抗性自动编码器(AAE),是检测未知数的流行方法,但它们无法为已知分类提供歧视性表示。在本文中,我们提出了一个新型框架,称为条件概率生成模型(CPGM),以进行开放式识别。我们工作的核心洞察力是将歧视性信息添加到概率生成模型中,以便提出的模型不仅可以检测未知的样本,而且可以通过强迫不同的潜在特征来近似条件高斯分布来对已知类别进行分类。我们讨论许多模型变体,并提供全面的实验来研究其特征。在多个基准数据集上的实验结果表明,所提出的方法显着超过了基准,并实现了新的最新性能。

Deep neural networks have made breakthroughs in a wide range of visual understanding tasks. A typical challenge that hinders their real-world applications is that unknown samples may be fed into the system during the testing phase, but traditional deep neural networks will wrongly recognize these unknown samples as one of the known classes. Open set recognition (OSR) is a potential solution to overcome this problem, where the open set classifier should have the flexibility to reject unknown samples and meanwhile maintain high classification accuracy in known classes. Probabilistic generative models, such as Variational Autoencoders (VAE) and Adversarial Autoencoders (AAE), are popular methods to detect unknowns, but they cannot provide discriminative representations for known classification. In this paper, we propose a novel framework, called Conditional Probabilistic Generative Models (CPGM), for open set recognition. The core insight of our work is to add discriminative information into the probabilistic generative models, such that the proposed models can not only detect unknown samples but also classify known classes by forcing different latent features to approximate conditional Gaussian distributions. We discuss many model variants and provide comprehensive experiments to study their characteristics. Experiment results on multiple benchmark datasets reveal that the proposed method significantly outperforms the baselines and achieves new state-of-the-art performance.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源