variational inference spin glasses and tap free energy
play

Variational inference, spin glasses, and TAP free energy Song Mei - PowerPoint PPT Presentation

Variational inference, spin glasses, and TAP free energy Song Mei Stanford University September 19, 2018 Joint work with Zhou Fan and Andrea Montanari Song Mei (Stanford University) TAP free energy September 19, 2018 1 / 29 General


  1. Variational inference, spin glasses, and TAP free energy Song Mei Stanford University September 19, 2018 Joint work with Zhou Fan and Andrea Montanari Song Mei (Stanford University) TAP free energy September 19, 2018 1 / 29

  2. General motivation ◮ Bayesian inference: high dimensional integration is hard! ◮ Variational inference: integration/summation ✦ optimization. A popular objective function: “mean field free energy”. ◮ Applications: topic modeling, stochastic block model, low rank matrix estimation, compressed sensing.... ... within which “MF free energy” is known to be not optimal. ◮ Today: introduce the optimal objective “TAP free energy”, and provide rigorous results. Song Mei (Stanford University) TAP free energy September 19, 2018 2 / 29

  3. General motivation ◮ Bayesian inference: high dimensional integration is hard! ◮ Variational inference: integration/summation ✦ optimization. A popular objective function: “mean field free energy”. ◮ Applications: topic modeling, stochastic block model, low rank matrix estimation, compressed sensing.... ... within which “MF free energy” is known to be not optimal. ◮ Today: introduce the optimal objective “TAP free energy”, and provide rigorous results. Song Mei (Stanford University) TAP free energy September 19, 2018 2 / 29

  4. General motivation ◮ Bayesian inference: high dimensional integration is hard! ◮ Variational inference: integration/summation ✦ optimization. A popular objective function: “mean field free energy”. ◮ Applications: topic modeling, stochastic block model, low rank matrix estimation, compressed sensing.... ... within which “MF free energy” is known to be not optimal. ◮ Today: introduce the optimal objective “TAP free energy”, and provide rigorous results. Song Mei (Stanford University) TAP free energy September 19, 2018 2 / 29

  5. General motivation ◮ Bayesian inference: high dimensional integration is hard! ◮ Variational inference: integration/summation ✦ optimization. A popular objective function: “mean field free energy”. ◮ Applications: topic modeling, stochastic block model, low rank matrix estimation, compressed sensing.... ... within which “MF free energy” is known to be not optimal. ◮ Today: introduce the optimal objective “TAP free energy”, and provide rigorous results. Song Mei (Stanford University) TAP free energy September 19, 2018 2 / 29

  6. General motivation ◮ Bayesian inference: high dimensional integration is hard! ◮ Variational inference: integration/summation ✦ optimization. A popular objective function: “mean field free energy”. ◮ Applications: topic modeling, stochastic block model, low rank matrix estimation, compressed sensing.... ... within which “MF free energy” is known to be not optimal. ◮ Today: introduce the optimal objective “TAP free energy”, and provide rigorous results. Song Mei (Stanford University) TAP free energy September 19, 2018 2 / 29

  7. General motivation ◮ Bayesian inference: high dimensional integration is hard! ◮ Variational inference: integration/summation ✦ optimization. A popular objective function: “mean field free energy”. ◮ Applications: topic modeling, stochastic block model, low rank matrix estimation, compressed sensing.... ... within which “MF free energy” is known to be not optimal. ◮ Today: introduce the optimal objective “TAP free energy”, and provide rigorous results. Song Mei (Stanford University) TAP free energy September 19, 2018 2 / 29

  8. Z ✷ synchronization ◮ Signal: x ❂ ❬ ① ✶ ❀ ✿ ✿ ✿ ❀ ① ♥ ❪ T ✷ Z ♥ ✐✿✐✿❞✿ ✷ ❀ ① ✐ ✘ ❯♥✐❢✭ Z ✷ ✮ ❀ Z ✷ ❂ ❢ ✰✶ ❀ � ✶ ❣ ✿ ◮ Observation: for ✶ ✔ ✐ ❁ ❥ ✔ ♥ ❨ ✐❥ ❂ ✕ ♥① ✐ ① ❥ ✰ ❲ ✐❥ ✿ ◮ Noise ❲ ✐❥ ✘ ◆ ✭✵ ❀ ✶ ❂♥ ✮ . ◮ SNR ✕ ✷ ❬✵ ❀ ✶ ✮ fixed, dimension ♥ ✦ ✶ . ◮ In matrix notation: Y ❂ ✕ ♥ xx T ✰ W ✿ ◮ Task: given Y ❂ ✭ ❨ ✐❥ ✮ , estimate x (or say X ❂ xx T ). Song Mei (Stanford University) TAP free energy September 19, 2018 3 / 29

  9. Z ✷ synchronization ◮ Signal: x ❂ ❬ ① ✶ ❀ ✿ ✿ ✿ ❀ ① ♥ ❪ T ✷ Z ♥ ✐✿✐✿❞✿ ✷ ❀ ① ✐ ✘ ❯♥✐❢✭ Z ✷ ✮ ❀ Z ✷ ❂ ❢ ✰✶ ❀ � ✶ ❣ ✿ ◮ Observation: for ✶ ✔ ✐ ❁ ❥ ✔ ♥ ❨ ✐❥ ❂ ✕ ♥① ✐ ① ❥ ✰ ❲ ✐❥ ✿ ◮ Noise ❲ ✐❥ ✘ ◆ ✭✵ ❀ ✶ ❂♥ ✮ . ◮ SNR ✕ ✷ ❬✵ ❀ ✶ ✮ fixed, dimension ♥ ✦ ✶ . ◮ In matrix notation: Y ❂ ✕ ♥ xx T ✰ W ✿ ◮ Task: given Y ❂ ✭ ❨ ✐❥ ✮ , estimate x (or say X ❂ xx T ). Song Mei (Stanford University) TAP free energy September 19, 2018 3 / 29

  10. Z ✷ synchronization ◮ Signal: x ❂ ❬ ① ✶ ❀ ✿ ✿ ✿ ❀ ① ♥ ❪ T ✷ Z ♥ ✐✿✐✿❞✿ ✷ ❀ ① ✐ ✘ ❯♥✐❢✭ Z ✷ ✮ ❀ Z ✷ ❂ ❢ ✰✶ ❀ � ✶ ❣ ✿ ◮ Observation: for ✶ ✔ ✐ ❁ ❥ ✔ ♥ ❨ ✐❥ ❂ ✕ ♥① ✐ ① ❥ ✰ ❲ ✐❥ ✿ ◮ Noise ❲ ✐❥ ✘ ◆ ✭✵ ❀ ✶ ❂♥ ✮ . ◮ SNR ✕ ✷ ❬✵ ❀ ✶ ✮ fixed, dimension ♥ ✦ ✶ . ◮ In matrix notation: Y ❂ ✕ ♥ xx T ✰ W ✿ ◮ Task: given Y ❂ ✭ ❨ ✐❥ ✮ , estimate x (or say X ❂ xx T ). Song Mei (Stanford University) TAP free energy September 19, 2018 3 / 29

  11. Z ✷ synchronization ◮ Signal: x ❂ ❬ ① ✶ ❀ ✿ ✿ ✿ ❀ ① ♥ ❪ T ✷ Z ♥ ✐✿✐✿❞✿ ✷ ❀ ① ✐ ✘ ❯♥✐❢✭ Z ✷ ✮ ❀ Z ✷ ❂ ❢ ✰✶ ❀ � ✶ ❣ ✿ ◮ Observation: for ✶ ✔ ✐ ❁ ❥ ✔ ♥ ❨ ✐❥ ❂ ✕ ♥① ✐ ① ❥ ✰ ❲ ✐❥ ✿ ◮ Noise ❲ ✐❥ ✘ ◆ ✭✵ ❀ ✶ ❂♥ ✮ . ◮ SNR ✕ ✷ ❬✵ ❀ ✶ ✮ fixed, dimension ♥ ✦ ✶ . ◮ In matrix notation: Y ❂ ✕ ♥ xx T ✰ W ✿ ◮ Task: given Y ❂ ✭ ❨ ✐❥ ✮ , estimate x (or say X ❂ xx T ). Song Mei (Stanford University) TAP free energy September 19, 2018 3 / 29

  12. Z ✷ synchronization ◮ Signal: x ❂ ❬ ① ✶ ❀ ✿ ✿ ✿ ❀ ① ♥ ❪ T ✷ Z ♥ ✐✿✐✿❞✿ ✷ ❀ ① ✐ ✘ ❯♥✐❢✭ Z ✷ ✮ ❀ Z ✷ ❂ ❢ ✰✶ ❀ � ✶ ❣ ✿ ◮ Observation: for ✶ ✔ ✐ ❁ ❥ ✔ ♥ ❨ ✐❥ ❂ ✕ ♥① ✐ ① ❥ ✰ ❲ ✐❥ ✿ ◮ Noise ❲ ✐❥ ✘ ◆ ✭✵ ❀ ✶ ❂♥ ✮ . ◮ SNR ✕ ✷ ❬✵ ❀ ✶ ✮ fixed, dimension ♥ ✦ ✶ . ◮ In matrix notation: Y ❂ ✕ ♥ xx T ✰ W ✿ ◮ Task: given Y ❂ ✭ ❨ ✐❥ ✮ , estimate x (or say X ❂ xx T ). Song Mei (Stanford University) TAP free energy September 19, 2018 3 / 29

  13. Z ✷ synchronization ◮ Signal: x ❂ ❬ ① ✶ ❀ ✿ ✿ ✿ ❀ ① ♥ ❪ T ✷ Z ♥ ✐✿✐✿❞✿ ✷ ❀ ① ✐ ✘ ❯♥✐❢✭ Z ✷ ✮ ❀ Z ✷ ❂ ❢ ✰✶ ❀ � ✶ ❣ ✿ ◮ Observation: for ✶ ✔ ✐ ❁ ❥ ✔ ♥ ❨ ✐❥ ❂ ✕ ♥① ✐ ① ❥ ✰ ❲ ✐❥ ✿ ◮ Noise ❲ ✐❥ ✘ ◆ ✭✵ ❀ ✶ ❂♥ ✮ . ◮ SNR ✕ ✷ ❬✵ ❀ ✶ ✮ fixed, dimension ♥ ✦ ✶ . ◮ In matrix notation: Y ❂ ✕ ♥ xx T ✰ W ✿ ◮ Task: given Y ❂ ✭ ❨ ✐❥ ✮ , estimate x (or say X ❂ xx T ). Song Mei (Stanford University) TAP free energy September 19, 2018 3 / 29

  14. Bayes estimation in Z ✷ synchronization ◮ Settings: Y ❂ ✭ ✕❂♥ ✮ xx T ✰ W ✿ x ✘ ❯♥✐❢✭ Z ♥ ✷ ✮ ❀ ◮ Estimate X ❂ xx T with loss: ❵ ✭ X ❀ ❝ X ✮ ❂ ✭✶ ❂♥ ✷ ✮ ❦ X � ❝ X ❦ ✷ ❋ ✿ ◮ For ✕ ❁ ✶ , estimation is impossible. ◮ For ✕ ❃ ✶ , estimation is possible and efficient, e.g., spectral estimator (Baik, Ben Arous, Peche phase transition). ◮ The optimal estimator is the Bayes estimator (also minimax estimator): X ❇❛②❡s ❂ E ❬ xx T ❥ Y ❪ ✿ ❝ Song Mei (Stanford University) TAP free energy September 19, 2018 4 / 29

  15. Bayes estimation in Z ✷ synchronization ◮ Settings: Y ❂ ✭ ✕❂♥ ✮ xx T ✰ W ✿ x ✘ ❯♥✐❢✭ Z ♥ ✷ ✮ ❀ ◮ Estimate X ❂ xx T with loss: ❵ ✭ X ❀ ❝ X ✮ ❂ ✭✶ ❂♥ ✷ ✮ ❦ X � ❝ X ❦ ✷ ❋ ✿ ◮ For ✕ ❁ ✶ , estimation is impossible. ◮ For ✕ ❃ ✶ , estimation is possible and efficient, e.g., spectral estimator (Baik, Ben Arous, Peche phase transition). ◮ The optimal estimator is the Bayes estimator (also minimax estimator): X ❇❛②❡s ❂ E ❬ xx T ❥ Y ❪ ✿ ❝ Song Mei (Stanford University) TAP free energy September 19, 2018 4 / 29

  16. Bayes estimation in Z ✷ synchronization ◮ Settings: Y ❂ ✭ ✕❂♥ ✮ xx T ✰ W ✿ x ✘ ❯♥✐❢✭ Z ♥ ✷ ✮ ❀ ◮ Estimate X ❂ xx T with loss: ❵ ✭ X ❀ ❝ X ✮ ❂ ✭✶ ❂♥ ✷ ✮ ❦ X � ❝ X ❦ ✷ ❋ ✿ ◮ For ✕ ❁ ✶ , estimation is impossible. ◮ For ✕ ❃ ✶ , estimation is possible and efficient, e.g., spectral estimator (Baik, Ben Arous, Peche phase transition). ◮ The optimal estimator is the Bayes estimator (also minimax estimator): X ❇❛②❡s ❂ E ❬ xx T ❥ Y ❪ ✿ ❝ Song Mei (Stanford University) TAP free energy September 19, 2018 4 / 29

Download Presentation
Download Policy: The content available on the website is offered to you 'AS IS' for your personal information and use only. It cannot be commercialized, licensed, or distributed on other websites without prior consent from the author. To download a presentation, simply click this link. If you encounter any difficulties during the download process, it's possible that the publisher has removed the file from their server.

Recommend


More recommend