Semiparametric KSD test: unifying score and distance-based approaches for goodness-of-fit testing
Zhihan Huang, Ziang Niu
TL;DR
This work reframes GoF testing through exponentially tilted models, showing that score-based tests are equivalent to IPM-based tests when indexed by a suitable function class. It unifies KS, Wasserstein, MMD, and KSD within a single score-based perspective and introduces SKSD, a computationally efficient, semiparametric GoF test using kernelized Stein functions that accommodates nuisance estimators via bootstrap. The SKSD framework is shown to be universally consistent and Pitman efficient, and it applies to models with intractable likelihoods by relying on tractable scores and Stein identities. Through theoretical guarantees and extensive experiments (normality, kernel exponential families, and conditional Gaussian graphs), SKSD demonstrates competitive power with task-specific tests and favorable scalability, suggesting broad applicability for complex GoF diagnostics in high dimensions.
Abstract
Goodness-of-fit (GoF) tests are fundamental for assessing model adequacy. Score-based tests are appealing because they require fitting the model only once under the null. However, extending them to powerful nonparametric alternatives is difficult due to the lack of suitable score functions. Through a class of exponentially tilted models, we show that the resulting score-based GoF tests are equivalent to the tests based on integral probability metrics (IPMs) indexed by a function class. When the class is rich, the test is universally consistent. This simple yet insightful perspective enables reinterpretation of classical distance-based testing procedures-including those based on Kolmogorov-Smirnov distance, Wasserstein-1 distance, and maximum mean discrepancy-as arising from score-based constructions. Building on this insight, we propose a new nonparametric score-based GoF test through a special class of IPM induced by kernelized Stein's function class, called semiparametric kernelized Stein discrepancy (SKSD) test. Compared with other nonparametric score-based tests, the SKSD test is computationally efficient and accommodates general nuisance-parameter estimators, supported by a generic parametric bootstrap procedure. The SKSD test is universally consistent and attains Pitman efficiency. Moreover, SKSD test provides simple GoF tests for models with intractable likelihoods but tractable scores with the help of Stein's identity and we use two popular models, kernel exponential family and conditional Gaussian models, to illustrate the power of our method. Our method achieves power comparable to task-specific normality tests such as Anderson-Darling and Lilliefors, despite being designed for general nonparametric alternatives.
