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 Neural Collapse (NC) for VGG17 Baseline (without NC control):
nc_eval_vgg.sh -
To measure Neural Collapse (NC) for VGG17 with NC control (Ours):
nc_eval_vgg_nc.sh -
To measure Neural Collapse (NC) for ResNet18/34 Baseline (without NC control):
nc_eval_resnet.sh -
To measure Neural Collapse (NC) for ResNet18/34 with NC control (Ours):
nc_eval_resnet_nc.sh
To get results for other configures, change the relevant arguments.