Meta-probabilistic Modeling

Kevin Zhang, Yixin Wang

TL;DR: Probabilistic modeling for multiple datasets simultaneously

Abstract: Probabilistic graphical models (PGMs) are widely used to discover latent structure in data, but their success hinges on selecting an appropriate model design. In practice, model specification is difficult and often requires iterative trial-and-error. This challenge arises because classical PGMs typically operate on individual datasets. In this work, we consider settings involving collections of related datasets and propose meta-probabilistic modeling (MPM) to learn the generative model structure itself. MPM uses a hierarchical formulation in which global components encode shared patterns across datasets, while local parameters capture dataset-specific latent structure. For scalable learning and inference, we derive a tractable VAE-inspired surrogate objective together with a bi-level optimization algorithm. Our methodology supports a broad class of expressive probabilistic models and has connections to existing architectures, such as Slot Attention. Experiments on object-centric representation learning and sequential text modeling demonstrate that MPM effectively adapts generative models to data while recovering meaningful latent representations.

Code Dataset Promise: Yes

Code Dataset Url: https://github.com/kzhangm02/mpm

Format Confirmation: I agree that I have read and followed the formatting instructions for the camera ready version.

Submission Number: 2067