Code to reproduce results presented in the paper Controlling Neural Collapse Enhances Out-of-Distribution Detection and Transfer Learning
To use the code, simply run the following:
-
To measure OOD detection performance for VGG17 Baseline (without NC control):
ood_det_vgg.sh -
To measure OOD detection performance for VGG17 with NC control (Ours):
ood_det_vgg_nc.sh -
To measure OOD detection performance for ResNet18/34 Baseline (without NC control):
ood_det_resnet.sh -
To measure OOD detection performance for ResNet18/34 with NC control (Ours):
ood_det_resnet_nc.sh
To get results for other configures, change the relevant arguments.