The Generalised Kernel Covariance Measure
Organizations: Leibniz Institute for Prevention Research and Epidemiology – BIPS, Bremen, Germany · Faculty of Mathematics and Computer Science, University of Bremen, Bremen, Germany · School of Mathematical Sciences and Australian Institute for Machine Learning Adelaide University, Adelaide, Australia
Abstract
We consider the problem of conditional independence (CI) testing and adopt a kernel-based approach. Kernel-based CI tests embed variables in reproducing kernel Hilbert spaces, regress their embeddings on the conditioning variables, and test the resulting residuals for marginal independence. This approach yields tests that are sensitive to a broad range of conditional dependencies. Existing methods, however, rely heavily on kernel ridge regression, which is computationally expensive when properly tuned and yields poorly calibrated tests when left untuned, which limits their practical usefulness. We propose the Generalised Kernel Covariance Measure (GKCM), a regression-model-agnostic kernel-based CI test that accommodates a broad class of regression estimators. Building on the Generalised Hilbertian Covariance Measure framework (Lundborg et al., 2022), we characterise conditions under which GKCM satisfies uniform asymptotic level guarantees. In simulations, GKCM paired with tree-based regression models frequently outperforms state-of-the-art CI tests across a diverse range of data-generating processes, achieving better type I error control and competitive or superior power.
Figures & tables
| GCM | wGCM | PCM | KCIT | KRESIT | RCIT | RCoT | GKCM | |
| Regression methods | Any | Any | Any | KRR | KRR | KRR | KRR | Any |
| Type-I error control | Uniform | Uniform | Uniform | Pointw. | Pointw. | Pointw. | Pointw. | Uniform |
| Function spaces | Linear | Linear | Mixed | RKHS | RKHS | RKHS | RKHS | RKHS |
| Conditional covariances | Mean-z. | A.s.-zero | A.s.-zero | A.s.-zero | Mean-z. | A.s.-zero | Mean-z. | Mean-z. |
| Approximated using Random Fourier Features | ||||||||
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
Conditional Independence Testing
Conditional independence testing asks whether two variables remain independent after accounting for other variables, supporting causal discovery and valid inference.
See all 19 papers