A Witness Two-Sample Test
This work addresses the two-sample problem for statistical testing, offering an incremental improvement in data efficiency over the state-of-the-art MMD test.
The paper tackles the two-sample problem by proposing a new nonparametric test that uses training data more efficiently to define weights and basis points, showing it is consistent, controls type-I error, and achieves comparable or better test power than existing methods like MMD on synthetic and real datasets such as Higgs data.
The Maximum Mean Discrepancy (MMD) has been the state-of-the-art nonparametric test for tackling the two-sample problem. Its statistic is given by the difference in expectations of the witness function, a real-valued function defined as a weighted sum of kernel evaluations on a set of basis points. Typically the kernel is optimized on a training set, and hypothesis testing is performed on a separate test set to avoid overfitting (i.e., control type-I error). That is, the test set is used to simultaneously estimate the expectations and define the basis points, while the training set only serves to select the kernel and is discarded. In this work, we propose to use the training data to also define the weights and the basis points for better data efficiency. We show that 1) the new test is consistent and has a well-controlled type-I error; 2) the optimal witness function is given by a precision-weighted mean in the reproducing kernel Hilbert space associated with the kernel; and 3) the test power of the proposed test is comparable or exceeds that of the MMD and other modern tests, as verified empirically on challenging synthetic and real problems (e.g., Higgs data).