vix.ing · top · new · best · stats · spec

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

2025/05/21 by Mariia Seleznova, Seleznova, Mariia, Hung-Hsu Chou +5
Computer Science · Engineering · #Anomaly Detection Techniques and Applications #Computer Vision and Pattern Recognition (cs.CV) #Distributed Sensor Networks and Detection Algorithms #FOS: Computer and information sciences #Fault Detection and Control Systems #Machine Learning (cs.LG)

paper · pdf · doi:10.48550/arxiv.2505.16017

openalex publication_date 2025/05/21 · openalex created_date 2025/10/19 · openalex updated_date 2026/07/28

Abstract

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.

Citations

Related