Tensor Network Moral Graph Recovery of Discrete Probability Distributions

arXiv:2609.09258 · stat.ML, cs.IT, cs.LG, math.IT, math.PR, quant-ph · Submitted 2026-09-08 · Read on arXiv

stat.ML, cs.IT, cs.LG, math.IT, math.PR, quant-ph

Submitted: 2026-09-08

Updated: 2026-09-08

Comments: 24 pages

License: http://creativecommons.org/licenses/by/4.0/

The gist: We present a method for recovering the moral graph of a causal DAG from a probability distribution over discrete variables, using fully connected tensor networks (FCTNs) with nuclear-norm-regularized

Terminology

Abstract

We present a method for recovering the moral graph of a causal DAG from a probability distribution over discrete variables, using fully connected tensor networks (FCTNs) with nuclear-norm-regularized bond corrections. Each bond matrix is parameterized as a baseline all-ones matrix plus a low-rank correction C ij = U ijV ij, and the nuclear norm of the correction implemented via the variational Frobenius norm penalty on the factors drives unnecessary bonds to zero. We prove that under faithfulness, positivity, and a no-implicit-rerouting assumption on the local tensor architecture, every optimal FCTN with zero reconstruction error epsilon = 0 has effective graph exactly equal to the moral graph. For the approximate regime (epsilon > 0), we provide explicit recovery bounds using the Fannes-Audenaert continuity of conditional mutual information, and derive a sufficient condition on the regularization parameter β. The effective graph is read directly from the optimized bond matrices.

Related papers