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

GradPCA: Leveraging NTK Alignment for Reliable Out-of-Distribution Detection

About

We introduce GradPCA, an Out-of-Distribution (OOD) detection method that exploits the low-rank structure of neural network gradients induced by Neural Tangent Kernel (NTK) alignment. GradPCA applies Principal Component Analysis (PCA) to gradient class-means, achieving more consistent performance than existing methods across standard image classification benchmarks. We provide a theoretical perspective on spectral OOD detection in neural networks to support GradPCA, highlighting feature-space properties that enable effective detection and naturally emerge from NTK alignment. Our analysis further reveals that feature quality -- particularly the use of pretrained versus non-pretrained representations -- plays a crucial role in determining which detectors will succeed. Extensive experiments validate the strong performance of GradPCA, and our theoretical framework offers guidance for designing more principled spectral OOD detectors.

Mariia Seleznova, Hung-Hsu Chou, Claudio Mayrink Verdun, Gitta Kutyniok• 2025

Related benchmarks

TaskDatasetResultRank
OOD DetectionCIFAR-10 (IND) SVHN (OOD)
AUROC0.9973
131
OOD DetectionCIFAR-10 (ID) vs Places 365 (OOD)
AUROC99.3
117
OOD DetectionPlaces (OOD)
AUROC89.02
93
OOD DetectionCIFAR-10 IND iSUN OOD
AUROC98.74
82
OOD DetectionSUN (OOD)
AUROC91.27
81
OOD DetectionTextures (OOD) with CIFAR-10 (ID) (test)
FPR@950.02
80
OOD DetectionLSUN-Resize (OOD) with CIFAR-10 (ID) (test)
FPR@956.48
70
Out-of-Distribution DetectionCIFAR100 (ID) vs SVHN (OOD) (test)
AUROC96.58
67
Out-of-Distribution DetectionCIFAR-10 ID CIFAR-100 OOD
AUC93.68
66
OOD DetectionCIFAR-10 IND LSUN C OOD
AUROC99.45
60
Showing 10 of 20 rows

Other info

Follow for update