arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 1907.04572cs.LGcs.CVstat.ML
收藏

基于神经渲染生成模型的分布外检测

Out-of-Distribution Detection Using Neural Rendering Generative Models

California Institute of Technology(加州理工学院);Rice University(莱斯大学)

Yujia Huang, Sihui Dai, Tan Nguyen, Richard G. Baraniuk, Anima Anandkumar

摘要:

分布外(OoD)检测是深度生成模型的自然下游任务,因为它们能够学习输入概率分布。主要有两类使用深度生成模型的OoD检测方法,即,基于似然测度和重建损失。然而,这两种方法都不能有效地进行OoD检测,特别是当OoD样本的方差小于训练样本时。例如,当在CIFAR-10图像上训练时,基于流的模型和VAE模型都为来自SVHN的图像分配了更高的可能性。我们使用最近提出的生成模型称为神经渲染模型(NRM)和OoD的指标。我们表明,NRM统一了这两种方法,因为它提供了一个可能性估计,也进行重建的神经网络的每一层。在各种措施中,我们发现潜在变量的联合可能性是最有效的OoD检测。我们的研究结果表明,当在CIFAR-10上训练时,SVHN图像的潜在变量的可能性较低。此外,我们还表明,这个指标在其他OoD数据集上是一致的。据我们所知,这是第一个显示OoD数据的可能性始终较低的工作,具有更小的深度生成模型方差。

英文摘要:

Out-of-distribution (OoD) detection is a natural downstream task for deep generative models, due to their ability to learn the input probability distribution. There are mainly two classes of approaches for OoD detection using deep generative models, viz., based on likelihood measure and the reconstruction loss. However, both approaches are unable to carry out OoD detection effectively, especially when the OoD samples have smaller variance than the training samples. For instance, both flow based and VAE models assign higher likelihood to images from SVHN when trained on CIFAR-10 images. We use a recently proposed generative model known as neural rendering model (NRM) and derive metrics for OoD. We show that NRM unifies both approaches since it provides a likelihood estimate and also carries out reconstruction in each layer of the neural network. Among various measures, we found the joint likelihood of latent variables to be the most effective one for OoD detection. Our results show that when trained on CIFAR-10, lower likelihood (of latent variables) is assigned to SVHN images. Additionally, we show that this metric is consistent across other OoD datasets. To the best of our knowledge, this is the first work to show consistently lower likelihood for OoD data with smaller variance with deep generative models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑