Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Approximate inference of marginals using the IBIA framework

About

Exact inference of marginals in probabilistic graphical models (PGM) is known to be intractable, necessitating the use of approximate methods. Most of the existing variational techniques perform iterative message passing in loopy graphs which is slow to converge for many benchmarks. In this paper, we propose a new algorithm for marginal inference that is based on the incremental build-infer-approximate (IBIA) paradigm. Our algorithm converts the PGM into a sequence of linked clique tree forests (SLCTF) with bounded clique sizes, and then uses a heuristic belief update algorithm to infer the marginals. For the special case of Bayesian networks, we show that if the incremental build step in IBIA uses the topological order of variables then (a) the prior marginals are consistent in all CTFs in the SLCTF and (b) the posterior marginals are consistent once all evidence variables are added to the SLCTF. In our approach, the belief propagation step is non-iterative and the accuracy-complexity trade-off is controlled using user-defined clique size bounds. Results for several benchmark sets from recent UAI competitions show that our method gives either better or comparable accuracy than existing variational and sampling based methods, with smaller runtimes.

Shivani Bathla, Vinita Vasudevan• 2023

Related benchmarks

TaskDatasetResultRank
Marginal InferenceBN
HD Avg7.00e-6
16
Marginal InferenceGridBN
HD Avg2.00e-5
10
Marginal InferencePedigree
HD avg0.008
10
Marginal InferencePromedas
HD Avg0.01
10
Marginal Inferencesegment
HD_avg1.00e-7
10
Marginal InferencePROTEIN
HD_avg3.00e-5
10
Marginal InferenceObjDetect
HD Avg4.00e-4
10
Marginal InferenceGrids
HD Average0.002
9
Marginal InferenceCSP
HD Avg2.00e-4
9
Marginal InferenceDBN
HD Avg0.003
8
Showing 10 of 23 rows

Other info

Follow for update