Tractable Representation Learning with Probabilistic Circuits

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Braun, Steven, Sidheekh, Sahil, Vergari, Antonio, Mundt, Martin, Natarajan, Sriraam, Kersting, Kristian
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866918104579178496
author Braun, Steven
Sidheekh, Sahil
Vergari, Antonio
Mundt, Martin
Natarajan, Sriraam
Kersting, Kristian
author_facet Braun, Steven
Sidheekh, Sahil
Vergari, Antonio
Mundt, Martin
Natarajan, Sriraam
Kersting, Kristian
contents Probabilistic circuits (PCs) are powerful probabilistic models that enable exact and tractable inference, making them highly suitable for probabilistic reasoning and inference tasks. While dominant in neural networks, representation learning with PCs remains underexplored, with prior approaches relying on external neural embeddings or activation-based encodings. To address this gap, we introduce autoencoding probabilistic circuits (APCs), a novel framework leveraging the tractability of PCs to model probabilistic embeddings explicitly. APCs extend PCs by jointly modeling data and embeddings, obtaining embedding representations through tractable probabilistic inference. The PC encoder allows the framework to natively handle arbitrary missing data and is seamlessly integrated with a neural decoder in a hybrid, end-to-end trainable architecture enabled by differentiable sampling. Our empirical evaluation demonstrates that APCs outperform existing PC-based autoencoding methods in reconstruction quality, generate embeddings competitive with, and exhibit superior robustness in handling missing data compared to neural autoencoders. These results highlight APCs as a powerful and flexible representation learning method that exploits the probabilistic inference capabilities of PCs, showing promising directions for robust inference, out-of-distribution detection, and knowledge distillation.
format Preprint
id arxiv_https___arxiv_org_abs_2507_04385
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Tractable Representation Learning with Probabilistic Circuits
Braun, Steven
Sidheekh, Sahil
Vergari, Antonio
Mundt, Martin
Natarajan, Sriraam
Kersting, Kristian
Machine Learning
Artificial Intelligence
Probabilistic circuits (PCs) are powerful probabilistic models that enable exact and tractable inference, making them highly suitable for probabilistic reasoning and inference tasks. While dominant in neural networks, representation learning with PCs remains underexplored, with prior approaches relying on external neural embeddings or activation-based encodings. To address this gap, we introduce autoencoding probabilistic circuits (APCs), a novel framework leveraging the tractability of PCs to model probabilistic embeddings explicitly. APCs extend PCs by jointly modeling data and embeddings, obtaining embedding representations through tractable probabilistic inference. The PC encoder allows the framework to natively handle arbitrary missing data and is seamlessly integrated with a neural decoder in a hybrid, end-to-end trainable architecture enabled by differentiable sampling. Our empirical evaluation demonstrates that APCs outperform existing PC-based autoencoding methods in reconstruction quality, generate embeddings competitive with, and exhibit superior robustness in handling missing data compared to neural autoencoders. These results highlight APCs as a powerful and flexible representation learning method that exploits the probabilistic inference capabilities of PCs, showing promising directions for robust inference, out-of-distribution detection, and knowledge distillation.
title Tractable Representation Learning with Probabilistic Circuits
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2507.04385