Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Low-Dimensional Gradient Helps Out-of-Distribution Detection

About

Detecting out-of-distribution (OOD) samples is essential for ensuring the reliability of deep neural networks (DNNs) in real-world scenarios. While previous research has predominantly investigated the disparity between in-distribution (ID) and OOD data through forward information analysis, the discrepancy in parameter gradients during the backward process of DNNs has received insufficient attention. Existing studies on gradient disparities mainly focus on the utilization of gradient norms, neglecting the wealth of information embedded in gradient directions. To bridge this gap, in this paper, we conduct a comprehensive investigation into leveraging the entirety of gradient information for OOD detection. The primary challenge arises from the high dimensionality of gradients due to the large number of network parameters. To solve this problem, we propose performing linear dimension reduction on the gradient using a designated subspace that comprises principal components. This innovative technique enables us to obtain a low-dimensional representation of the gradient with minimal information loss. Subsequently, by integrating the reduced gradient with various existing detection score functions, our approach demonstrates superior performance across a wide range of detection tasks. For instance, on the ImageNet benchmark with ResNet50 model, our method achieves an average reduction of 11.15$\%$ in the false positive rate at 95$\%$ recall (FPR95) compared to the current state-of-the-art approach. The code would be released.

Yingwen Wu, Tao Li, Xinwen Cheng, Jie Yang, Xiaolin Huang• 2023

Related benchmarks

TaskDatasetResultRank
OOD DetectionCIFAR-10 (IND) SVHN (OOD)
AUROC0.901
131
OOD DetectionCIFAR-10 (ID) vs Places 365 (OOD)
AUROC88.36
117
OOD DetectionCIFAR-10 IND iSUN OOD
AUROC94.78
82
OOD DetectionTextures (OOD) with CIFAR-10 (ID) (test)
FPR@9512.52
80
OOD DetectionLSUN-Resize (OOD) with CIFAR-10 (ID) (test)
FPR@9523.61
70
Out-of-Distribution DetectionCIFAR100 (ID) vs SVHN (OOD) (test)
AUROC93.01
67
OOD DetectionCIFAR-10 IND LSUN C OOD
AUROC95.09
60
OOD DetectionCIFAR-10 Average OOD ID (test)
FPR@9525.59
40
Out-of-Distribution DetectionImageNet
FPR95 (Places)71.04
34
Showing 9 of 9 rows

Other info

Follow for update