English

Regularized $f$-Divergence Kernel Tests

Machine Learning 2026-01-28 v1 Machine Learning

Abstract

We propose a framework to construct practical kernel-based two-sample tests from the family of ff-divergences. The test statistic is computed from the witness function of a regularized variational representation of the divergence, which we estimate using kernel methods. The proposed test is adaptive over hyperparameters such as the kernel bandwidth and the regularization parameter. We provide theoretical guarantees for statistical test power across our family of ff-divergence estimates. While our test covers a variety of ff-divergences, we bring particular focus to the Hockey-Stick divergence, motivated by its applications to differential privacy auditing and machine unlearning evaluation. For two-sample testing, experiments demonstrate that different ff-divergences are sensitive to different localized differences, illustrating the importance of leveraging diverse statistics. For machine unlearning, we propose a relative test that distinguishes true unlearning failures from safe distributional variations.

Keywords

Cite

@article{arxiv.2601.19755,
  title  = {Regularized $f$-Divergence Kernel Tests},
  author = {Mónica Ribero and Antonin Schrab and Arthur Gretton},
  journal= {arXiv preprint arXiv:2601.19755},
  year   = {2026}
}