HyperSPNs: Compact and Expressive Probabilistic Circuits

Andy Shih, Dorsa Sadigh, Stefano Ermon

Keywords: probabilistic circuits, sum product networks, discrete density estimation, generalization

Abstract: Probabilistic circuits (PCs) are a family of generative models which allows for the computation of exact likelihoods and marginals of its probability distributions. PCs are both expressive and tractable, and serve as popular choices for discrete density estimation tasks. However, large PCs are susceptible to overfitting, and only a few regularization strategies (e.g., dropout, weight-decay) have been explored. We propose HyperSPNs: a new paradigm of generating the mixture weights of large PCs using a small-scale neural network. Our framework can be viewed as a soft weight-sharing strategy, which combines the greater expressiveness of large models with the better generalization and memory-footprint properties of small models. We show the merits of our regularization strategy on two state-of-the-art PC families introduced in recent literature -- RAT-SPNs and EiNETs -- and demonstrate generalization improvements in both models on a suite of density estimation benchmarks in both discrete and continuous domains.

Code Of Conduct: I certify that all co-authors of this work have read and commit to adhering to the NeurIPS Statement on Ethics, Fairness, Inclusivity, and Code of Conduct.

TL;DR: We generate parameters of probabilistic circuits using an external neural network, leading to memory efficiencies and generalization improvements.

Code: https://github.com/AndyShih12/HyperSPN

Community Implementations: [ 2 code implementations](https://www.catalyzex.com/paper/arxiv:2112.00914/code)