English

Exact Fractional Inference via Re-Parametrization & Interpolation between Tree-Re-Weighted- and Belief Propagation- Algorithms

Machine Learning 2024-11-14 v4 Statistical Mechanics

Abstract

Computing the partition function, ZZ, of an Ising model over a graph of NN \enquote{spins} is most likely exponential in NN. Efficient variational methods, such as Belief Propagation (BP) and Tree Re-Weighted (TRW) algorithms, compute ZZ approximately by minimizing the respective (BP- or TRW-) free energy. We generalize the variational scheme by building a λ\lambda-fractional interpolation, Z(λ)Z^{(\lambda)}, where λ=0\lambda=0 and λ=1\lambda=1 correspond to TRW- and BP-approximations, respectively. This fractional scheme -- coined Fractional Belief Propagation (FBP) -- guarantees that in the attractive (ferromagnetic) case Z(TRW)Z(λ)Z(BP)Z^{(TRW)} \geq Z^{(\lambda)} \geq Z^{(BP)}, and there exists a unique (\enquote{exact}) λ\lambda_* such that Z=Z(λ)Z=Z^{(\lambda_*)}. Generalizing the re-parametrization approach of \citep{wainwright_tree-based_2002} and the loop series approach of \citep{chertkov_loop_2006}, we show how to express ZZ as a product, λ: Z=Z(λ)Z~(λ)\forall \lambda:\ Z=Z^{(\lambda)}{\tilde Z}^{(\lambda)}, where the multiplicative correction, Z~(λ){\tilde Z}^{(\lambda)}, is an expectation over a node-independent probability distribution built from node-wise fractional marginals. Our theoretical analysis is complemented by extensive experiments with models from Ising ensembles over planar and random graphs of medium and large sizes. Our empirical study yields a number of interesting observations, such as the ability to estimate Z~(λ){\tilde Z}^{(\lambda)} with O(N2::4)O(N^{2::4}) fractional samples and suppression of variation in λ\lambda_* estimates with an increase in NN for instances from a particular random Ising ensemble, where [2::4][2::4] indicates a range from 22 to 44. We also discuss the applicability of this approach to the problem of image de-noising.

Keywords

Cite

@article{arxiv.2301.10369,
  title  = {Exact Fractional Inference via Re-Parametrization & Interpolation between Tree-Re-Weighted- and Belief Propagation- Algorithms},
  author = {Hamidreza Behjoo and Michael Chertkov},
  journal= {arXiv preprint arXiv:2301.10369},
  year   = {2024}
}