HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Scaling for Training Time and Post-hoc Out-of-distribution Detection Enhancement

Kai Xu Rongyu Chen Gianni Franchi Angela Yao

Scaling for Training Time and Post-hoc Out-of-distribution Detection Enhancement

Abstract

The capacity of a modern deep learning system to determine if a sample falls within its realm of knowledge is fundamental and important. In this paper, we offer insights and analyses of recent state-of-the-art out-of-distribution (OOD) detection methods - extremely simple activation shaping (ASH). We demonstrate that activation pruning has a detrimental effect on OOD detection, while activation scaling enhances it. Moreover, we propose SCALE, a simple yet effective post-hoc network enhancement method for OOD detection, which attains state-of-the-art OOD detection performance without compromising in-distribution (ID) accuracy. By integrating scaling concepts into the training process to capture a sample's ID characteristics, we propose Intermediate Tensor SHaping (ISH), a lightweight method for training time OOD detection enhancement. We achieve AUROC scores of +1.85\% for near-OOD and +0.74\% for far-OOD datasets on the OpenOOD v1.5 ImageNet-1K benchmark. Our code and models are available at https://github.com/kai422/SCALE.

Code Repositories

kai422/scale
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
out-of-distribution-detection-on-far-oodSCALE (ResNet50)
AUROC: 96.53
FPR@95: 16.53
ID ACC: 76.18
out-of-distribution-detection-on-far-oodISH (ResNet50)
AUROC: 96.79
FPR@95: 15.62
ID ACC: 76.74
out-of-distribution-detection-on-imagenet-1k-10SCALE (ResNet50)
AUROC: 97.37
FPR95: 12.93
Latency, ms: 11.27
out-of-distribution-detection-on-imagenet-1k-12SCALE (ResNet50)
AUROC: 95.71
FPR95: 20.05
out-of-distribution-detection-on-imagenet-1k-3SCALE (ResNet50)
AUROC: 98.17
FPR95: 9.5
Latency, ms: 11.27
out-of-distribution-detection-on-imagenet-1k-8SCALE (ResNet50)
AUROC: 95.02
FPR95: 23.27
out-of-distribution-detection-on-imagenet-1k-9SCALE (ResNet50)
AUROC: 92.26
FPR95: 34.51
out-of-distribution-detection-on-near-oodSCALE (ResNet50)
AUROC: 81.36
FPR@95: 59.76
ID ACC: 76.18
out-of-distribution-detection-on-near-oodISH (ResNet50)
AUROC: 84.01
FPR@95: 55.73
ID ACC: 76.74

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