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

This is an offical PyTorch implementation of

Scaling for Training Time and Post-hoc Out-of-distribution Detection Enhancement.
ICLR 2024
Kai Xu, Rongyu Chen, Gianni Franchi, Angela Yao

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.

Environment and dataset

conda env create -f environment.yml

Prepare Dataset and pretrained network following OpenOOD official instruction. You need to prepare following dataset:

python ./scripts/download.py \
	--contents 'datasets' 'checkpoints' \
	--save_dir './data' './results' \
	--dataset_mode 'benchmark'

SCALE as post-hoc model enhancement.

python scripts/eval_ood_imagenet.py \
    --tvs-pretrained \
    --arch resnet50 \
    --postprocessor scale \
    --save-score --save-csv

ISH as training time model enhancement.

perform inference on ISH model:

python scripts/eval_ood_imagenet.py \
  --ckpt-path results/ish/ish_last.ckpt \
  --arch resnet50 \
  --postprocessor scale \
  --save-score --save-csv

You can download the model file from Google Drive. For the training code, please refer to OpenOOD.

Acknowledgment

Our Code is based on OpenOOD: Benchmarking Generalized OOD Detection.