arXiv · 2109.06587
Sum-Product-Attention Networks: Leveraging Self-Attention in Probabilistic Circuits
Abstract
Probabilistic circuits (PCs) have become the de-facto standard for learning and inference in probabilistic modeling. We introduce Sum-Product-Attention Networks (SPAN), a new generative model that integrates probabilistic circuits with Transformers. SPAN uses self-attention to select the most relevant parts of a probabilistic circuit, here sum-product networks, to improve the modeling capability of the underlying sum-product network. We show that while modeling, SPAN focuses on a specific set of independent assumptions in every product layer of the sum-product network. Our empirical evaluations show that SPAN outperforms state-of-the-art probabilistic generative models on various benchmark data sets as well is an efficient generative image model.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Zhongjie Yu, Devendra Singh Dhami, Kristian Kersting. 2021-09-14. Sum-Product-Attention Networks: Leveraging Self-Attention in Probabilistic Circuits. https://arxiv.org/abs/2109.06587
Cite the original work for its findings. Save a collection to share your selection of sources.