On Faster Marginalization with Squared Circuits via Orthonormalization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Loconte, Lorenzo, Vergari, Antonio
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929681477926912
author Loconte, Lorenzo
Vergari, Antonio
author_facet Loconte, Lorenzo
Vergari, Antonio
contents Squared tensor networks (TNs) and their generalization as parameterized computational graphs -- squared circuits -- have been recently used as expressive distribution estimators in high dimensions. However, the squaring operation introduces additional complexity when marginalizing variables or computing the partition function, which hinders their usage in machine learning applications. Canonical forms of popular TNs are parameterized via unitary matrices as to simplify the computation of particular marginals, but cannot be mapped to general circuits since these might not correspond to a known TN. Inspired by TN canonical forms, we show how to parameterize squared circuits to ensure they encode already normalized distributions. We then use this parameterization to devise an algorithm to compute any marginal of squared circuits that is more efficient than a previously known one. We conclude by formally showing the proposed parameterization comes with no expressiveness loss for many circuit classes.
format Preprint
id arxiv_https___arxiv_org_abs_2412_07883
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle On Faster Marginalization with Squared Circuits via Orthonormalization
Loconte, Lorenzo
Vergari, Antonio
Machine Learning
Artificial Intelligence
Squared tensor networks (TNs) and their generalization as parameterized computational graphs -- squared circuits -- have been recently used as expressive distribution estimators in high dimensions. However, the squaring operation introduces additional complexity when marginalizing variables or computing the partition function, which hinders their usage in machine learning applications. Canonical forms of popular TNs are parameterized via unitary matrices as to simplify the computation of particular marginals, but cannot be mapped to general circuits since these might not correspond to a known TN. Inspired by TN canonical forms, we show how to parameterize squared circuits to ensure they encode already normalized distributions. We then use this parameterization to devise an algorithm to compute any marginal of squared circuits that is more efficient than a previously known one. We conclude by formally showing the proposed parameterization comes with no expressiveness loss for many circuit classes.
title On Faster Marginalization with Squared Circuits via Orthonormalization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2412.07883