Leveraging Perturbation Robustness to Enhance Out-of-Distribution Detection
About
Out-of-distribution (OOD) detection is the task of identifying inputs that deviate from the training data distribution. This capability is essential for safely deploying deep computer vision models in open-world environments. In this work, we propose a post-hoc method, Perturbation-Rectified OOD detection (PRO), based on the insight that prediction confidence for OOD inputs is more susceptible to reduction under perturbation than in-distribution (IND) inputs. Based on the observation, we propose an adversarial score function that searches for the local minimum scores near the original inputs by applying gradient descent. This procedure enhances the separability between IND and OOD samples. Importantly, the approach improves OOD detection performance without complex modifications to the underlying model architectures. We conduct extensive experiments using the OpenOOD benchmark~\cite{yang2022openood}. Our approach further pushes the limit of softmax-based OOD detection and is the leading post-hoc method for small-scale models. On a CIFAR-10 model with adversarial training, PRO effectively detects near-OOD inputs, achieving a reduction of more than 10\% on FPR@95 compared to state-of-the-art methods.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Graph Out-of-Distribution Detection | BZR (ID) COX2 (OOD) | AUC0.5205 | 49 | |
| Graph OOD Detection | IMDB-M IMDB-B | AUC0.3752 | 36 | |
| Graph Out-of-Distribution Detection | AIDS DHFR ID OOD | AUC86.64 | 33 | |
| Graph Out-of-Distribution Detection | ClinTox ID LIPO OOD | AUPR58.86 | 20 | |
| Graph Out-of-Distribution Detection | ClinTox ID LIPO OOD (test) | AUC0.57 | 20 | |
| Graph Out-of-Distribution Detection | AIDS (ID) DHFR (OOD) | AUPR74.7 | 20 | |
| Graph Out-of-Distribution Detection | TUDataset Graph OOD Suite (BZR, PTC_MR, AIDS, ENZYMES, IMDB-M, Tox21, FreeSolv, BBBP, ClinTox, Esol) (test) | FPR95% (BZR/COX2)73.17 | 20 | |
| Graph Out-of-Distribution Detection | ENZYMES ID PROTEIN OOD (test) | AUC55.71 | 20 | |
| Graph Out-of-Distribution Detection | ENZYMES PROTEIN ID OOD | AUPR54.14 | 20 | |
| Graph Out-of-Distribution Detection | Tox21 ID SIDER OOD | AUPR44.98 | 20 |