arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.04851cs.LGstat.ML

你的分类器可能秘密地是一个基于似然的分布外检测器

Your Classifier Can Be Secretly a Likelihood-Based OOD Detector

Stanford University(斯坦福大学);University of Wisconsin-Madison(威斯康星大学麦迪逊分校)

Jirayu Burapacheep, Yixuan Li

AI总结:

针对判别式分类器缺乏显式输入似然的问题,本文提出INK,利用恒定范数超球面嵌入将分类器潜在表征解释为内在似然,并在OpenOOD的far/near-OOD设置中取得最先进性能。

摘要:

检测分布外(OOD)输入的能力,对于保障部署在开放环境中的分类模型的可靠性至关重要。OOD检测中的一个根本性挑战是:判别式分类器通常被训练为在给定输入z时估计类别y的后验概率p(y|z),但缺乏OOD检测理想情况下所需的对p(z)的显式似然估计。尽管已有许多用于分类模型的OOD评分函数被提出,但这些估计分数往往由启发式方法驱动,无法被严格解释为似然。为弥合这一差距,我们提出了Intrinsic Likelihood(INK,内在似然),它为现代基于判别式的分类器提供了严格的似然解释。具体而言,我们提出的INK分数作用于判别式分类器的受约束潜在嵌入,这些嵌入被建模为具有恒定范数的超球面嵌入混合。我们在超球面分布与内在似然之间建立了一种新颖联系,这种似然可以在现代神经网络中被有效优化。在OpenOOD基准上的大量实验从经验上表明,INK在包括far-OOD和near-OOD在内的多种OOD检测设置中确立了新的最先进水平。代码可在https://github.com/deeplearning-wisc/ink获取。

英文摘要:

The ability to detect out-of-distribution (OOD) inputs is critical to guarantee the reliability of classification models deployed in an open environment. A fundamental challenge in OOD detection is that a discriminative classifier is typically trained to estimate the posterior probability p(y|z) for class y given an input z, but lacks the explicit likelihood estimation of p(z) ideally needed for OOD detection. While numerous OOD scoring functions have been proposed for classification models, these estimate scores are often heuristic-driven and cannot be rigorously interpreted as likelihood. To bridge the gap, we propose Intrinsic Likelihood (INK), which offers rigorous likelihood interpretation to modern discriminative-based classifiers. Specifically, our proposed INK score operates on the constrained latent embeddings of a discriminative classifier, which are modeled as a mixture of hyperspherical embeddings with constant norm. We draw a novel connection between the hyperspherical distribution and the intrinsic likelihood, which can be effectively optimized in modern neural networks. Extensive experiments on the OpenOOD benchmark empirically demonstrate that INK establishes a new state-of-the-art in a variety of OOD detection setups, including both far-OOD and near-OOD. Code is available at https://github.com/deeplearning-wisc/ink.

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

↑