KSD Aggregated Goodness-of-fit Test
Antonin Schrab, Benjamin Guedj, Arthur Gretton
Abstract
We investigate properties of goodness-of-fit tests based on the Kernel Stein Discrepancy (KSD). We introduce a strategy to construct a test, called KSDAgg, which aggregates multiple tests with different kernels. KSDAgg avoids splitting the data to perform kernel selection (which leads to a loss in test power), and rather maximises the test power over a collection of kernels. We provide non-asymptotic guarantees on the power of KSDAgg: we show it achieves the smallest uniform separation rate of the collection, up to a logarithmic term. For compactly supported densities with bounded model score function, we derive the rate for KSDAgg over restricted Sobolev balls; this rate corresponds to the minimax optimal rate over unrestricted Sobolev balls, up to an iterated logarithmic term. KSDAgg can be computed exactly in practice as it relies either on a parametric bootstrap or on a wild bootstrap to estimate the quantiles and the level corrections. In particular, for the crucial choice of bandwidth of a fixed kernel, it avoids resorting to arbitrary heuristics (such as median or standard deviation) or to data splitting. We find on both synthetic and real-world data that KSDAgg outperforms other state-of-the-art quadratic-time adaptive KSD-based goodness-of-fit testing procedures.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 6bd012e4-1daa-46ff-9dd2-ea137cb7cd3bCited by top-tier papers8
- MMD-Fuse: Learning and Combining Kernels for Two-Sample Testing Without Data SplittingFelix Biggs, Antonin Schrab, Arthur GrettonNeurIPS 2023 · 49 citations
- Efficient Aggregated Kernel Tests using Incomplete -statisticsAntonin Schrab, Ilmun Kim, Benjamin Guedj, Arthur GrettonNeurIPS 2022 · 42 citations
- DUAL: Learning Diverse Kernels for Aggregated Two-sample and Independence TestingZhijian Zhou, Xunye Tian, Liuhua Peng, Chao Lei et al.NeurIPS 2025 · 8 citations
- Using Perturbation to Improve Goodness-of-Fit Tests based on Kernelized Stein DiscrepancyXing Liu, Andrew B. Duncan, Axel GandyICML 2023 · 8 citations
- Minimax Optimal Rate for Parameter Estimation in Multivariate Deviated ModelsDat Do, Huy Nguyen, Khai Nguyen, Nhat HoNeurIPS 2023 · 5 citations
Builds on10
- Learning Deep Kernels for Non-Parametric Two-Sample TestsFeng Liu, Wenkai Xu, Jie Lu, Guangquan Zhang et al.ICML 2020 · 213 citations
- Learning the Stein Discrepancy for Training and Evaluating Energy-Based Models without SamplingWill Grathwohl, Kuan-Chieh Wang, Jörn-Henrik Jacobsen, David Duvenaud et al.ICML 2020 · 93 citations
- Kernel Stein Discrepancy DescentAnna Korba, Pierre-Cyril Aubin-Frankowski, Szymon Majewski, Pierre AblinICML 2021 · 64 citations
- Efficient Aggregated Kernel Tests using Incomplete -statisticsAntonin Schrab, Ilmun Kim, Benjamin Guedj, Arthur GrettonNeurIPS 2022 · 42 citations
- Learning Kernel Tests Without Data SplittingJonas M. Kübler, Wittawat Jitkrittum, Bernhard Schölkopf, Krikamol MuandetNeurIPS 2020 · 27 citations
Related papers
- Active Slices for Sliced Stein DiscrepancyWenbo Gong, Kaibo Zhang, Yingzhen Li, José Miguel Hernández-LobatoICML 2021 · 8 citations
- A Kernel Stein Test of Goodness of Fit for Sequential ModelsJerome Baum, Heishiro Kanagawa, Arthur GrettonICML 2023 · 12 citations
- Sliced Kernelized Stein DiscrepancyWenbo Gong, Yingzhen Li, José Miguel Hernández-LobatoICLR 2021 · 14 citations
- Kernelized Stein Discrepancy Tests of Goodness-of-fit for Time-to-Event DataTamara Fernandez, Nicolas Rivera, Wenkai Xu, Arthur GrettonICML 2020 · 16 citations
- The Polynomial Stein Discrepancy for Assessing Moment ConvergenceNarayan Srinivasan, Matthew Sutton, Christopher C. Drovandi, Leah F. SouthICML 2025
