GradPCA: Leveraging NTK Alignment for Reliable Out-of-Distribution Detection
GradPCA: Leveraging NTK Alignment for Reliable Out-of-Distribution Detection
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
计算技术、计算机技术
Mariia Seleznova,Hung-Hsu Chou,Claudio Mayrink Verdun,Gitta Kutyniok.GradPCA: Leveraging NTK Alignment for Reliable Out-of-Distribution Detection[EB/OL].(2025-05-21)[2025-07-02].https://arxiv.org/abs/2505.16017.点此复制
评论