Markov Chains Approximate Message Passing
Amit Rajaraman, David X. Wu
Abstract
Markov chain Monte Carlo algorithms have long been observed to obtain near-optimal performance in various Bayesian inference settings. However, developing a supporting theory that makes these studies rigorous has proved challenging. In this paper, we study the classical spiked Wigner inference problem, where one aims to recover a planted Boolean spike from a noisy matrix measurement. We relate the recovery performance of Glauber dynamics on the annealed posterior to the performance of Approximate Message Passing (AMP), which is known to achieve Bayes-optimal performance. Our main results rely on the analysis of an auxiliary Markov chain called restricted Gaussian dynamics (RGD). Concretely, we establish the following three results. First, RGD can be reduced to an effective one-dimensional recursion which mirrors the evolution of the AMP iterates. Second, from a warm start, RGD rapidly converges to a fixed point in correlation space, which recovers Bayes-optimal performance when run on the posterior. Third, conditioned on widely believed mixing results for the SK model, we recover the phase transition for non-trivial inference. The full version of this paper can be found on arXiv (arXiv ID: 2512.02384).
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on7
- Sampling from the Sherrington-Kirkpatrick Gibbs measure via algorithmic stochastic localizationAhmed El Alaoui, Andrea Montanari, Mark SellkeFOCS 2022 · 29 citations
- Taming Imperfect Process Verifiers: A Sampling Perspective on BacktrackingDhruv Rohatgi, Abhishek Shetty, Donya Saless, Yuchen Li et al.ICLR 2026 · 15 citations
- Fast Mixing in Sparse Random Ising ModelsKuikui Liu, Sidhanth Mohanty, Amit Rajaraman, David X. WuFOCS 2024 · 14 citations
- Weak Poincaré Inequalities, Simulated Annealing, and Sampling from Spherical Spin GlassesBrice Huang, Sidhanth Mohanty, Amit Rajaraman, David X. WuSTOC 2025 · 13 citations
- Trickle-Down in Localization Schemes and ApplicationsNima Anari, Frederic Koehler, Thuy-Duong VuongSTOC 2024 · 2 citations
Related papers
- Optimal Algorithms for the Inhomogeneous Spiked Wigner ModelAleksandr Pak, Justin Ko, Florent KrzakalaNeurIPS 2023 · 17 citations
- The price of ignorance: how much does it cost to forget noise structure in low-rank matrix estimation?Jean Barbier, TianQi Hou, Marco Mondelli, Manuel SáenzNeurIPS 2022 · 25 citations
- PCA Initialization for Approximate Message Passing in Rotationally Invariant ModelsMarco Mondelli, Ramji VenkataramananNeurIPS 2021 · 23 citations
- All-or-nothing statistical and computational phase transitions in sparse spiked matrix estimationJean Barbier, Nicolas Macris, Cynthia RushNeurIPS 2020 · 42 citations
- Subspace clustering in high-dimensions: Phase transitions & Statistical-to-Computational gapLuca Pesce, Bruno Loureiro, Florent Krzakala, Lenka ZdeborováNeurIPS 2022 · 4 citations
