LGAIFeb 7, 2022

Training OOD Detectors in their Natural Habitats

arXiv:2202.03299v2118 citations
AI Analysis

This addresses the challenge of reliable OOD detection for machine learning models deployed in real-world settings, offering a more practical approach by leveraging naturally occurring data.

The paper tackles the problem of out-of-distribution (OOD) detection by proposing a framework that uses wild mixture data containing both in-distribution and OOD samples, overcoming the limitation of existing methods that assume separable auxiliary outlier data, and demonstrates superior performance on common tasks.

Out-of-distribution (OOD) detection is important for machine learning models deployed in the wild. Recent methods use auxiliary outlier data to regularize the model for improved OOD detection. However, these approaches make a strong distributional assumption that the auxiliary outlier data is completely separable from the in-distribution (ID) data. In this paper, we propose a novel framework that leverages wild mixture data, which naturally consists of both ID and OOD samples. Such wild data is abundant and arises freely upon deploying a machine learning classifier in their natural habitats. Our key idea is to formulate a constrained optimization problem and to show how to tractably solve it. Our learning objective maximizes the OOD detection rate, subject to constraints on the classification error of ID data and on the OOD error rate of ID examples. We extensively evaluate our approach on common OOD detection tasks and demonstrate superior performance.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes