Saved in:
Bibliographic Details
Main Authors: Dubey, Harsh Vardhan, Lee, Ji Ah, Flaherty, Patrick
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.19131
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912095276105728
author Dubey, Harsh Vardhan
Lee, Ji Ah
Flaherty, Patrick
author_facet Dubey, Harsh Vardhan
Lee, Ji Ah
Flaherty, Patrick
contents Many Bayesian statistical inference problems come down to computing a maximum a-posteriori (MAP) assignment of latent variables. Yet, standard methods for estimating the MAP assignment do not have a finite time guarantee that the algorithm has converged to a fixed point. Previous research has found that MAP inference can be represented in dual form as a linear programming problem with a non-polynomial number of constraints. A Lagrangian relaxation of the dual yields a statistical inference algorithm as a linear programming problem. However, the decision as to which constraints to remove in the relaxation is often heuristic. We present a method for maximum a-posteriori inference in general Bayesian factor models that sequentially adds constraints to the fully relaxed dual problem using Benders' decomposition. Our method enables the incorporation of expressive integer and logical constraints in clustering problems such as must-link, cannot-link, and a minimum number of whole samples allocated to each cluster. Using this approach, we derive MAP estimation algorithms for the Bayesian Gaussian mixture model and latent Dirichlet allocation. Empirical results show that our method produces a higher optimal posterior value compared to Gibbs sampling and variational Bayes methods for standard data sets and provides certificate of convergence.
format Preprint
id arxiv_https___arxiv_org_abs_2410_19131
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Maximum a Posteriori Inference for Factor Graphs via Benders' Decomposition
Dubey, Harsh Vardhan
Lee, Ji Ah
Flaherty, Patrick
Machine Learning
Many Bayesian statistical inference problems come down to computing a maximum a-posteriori (MAP) assignment of latent variables. Yet, standard methods for estimating the MAP assignment do not have a finite time guarantee that the algorithm has converged to a fixed point. Previous research has found that MAP inference can be represented in dual form as a linear programming problem with a non-polynomial number of constraints. A Lagrangian relaxation of the dual yields a statistical inference algorithm as a linear programming problem. However, the decision as to which constraints to remove in the relaxation is often heuristic. We present a method for maximum a-posteriori inference in general Bayesian factor models that sequentially adds constraints to the fully relaxed dual problem using Benders' decomposition. Our method enables the incorporation of expressive integer and logical constraints in clustering problems such as must-link, cannot-link, and a minimum number of whole samples allocated to each cluster. Using this approach, we derive MAP estimation algorithms for the Bayesian Gaussian mixture model and latent Dirichlet allocation. Empirical results show that our method produces a higher optimal posterior value compared to Gibbs sampling and variational Bayes methods for standard data sets and provides certificate of convergence.
title Maximum a Posteriori Inference for Factor Graphs via Benders' Decomposition
topic Machine Learning
url https://arxiv.org/abs/2410.19131