English

Score Operator Newton transport

Statistics Theory 2024-03-12 v3 Machine Learning Numerical Analysis Numerical Analysis Statistics Theory

Abstract

We propose a new approach for sampling and Bayesian computation that uses the score of the target distribution to construct a transport from a given reference distribution to the target. Our approach is an infinite-dimensional Newton method, involving a linear PDE, for finding a zero of a ``score-residual'' operator. We prove sufficient conditions for convergence to a valid transport map. Our Newton iterates can be computed by exploiting fast solvers for elliptic PDEs, resulting in new algorithms for Bayesian inference and other sampling tasks. We identify elementary settings where score-operator Newton transport achieves fast convergence while avoiding mode collapse.

Keywords

Cite

@article{arxiv.2305.09792,
  title  = {Score Operator Newton transport},
  author = {Nisha Chandramoorthy and Florian Schaefer and Youssef Marzouk},
  journal= {arXiv preprint arXiv:2305.09792},
  year   = {2024}
}

Comments

24 pages; AISTATS 2024

R2 v1 2026-06-28T10:36:26.840Z