Scaling Continuous Latent Variable Models as Probabilistic Integral Circuits

التفاصيل البيبلوغرافية
العنوان: Scaling Continuous Latent Variable Models as Probabilistic Integral Circuits
المؤلفون: Gala, Gennaro, de Campos, Cassio, Vergari, Antonio, Quaeghebeur, Erik
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Machine Learning, Computer Science - Artificial Intelligence
الوصف: Probabilistic integral circuits (PICs) have been recently introduced as probabilistic models enjoying the key ingredient behind expressive generative models: continuous latent variables (LVs). PICs are symbolic computational graphs defining continuous LV models as hierarchies of functions that are summed and multiplied together, or integrated over some LVs. They are tractable if LVs can be analytically integrated out, otherwise they can be approximated by tractable probabilistic circuits (PC) encoding a hierarchical numerical quadrature process, called QPCs. So far, only tree-shaped PICs have been explored, and training them via numerical quadrature requires memory-intensive processing at scale. In this paper, we address these issues, and present: (i) a pipeline for building DAG-shaped PICs out of arbitrary variable decompositions, (ii) a procedure for training PICs using tensorized circuit architectures, and (iii) neural functional sharing techniques to allow scalable training. In extensive experiments, we showcase the effectiveness of functional sharing and the superiority of QPCs over traditional PCs.
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2406.06494
رقم الأكسشن: edsarx.2406.06494
قاعدة البيانات: arXiv