Springe direkt zu Inhalt

WeiPer: OOD Detection using Weight Perturbations of Class Projections

Tim Landgraf, Manuel Heurich, Maximilian Granz – 2024

Recent advances in out-of-distribution (OOD) detection on image data show that pre-trained neural network classifiers can separate in-distribution (ID) from OOD data well, leveraging the class-discriminative ability of the model itself. Methods have been proposed that either use logit information directly or that process the model's penultimate layer activations. With "WeiPer", we introduce perturbations of the class projections in the final fully connected layer which creates a richer representation of the input. We show that this simple trick can improve the OOD detection performance of a variety of methods and additionally propose a distance-based method that leverages the properties of the augmented WeiPer space. We achieve state-of-the-art OOD detection results across multiple benchmarks of the OpenOOD framework, especially pronounced in difficult settings in which OOD samples are positioned close to the training set distribution. We support our findings with theoretical motivations and empirical observations, and run extensive ablations to provide insights into why WeiPer works. Our code is available at: https://github.com/mgranz/weiper.

Titel
WeiPer: OOD Detection using Weight Perturbations of Class Projections
Verfasser
Tim Landgraf, Manuel Heurich, Maximilian Granz
Verlag
Curran Associates Inc.
Schlagwörter
WeiPer; OOD Detection
Datum
2024
Kennung
ISBN: 9798331314385
Quelle/n
Erschienen in
Advances in Neural Information Processing Systems 37 (NeurIPS 2024) Main Conference Track
Größe oder Länge
pp 35879-35908
BibTeX Code
@inproceedings{NEURIPS2024_3effb915,
author = {Granz, Maximilian and Heurich, Manuel and Landgraf, Tim},
booktitle = {Advances in Neural Information Processing Systems},
editor = {A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang},
pages = {35879--35908},
publisher = {Curran Associates, Inc.},
title = {WeiPer: OOD Detection using Weight Perturbations of Class Projections},
url = {https://proceedings.neurips.cc/paper_files/paper/2024/file/3effb91593c4fb42b1da1528328eff49-Paper-Conference.pdf},
volume = {37},
year = {2024}
}