Composite goodness-of-fit test with the Kernel Stein Discrepancy and a bootstrap for degenerate U-statistics with estimated parameters
Florian Brueck, Veronika Reimoser, Fabian Baier
TL;DR
This work addresses the problem of composite goodness-of-fit testing when the null encompasses a parametric family of distributions, using Kernel Stein Discrepancy (KSD) as the testing statistic. It derives the full asymptotic distribution of the composite KSD estimator under parameter estimation, revealing a limit that combines a weighted sum of chi-square terms with a disturbance from estimating the parameter, and provides a joint limit involving the estimator and score derivatives. To enable valid inference, the authors develop a general bootstrap CLT for degenerate, parameter-dependent U-statistics, showing that naive bootstrap procedures fail and introducing a corrected bootstrap term that delivers the correct limiting law; this framework is then specialized to the KSD. Simulations demonstrate that the proposed bootstrap-based KSD test maintains level under the null and exhibits higher power than competing MMD-based tests and the prior wild-bootstrap KSD approach, particularly in moderate to high dimensions. Overall, the paper offers both a rigorous theoretical foundation and a practical resampling strategy for composite goodness-of-fit testing with KSD in the presence of estimated parameters, with broader relevance to degenerate U-statistics beyond the KSD case.
Abstract
This paper formally derives the asymptotic distribution of a goodness-of-fit test based on the Kernel Stein Discrepancy introduced in (Oscar Key et al., "Composite Goodness-of-fit Tests with Kernels", Journal of Machine Learning Research 26.51 (2025), pp. 1-60). The test enables the simultaneous estimation of the optimal parameter within a parametric family of candidate models. Its asymptotic distribution is shown to be a weighted sum of infinitely many $χ^2$-distributed random variables plus an additional disturbance term, which is due to the parameter estimation. Further, we provide a general framework to bootstrap degenerate parameter-dependent $U$-statistics and use it to derive a new Kernel Stein Discrepancy composite goodness-of-fit test.
