Antonin Schrab

dblp:305/0565 · DBLP profile ↗
← Back
7ranked-venue papers
4as first author
7since 2021 · last 2025
0000-0002-3218-8613ORCID · verified

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 7 · 4 first-author · 7 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
5 papers
Learning theory · 62% Kernel, tree and ensemble methods · 27% Probabilistic and Bayesian machine learning · 11%

Topics — the 14 heaviest of 14, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Learning theory
hypothesis testing
2.742025
DUAL: Learning Diverse Kernels for Aggregated Two-sample and Independence Testing · NeurIPS 2025
MMD-Fuse: Learning and Combining Kernels for Two-Sample Testing Without Data Splitting · NeurIPS 2023
Efficient Aggregated Kernel Tests using Incomplete $U$-statistics · NeurIPS 2022
Machine learning › Kernel, tree and ensemble methods
kernel methods
2.742025
DUAL: Learning Diverse Kernels for Aggregated Two-sample and Independence Testing · NeurIPS 2025
MMD-Fuse: Learning and Combining Kernels for Two-Sample Testing Without Data Splitting · NeurIPS 2023
Efficient Aggregated Kernel Tests using Incomplete $U$-statistics · NeurIPS 2022
Machine learning › Learning theory › hypothesis testing › two-sample testing
kernel two-sample test
2.132025
DUAL: Learning Diverse Kernels for Aggregated Two-sample and Independence Testing · NeurIPS 2025
MMD Aggregated Two-Sample Test · J. Mach. Learn. Res. 2023
Efficient Aggregated Kernel Tests using Incomplete $U$-statistics · NeurIPS 2022
Machine learning › Learning theory › hypothesis testing
independence testing
1.422025
DUAL: Learning Diverse Kernels for Aggregated Two-sample and Independence Testing · NeurIPS 2025
Efficient Aggregated Kernel Tests using Incomplete $U$-statistics · NeurIPS 2022
Machine learning › Learning theory › hypothesis testing
two-sample testing
1.322023
MMD Aggregated Two-Sample Test · J. Mach. Learn. Res. 2023
MMD-Fuse: Learning and Combining Kernels for Two-Sample Testing Without Data Splitting · NeurIPS 2023
Machine learning › Learning theory › probability metric › integral probability metric
maximum mean discrepancy
1.222023
MMD-Fuse: Learning and Combining Kernels for Two-Sample Testing Without Data Splitting · NeurIPS 2023
Efficient Aggregated Kernel Tests using Incomplete $U$-statistics · NeurIPS 2022
Machine learning › Kernel, tree and ensemble methods
kernel selection
1.032023
MMD-Fuse: Learning and Combining Kernels for Two-Sample Testing Without Data Splitting · NeurIPS 2023
Efficient Aggregated Kernel Tests using Incomplete $U$-statistics · NeurIPS 2022
KSD Aggregated Goodness-of-fit Test · NeurIPS 2022
Machine learning › Kernel, tree and ensemble methods › kernel methods › kernel learning
multiple kernel learning
0.912025
DUAL: Learning Diverse Kernels for Aggregated Two-sample and Independence Testing · NeurIPS 2025
Machine learning › Probabilistic and Bayesian machine learning › experimental design
adaptive testing
0.712023
MMD Aggregated Two-Sample Test · J. Mach. Learn. Res. 2023
Machine learning › Learning theory
minimax optimality
0.712023
MMD Aggregated Two-Sample Test · J. Mach. Learn. Res. 2023
Machine learning › Learning theory › hypothesis testing
nonparametric hypothesis testing
0.712023
MMD Aggregated Two-Sample Test · J. Mach. Learn. Res. 2023
Machine learning › Probabilistic and Bayesian machine learning › statistical inference
goodness-of-fit testing
0.612022
KSD Aggregated Goodness-of-fit Test · NeurIPS 2022
Machine learning › Probabilistic and Bayesian machine learning › divergence measure
kernel stein discrepancy
0.612022
KSD Aggregated Goodness-of-fit Test · NeurIPS 2022
Machine learning › Learning theory › hypothesis testing
permutation test
0.212023
MMD-Fuse: Learning and Combining Kernels for Two-Sample Testing Without Data Splitting · NeurIPS 2023

Methods — techniques the papers use, named apart from their topics

wild bootstrap · 1.8submodular selection · 0.9kernel diversity · 0.9asymptotic analysis · 0.9permutation test · 0.7maximum mean discrepancy · 0.7exponential concentration bounds · 0.7deep kernel · 0.7autoencoder · 0.7parametric bootstrap · 0.6
YearPublicationVenuePosition
2025 Credal Two-Sample Tests of Epistemic Uncertainty
abstract
We introduce credal two-sample testing, a new hypothesis testing framework for comparing credal sets—convex sets of probability measures where each element captures aleatoric uncertainty and the set itself represents epistemic uncertainty that arises from the modeller’s partial ignorance. Compared to classical two-sample tests, which focus on comparing precise distributions, the proposed framework provides a broader and more versatile set of hypotheses. This approach enables the direct integration of epistemic uncertainty, effectively addressing the challenges arising from partial ignorance in hypothesis testing. By generalising two-sample test to compare credal sets, our framework enables reasoning for equality, inclusion, intersection, and mutual exclusivity, each offering unique insights into the modeller’s epistemic beliefs. As the first work on nonparametric hypothesis testing for comparing credal sets, we focus on finitely generated credal sets derived from i.i.d. samples from multiple distributions—referred to as \emph{credal samples}. We formalise these tests as two-sample tests with nuisance parameters and introduce the first permutation-based solution for this class of problems, significantly improving upon existing methods. Our approach properly incorporates the modeller’s epistemic uncertainty into hypothesis testing, leading to more robust and credible conclusions, with kernel-based implementations for real-world applications.
Siu Lun Chau, Antonin Schrab, Arthur Gretton, Dino Sejdinovic, Krikamol Muandet
AISTATS2
2025 Robust Kernel Hypothesis Testing under Data Corruption
abstract
We propose a general method for constructing robust permutation tests under data corruption. The proposed tests effectively control the non-asymptotic type I error under data corruption, and we prove their consistency in power under minimal conditions. This contributes to the practical deployment of hypothesis tests for real-world applications with potential adversarial attacks. For the two-sample and independence settings, we show that our kernel robust tests are minimax optimal, in the sense that they are guaranteed to be non-asymptotically powerful against alternatives uniformly separated from the null in the kernel MMD and HSIC metrics at some optimal rate (tight with matching lower bound). We point out that existing differentially private tests can be adapted to be robust to data corruption, and we demonstrate in experiments that our proposed tests achieve much higher power than these private tests. Finally, we provide publicly available implementations and empirically illustrate the practicality of our robust tests.
Antonin Schrab, Ilmun Kim
AISTATS1
2025 DUAL: Learning Diverse Kernels for Aggregated Two-sample and Independence Testing
abstract
To adapt kernel two-sample and independence testing to complex structured data, aggregation of multiple kernels is frequently employed to boost testing power compared to single-kernel tests. However, we observe a phenomenon that directly maximizing multiple kernel-based statistics may result in highly similar kernels that capture highly overlapping information, limiting the effectiveness of aggregation. To address this, we propose an aggregated statistic that explicitly incorporates kernel diversity based on the covariance between different kernels. Moreover, we identify a fundamental challenge: a trade-off between the diversity among kernels and the test power of individual kernels, i.e., the selected kernels should be both effective and diverse. This motivates a testing framework with selection inference, which leverages information from the training phase to select kernels with strong individual performance from the learned diverse kernel pool. We provide rigorous theoretical statements and proofs to show the consistency on the test power and control of Type-I error, along with asymptotic analysis of the proposed statistics. Lastly, we conducted extensive empirical experiments demonstrating the superior performance of our proposed approach across various benchmarks for both two-sample and independence testing.
Zhijian Zhou, Xunye Tian, Liuhua Peng, Antonin Schrab, Danica J. Sutherland, Feng Liu 0003
NeurIPS5
2023 MMD-Fuse: Learning and Combining Kernels for Two-Sample Testing Without Data Splitting
abstract
We propose novel statistics which maximise the power of a two-sample test based on the Maximum Mean Discrepancy (MMD), by adapting over the set of kernels used in defining it. For finite sets, this reduces to combining (normalised) MMD values under each of these kernels via a weighted soft maximum. Exponential concentration bounds are proved for our proposed statistics under the null and alternative. We further show how these kernels can be chosen in a data-dependent but permutation-independent way, in a well-calibrated test, avoiding data splitting. This technique applies more broadly to general permutation-based MMD testing, and includes the use of deep kernels with features learnt using unsupervised models such as auto-encoders. We highlight the applicability of our MMD-Fuse tests on both synthetic low-dimensional and real-world high-dimensional data, and compare its performance in terms of power against current state-of-the-art kernel tests.
Felix Biggs, Antonin Schrab, Arthur Gretton
NeurIPS2
2023 MMD Aggregated Two-Sample Test
abstract
We propose two novel nonparametric two-sample kernel tests based on the Maximum Mean Discrepancy (MMD). First, for a fixed kernel, we construct an MMD test using either permutations or a wild bootstrap, two popular numerical procedures to determine the test threshold. We prove that this test controls the probability of type I error non-asymptotically. Hence, it can be used reliably even in settings with small sample sizes as it remains well-calibrated, which differs from previous MMD tests which only guarantee correct test level asymptotically. When the difference in densities lies in a Sobolev ball, we prove minimax optimality of our MMD test with a specific kernel depending on the smoothness parameter of the Sobolev ball. In practice, this parameter is unknown and, hence, the optimal MMD test with this particular kernel cannot be used. To overcome this issue, we construct an aggregated test, called MMDAgg, which is adaptive to the smoothness parameter. The test power is maximised over the collection of kernels used, without requiring held-out data for kernel selection (which results in a loss of test power), or arbitrary kernel choices such as the median heuristic. We prove that MMDAgg still controls the level non-asymptotically, and achieves the minimax rate over Sobolev balls, up to an iterated logarithmic term. Our guarantees are not restricted to a specific type of kernel, but hold for any product of one-dimensional translation invariant characteristic kernels. We provide a user-friendly parameter-free implementation of MMDAgg using an adaptive collection of bandwidths. We demonstrate that MMDAgg significantly outperforms alternative state-of-the-art MMD-based two-sample tests on synthetic data satisfying the Sobolev smoothness assumption, and that, on real-world image data, MMDAgg closely matches the power of tests leveraging the use of models such as neural networks.
Antonin Schrab, Ilmun Kim, Mélisande Albert, Béatrice Laurent, Benjamin Guedj, Arthur Gretton
J. Mach. Learn. Res.1
2022 KSD Aggregated Goodness-of-fit Test
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 theoretical 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 score function for the model, 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.
Antonin Schrab, Benjamin Guedj, Arthur Gretton
NeurIPS1
2022 Efficient Aggregated Kernel Tests using Incomplete $U$-statistics
abstract
We propose a series of computationally efficient, nonparametric tests for the two-sample, 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 trade-off 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 more widespread permutation-based approaches, 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, our proposed linear-time tests outperform the current linear-time state-of-the-art tests (or at least match their test power).
Antonin Schrab, Ilmun Kim, Benjamin Guedj, Arthur Gretton
NeurIPS1