Efficient Aggregated Kernel Tests using Incomplete -statistics
Antonin Schrab, Ilmun Kim, Benjamin Guedj, Arthur Gretton
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper14
- MMD-Fuse: Learning and Combining Kernels for Two-Sample Testing Without Data SplittingFelix Biggs, Antonin Schrab, Arthur GrettonNeurIPS 2023 · 被引用 49 次
- KSD Aggregated Goodness-of-fit TestAntonin Schrab, Benjamin Guedj, Arthur GrettonNeurIPS 2022 · 被引用 26 次
- Subspace Recovery from Heterogeneous Data with Non-isotropic NoiseJohn C. Duchi, Vitaly Feldman, Lunjia Hu, Kunal TalwarNeurIPS 2022 · 被引用 16 次
- DUAL: Learning Diverse Kernels for Aggregated Two-sample and Independence TestingZhijian Zhou, Xunye Tian, Liuhua Peng, Chao Lei 等NeurIPS 2025 · 被引用 8 次
- On the Exploration of Local Significant Differences For Two-Sample TestZhijian Zhou, Jie Ni, Jia-He Yao, Wei GaoNeurIPS 2023 · 被引用 6 次
它引用的顶会 Paper4
- Learning the Stein Discrepancy for Training and Evaluating Energy-Based Models without SamplingWill Grathwohl, Kuan-Chieh Wang, Jörn-Henrik Jacobsen, David Duvenaud 等ICML 2020 · 被引用 93 次
- Learning Kernel Tests Without Data SplittingJonas M. Kübler, Wittawat Jitkrittum, Bernhard Schölkopf, Krikamol MuandetNeurIPS 2020 · 被引用 27 次
- KSD Aggregated Goodness-of-fit TestAntonin Schrab, Benjamin Guedj, Arthur GrettonNeurIPS 2022 · 被引用 26 次
- Post-selection inference with HSIC-LassoTobias Freidling, Benjamin Poignard, Héctor Climente-González, Makoto YamadaICML 2021 · 被引用 17 次
相关 Paper
- A permutation-free kernel two-sample testShubhanshu Shekhar, Ilmun Kim, Aaditya RamdasNeurIPS 2022 · 被引用 40 次
- Active Slices for Sliced Stein DiscrepancyWenbo Gong, Kaibo Zhang, Yingzhen Li, José Miguel Hernández-LobatoICML 2021 · 被引用 8 次
- Kernelized Stein Discrepancy Tests of Goodness-of-fit for Time-to-Event DataTamara Fernandez, Nicolas Rivera, Wenkai Xu, Arthur GrettonICML 2020 · 被引用 16 次
- 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 次
