A permutation-free kernel two-sample test
Shubhanshu Shekhar, Ilmun Kim, Aaditya Ramdas
Abstract
The kernel Maximum Mean Discrepancy (MMD) is a popular multivariate distance metric between distributions that has found utility in two-sample testing. The usual kernel-MMD test statistic is a degenerate U-statistic under the null, and thus it has an intractable limiting distribution. Hence, to design a level- test, one usually selects the rejection threshold as the -quantile of the permutation distribution. The resulting nonparametric test has finite-sample validity but suffers from large computational cost, since every permutation takes quadratic time. We propose the cross-MMD, a new quadratic-time MMD test statistic based on sample-splitting and studentization. We prove that under mild assumptions, the cross-MMD has a limiting standard Gaussian distribution under the null. Importantly, we also show that the resulting test is consistent against any fixed alternative, and when using the Gaussian kernel, it has minimax rate-optimal power against local alternatives. For large sample sizes, our new cross-MMD provides a significant speedup over the MMD, for only a slight loss in power.
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 68a33733-b867-4278-91ca-e67525805b5fCited by top-tier papers9
- MMD-Fuse: Learning and Combining Kernels for Two-Sample Testing Without Data SplittingFelix Biggs, Antonin Schrab, Arthur GrettonNeurIPS 2023 · 49 citations
- AutoML Two-Sample TestJonas M. Kübler, Vincent Stimper, Simon Buchholz, Krikamol Muandet et al.NeurIPS 2022 · 29 citations
- An Efficient Doubly-Robust Test for the Kernel Treatment EffectDiego Martinez-Taboada, Aaditya Ramdas, Edward KennedyNeurIPS 2023 · 17 citations
- A Bias-Variance-Covariance Decomposition of Kernel Scores for Generative ModelsSebastian Gregor Gruber, Florian BuettnerICML 2024 · 6 citations
- On the Exploration of Local Significant Differences For Two-Sample TestZhijian Zhou, Jie Ni, Jia-He Yao, Wei GaoNeurIPS 2023 · 6 citations
Related papers
- Efficient Aggregated Kernel Tests using Incomplete -statisticsAntonin Schrab, Ilmun Kim, Benjamin Guedj, Arthur GrettonNeurIPS 2022 · 42 citations
- Neural Tangent Kernel Maximum Mean DiscrepancyXiuyuan Cheng, Yao XieNeurIPS 2021 · 26 citations
- Kernel-based Maximum-of-difference Test for Two-sample ComparisonDan Pu, Tianyi Zhu, Yao Yan, Wei LanICML 2026
- Learning Kernel Tests Without Data SplittingJonas M. Kübler, Wittawat Jitkrittum, Bernhard Schölkopf, Krikamol MuandetNeurIPS 2020 · 27 citations
- Kernel Quantile Embeddings and Associated Probability MetricsMasha Naslidnyk, Siu Lun Chau, François-Xavier Briol, Krikamol MuandetICML 2025
