HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Semi-supervised novelty detection using ensembles with regularized disagreement

Alexandru Ţifrea Eric Stavarache Fanny Yang

Semi-supervised novelty detection using ensembles with regularized disagreement

Abstract

Deep neural networks often predict samples with high confidence even when they come from unseen classes and should instead be flagged for expert evaluation. Current novelty detection algorithms cannot reliably identify such near OOD points unless they have access to labeled data that is similar to these novel samples. In this paper, we develop a new ensemble-based procedure for semi-supervised novelty detection (SSND) that successfully leverages a mixture of unlabeled ID and novel-class samples to achieve good detection performance. In particular, we show how to achieve disagreement only on OOD data using early stopping regularization. While we prove this fact for a simple data distribution, our extensive experiments suggest that it holds true for more complex scenarios: our approach significantly outperforms state-of-the-art SSND methods on standard image data sets (SVHN/CIFAR-10/CIFAR-100) and medical image data sets with only a negligible increase in computation cost.

Code Repositories

ericpts/reto
Official
tf

Benchmarks

BenchmarkMethodologyMetrics
out-of-distribution-detection-on-cifar-10-vsERD (ResNet18)
AUROC: 95.1
out-of-distribution-detection-on-cifar-10-vs-10ERD (ResNet18)
AUROC: 91.4
out-of-distribution-detection-on-cifar-100-vsERD (ResNet18)
AUROC: 94.3

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp