MOODv2: Masked Image Modeling for Out-of-Distribution Detection

Jingyao Li, Pengguang Chen, Shaozuo Yu, Shu Liu, Jiaya Jia
1/5/2024
cs.CV

No official code found. Best implementation: JIA-Lab-research/MOOD (137 stars, PyTorch, last commit 2024-07-02).

Abstract

The crux of effective out-of-distribution (OOD) detection lies in acquiring a robust in-distribution (ID) representation, distinct from OOD samples. While previous methods predominantly leaned on recognition-based techniques for this purpose, they often resulted in shortcut learning, lacking comprehensive representations. In our study, we conducted a comprehensive analysis, exploring distinct pretraining tasks and employing various OOD score functions. The results highlight that the feature representations pre-trained through reconstruction yield a notable enhancement and narrow the performance gap among various score functions. This suggests that even simple score functions can rival complex ones when leveraging reconstruction-based pretext tasks. Reconstruction-based pretext tasks adapt well to various score functions. As such, it holds promising potential for further expansion. Our OOD detection framework, MOODv2, employs the masked image modeling pretext task. Without bells and whistles, MOODv2 impressively enhances 14.30% AUROC to 95.68% on ImageNet and achieves 99.98% on CIFAR-10.

View on arXivView PDF

Code Implementations(1)

PyTorch

Official PyTorch implementation of MOOD series: (1) MOODv1: Rethinking Out-of-distributionDetection: Masked Image Modeling Is All You Need. (2) MOODv2: Masked Image Modeling for Out-of-Distribution Detection.

1377PythonFeb 6, 20232 years ago
cvpr2023masked-image-modelingood-detectionoutlier-detectionpytorch

Similar Papers

Cite this paper

@article{li2024moodv,
  title  = {MOODv2: Masked Image Modeling for Out-of-Distribution Detection},
  author = {Jingyao Li and Pengguang Chen and Shaozuo Yu and Shu Liu and Jiaya Jia},
  year   = {2024},
  eprint = {2401.02611},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2401.02611v1}
}

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/2401.02611.svg)](https://codeofpaper.com/paper/2401.02611)

Discussion