PGMax: Factor Graphs for Discrete Probabilistic Graphical Models and Loopy Belief Propagation in JAX.
Guangyao ZhouNishanth KumarMiguel Lázaro-GredillaShrinu KushagraDileep GeorgePublished in: CoRR (2022)
Keyphrases
- probabilistic graphical models
- factor graphs
- approximate inference
- graphical models
- belief propagation
- exact inference
- message passing
- markov networks
- probabilistic inference
- bayesian networks
- probabilistic model
- parameter estimation
- latent variables
- markov random field
- conditional random fields
- variational methods
- random variables
- gaussian process
- stereo matching
- parameter learning
- structure learning
- dynamic bayesian networks
- graph cuts
- belief networks
- conditional independence
- structured prediction
- model selection
- bayesian inference
- machine learning
- computer vision