How to Square Tensor Networks and Circuits Without Squaring Them 文章

ArXiv CS.AI2026-05-27NEWSen作者: Lorenzo Loconte, Adri\'an Javaloy, Antonio Vergari

摘要

arXiv:2512.17090v2 Announce Type: replace-cross Abstract: Squared tensor networks (TNs) and their extension as computational graphs--squared circuits--have been used as expressive distribution estimators, yet supporting closed-form marginalization. However, the squaring operation introduces additional complexity when computing the partition function or marginalizing variables, which hinders their applicability in ML. To solve this issue, canonical forms of TNs are parameterized via unitary matrices to simplify the computation of marginals. However, these canonical forms do not apply to circuits, as they can represent factorizations that do not directly map to a known TN. Inspired by the ideas of orthogonality in canonical forms and determinism in circuits enabling tractable maximization, we show how to parameterize squared circuits to overcome their marginalization overhead.