Energy-based Hopfield Boosting for Out-of-Distribution Detection

Claus Hofmann, Simon Schmid, Bernhard Lehner, Daniel Klotz, Sepp Hochreiter
5/14/2024
cs.LGcs.CV

No official code found. Best implementation: ml-jku/hopfield-boosting (34 stars, MIT, last commit 2024-05-15).

Abstract

Out-of-distribution (OOD) detection is critical when deploying machine learning models in the real world. Outlier exposure methods, which incorporate auxiliary outlier data in the training process, can drastically improve OOD detection performance compared to approaches without advanced training strategies. We introduce Hopfield Boosting, a boosting approach, which leverages modern Hopfield energy (MHE) to sharpen the decision boundary between the in-distribution and OOD data. Hopfield Boosting encourages the model to concentrate on hard-to-distinguish auxiliary outlier examples that lie close to the decision boundary between in-distribution and auxiliary outlier data. Our method achieves a new state-of-the-art in OOD detection with outlier exposure, improving the FPR95 metric from 2.28 to 0.92 on CIFAR-10 and from 11.76 to 7.94 on CIFAR-100.

View on arXivView PDF

Code Implementations(1)

MIT
340Python, Jupyter NotebookFeb 8, 20242 years agoMIT

Similar Papers

Cite this paper

@article{hofmann2024energybased,
  title  = {Energy-based Hopfield Boosting for Out-of-Distribution Detection},
  author = {Claus Hofmann and Simon Schmid and Bernhard Lehner and Daniel Klotz and Sepp Hochreiter},
  year   = {2024},
  eprint = {2405.08766},
  archivePrefix = {arXiv},
  doi    = {10.48550/arXiv.2405.08766},
  url    = {http://arxiv.org/abs/2405.08766v2},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.08766.svg)](https://codeofpaper.com/paper/2405.08766)

Discussion