Efficient Aggregated Kernel Tests using Incomplete -statistics
Antonin Schrab, Ilmun Kim, Benjamin Guedj, Arthur Gretton
Abstract
We propose a series of computationally efficient nonparametric tests for the twosample, independence, and goodness-of-fit problems, using the Maximum Mean Discrepancy (MMD), Hilbert Schmidt Independence Criterion (HSIC), and Kernel Stein Discrepancy (KSD), respectively. Our test statistics are incomplete U -statistics, with a computational cost that interpolates between linear time in the number of samples, and quadratic time, as associated with classical U -statistic tests. The three proposed tests aggregate over several kernel bandwidths to detect departures from the null on various scales: we call the resulting tests MMDAggInc, HSICAggInc and KSDAggInc. This procedure provides a solution to the fundamental kernel selection problem as we can aggregate a large number of kernels with several bandwidths without incurring a significant loss of test power. For the test thresholds, we derive a quantile bound for wild bootstrapped incomplete U -statistics, which is of independent interest. We derive non-asymptotic uniform separation rates for MMDAggInc and HSICAggInc, and quantify exactly the tradeoff between computational efficiency and the attainable rates: this result is novel for tests based on incomplete U -statistics, to our knowledge. We further show that in the quadratic-time case, the wild bootstrap incurs no penalty to test power over the more widespread permutation-based approach, since both attain the same minimax optimal rates (which in turn match the rates that use oracle quantiles). We support our claims with numerical experiments on the trade-off between computational efficiency and test power. In all three testing frameworks, the linear-time versions of our proposed tests perform at least as well as the current linear-time state-of-the-art tests. Background In this section, we briefly describe our main problems of interest, comprising the two-sample, independence and goodness-of-fit problems. We approach these problems from a nonparametric point of view using the kernel-based statistics: MMD, HSIC, and KSD. We briefly introduce original forms of these statistics, which can be computed in quadratic time, and also discuss ways of calibrating tests proposed in the literature. The three quadratic-time expressions are presented in Appendix B.
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 926e1cf7-e222-4abb-be27-a5ca0bd22634Cited by top-tier papers14
- MMD-Fuse: Learning and Combining Kernels for Two-Sample Testing Without Data SplittingFelix Biggs, Antonin Schrab, Arthur GrettonNeurIPS 2023 · 49 citations
- KSD Aggregated Goodness-of-fit TestAntonin Schrab, Benjamin Guedj, Arthur GrettonNeurIPS 2022 · 26 citations
- Subspace Recovery from Heterogeneous Data with Non-isotropic NoiseJohn C. Duchi, Vitaly Feldman, Lunjia Hu, Kunal TalwarNeurIPS 2022 · 16 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
- On the Exploration of Local Significant Differences For Two-Sample TestZhijian Zhou, Jie Ni, Jia-He Yao, Wei GaoNeurIPS 2023 · 6 citations
Builds on4
- 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
- Learning Kernel Tests Without Data SplittingJonas M. Kübler, Wittawat Jitkrittum, Bernhard Schölkopf, Krikamol MuandetNeurIPS 2020 · 27 citations
- KSD Aggregated Goodness-of-fit TestAntonin Schrab, Benjamin Guedj, Arthur GrettonNeurIPS 2022 · 26 citations
- Post-selection inference with HSIC-LassoTobias Freidling, Benjamin Poignard, Héctor Climente-González, Makoto YamadaICML 2021 · 17 citations
Related papers
- A permutation-free kernel two-sample testShubhanshu Shekhar, Ilmun Kim, Aaditya RamdasNeurIPS 2022 · 40 citations
- Active Slices for Sliced Stein DiscrepancyWenbo Gong, Kaibo Zhang, Yingzhen Li, José Miguel Hernández-LobatoICML 2021 · 8 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
- A Kernel-based Test of Independence for Cluster-correlated DataHongjiao Liu, Anna M. Plantinga, Yunhua Xiang, Michael C. WuNeurIPS 2021 · 3 citations
