arXiv Statistics ML

Learning discrete Bayesian networks with hierarchical Dirichlet shrinkage

arXiv Machine Learning
4d ago

Joint Bayesian Inference of Graphical Structure and Parameters with a Single Generative Flow Network

The paper introduces JSP-GFN, a Generative Flow Network that jointly infers the structure and parameters of a Bayesian Network. It sequentially generates a directed acyclic graph edge by edge and then samples the corresponding conditional probability parameters once the full structure is known. Experiments on simulated and real data show that JSP‑GFN accurately approximates the joint posterior and outperforms existing methods.

By Tristan Deleu, Mizu Nishikawa-Toomey, Jithendaraa Subramanian, Esmeralda S. Whitammer, Laurent Charlin, Yoshua Bengio
arXiv AI
Aug 25

Joint Causal Structure and Cluster Discovery Using Variational Inference

The paper introduces a variational inference framework that jointly discovers latent clusters of variables and the causal relationships among those clusters. It models clusters with categorical distributions and graph structures with Bernoulli distributions, deriving variational lower bounds and estimation techniques for learning both cluster assignments and causal links. The method’s effectiveness is shown on synthetic and real datasets.

By Avni Rajpal, Anubhav Kumar, Rishabh Karnad, Mohammad Emtiyaz Khan, P. K. Srijith