Toward Scalable and Valid Conditional Independence Testing with Spectral Representations

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Frohlich, Alek, Kostic, Vladimir, Lounici, Karim, Perazzo, Daniel, Pontil, Massimiliano
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911332705501184
author Frohlich, Alek
Kostic, Vladimir
Lounici, Karim
Perazzo, Daniel
Pontil, Massimiliano
author_facet Frohlich, Alek
Kostic, Vladimir
Lounici, Karim
Perazzo, Daniel
Pontil, Massimiliano
contents Conditional independence (CI) is central to causal inference, feature selection, and graphical modeling, yet it is untestable in many settings without additional assumptions. Existing CI tests often rely on restrictive structural conditions, limiting their validity on real-world data. Kernel methods using the partial covariance operator offer a more principled approach but suffer from limited adaptivity, slow convergence, and poor scalability. In this work, we explore whether representation learning can help address these limitations. Specifically, we focus on representations derived from the singular value decomposition of the partial covariance operator and use them to construct a simple test statistic, reminiscent of the Hilbert-Schmidt Independence Criterion (HSIC). We also introduce a practical bi-level contrastive algorithm to learn these representations. Our theory links representation learning error to test performance and establishes asymptotic validity and power guarantees. Preliminary experiments suggest that this approach offers a practical and statistically grounded path toward scalable CI testing, bridging kernel-based theory with modern representation learning.
format Preprint
id arxiv_https___arxiv_org_abs_2512_19510
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Toward Scalable and Valid Conditional Independence Testing with Spectral Representations
Frohlich, Alek
Kostic, Vladimir
Lounici, Karim
Perazzo, Daniel
Pontil, Massimiliano
Machine Learning
Conditional independence (CI) is central to causal inference, feature selection, and graphical modeling, yet it is untestable in many settings without additional assumptions. Existing CI tests often rely on restrictive structural conditions, limiting their validity on real-world data. Kernel methods using the partial covariance operator offer a more principled approach but suffer from limited adaptivity, slow convergence, and poor scalability. In this work, we explore whether representation learning can help address these limitations. Specifically, we focus on representations derived from the singular value decomposition of the partial covariance operator and use them to construct a simple test statistic, reminiscent of the Hilbert-Schmidt Independence Criterion (HSIC). We also introduce a practical bi-level contrastive algorithm to learn these representations. Our theory links representation learning error to test performance and establishes asymptotic validity and power guarantees. Preliminary experiments suggest that this approach offers a practical and statistically grounded path toward scalable CI testing, bridging kernel-based theory with modern representation learning.
title Toward Scalable and Valid Conditional Independence Testing with Spectral Representations
topic Machine Learning
url https://arxiv.org/abs/2512.19510