Command Palette
Search for a command to run...
Scaling for Training Time and Post-hoc Out-of-distribution Detection Enhancement
Kai Xu Rongyu Chen Gianni Franchi Angela Yao

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
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| out-of-distribution-detection-on-far-ood | SCALE (ResNet50) | AUROC: 96.53 FPR@95: 16.53 ID ACC: 76.18 |
| out-of-distribution-detection-on-far-ood | ISH (ResNet50) | AUROC: 96.79 FPR@95: 15.62 ID ACC: 76.74 |
| out-of-distribution-detection-on-imagenet-1k-10 | SCALE (ResNet50) | AUROC: 97.37 FPR95: 12.93 Latency, ms: 11.27 |
| out-of-distribution-detection-on-imagenet-1k-12 | SCALE (ResNet50) | AUROC: 95.71 FPR95: 20.05 |
| out-of-distribution-detection-on-imagenet-1k-3 | SCALE (ResNet50) | AUROC: 98.17 FPR95: 9.5 Latency, ms: 11.27 |
| out-of-distribution-detection-on-imagenet-1k-8 | SCALE (ResNet50) | AUROC: 95.02 FPR95: 23.27 |
| out-of-distribution-detection-on-imagenet-1k-9 | SCALE (ResNet50) | AUROC: 92.26 FPR95: 34.51 |
| out-of-distribution-detection-on-near-ood | SCALE (ResNet50) | AUROC: 81.36 FPR@95: 59.76 ID ACC: 76.18 |
| out-of-distribution-detection-on-near-ood | ISH (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.