Paper Reading Notes

Problem Setup

    Let be a probability measure on and let be the set of probability measures on with bounded second moment. Let and be real valued functions on . We will find it convenient to define vector valued functions whose components are the functions respectively.
    The paper proposes a primal-dual Langevin Monte carlo algorithm to sample from the measure that solves satisfying the constraints
    The constrained sampling problem encompasses a number of practical problems such as sampling from convex sets, by sampling from for a close convex set .Rate constrained Bayesian models

Background

    Let . be a smooth and strongly convex function. To the measure with density we can associate a diffusion process called Langevin diffusion, defined through the SDE The density of satisfies the Fokker-Planck equation In particular the KL divergence has the form In some sense the Fokker-Planck equation is the gradient flow of KL divergence in 2-Wasserstein space. The diffusion brings the distribution of closer to the measure from which we wish to draw samples.
    Discretization of the SDE through the forward Euler-Maruyama method leads to the Langevin Monte Carlo algorithm whereby is the size of the step. The convergence rate of this discretization has been established for potentials satisfying various smoothness and convexity conditions, or the measure satisfies appropriate LSI condition.

Duality

    Define the Lagrangian Put where is the normalizing constant. Then
    One has
    Define the dual function The Lagrangian minimizer is From the relaxation of the primal problem, we have the bound for all feasible .The dual problem seeks the optimal lower bound

Theorem

    Suppose there exists together with constants and such that
  1. for all
  2. for all
Then
This reduces the sampling from the solution distribution of the primal problem to sampling from One can obtain via dual ascent. However it requires computing intractible integrals with respect to the distribution .

References

1.  L. Chamon et al, Constrained Sampling with Primal-Dual Langevin Monte Carlo, NeurIPS 2024
2.  K. Ahn, S. Chewi, Efficient Constrained Sampling via the Mirror-Langevin Algorithm, NeurIPS 2021
3.  L. Li et al, Sampling with Mollified Interaction Energy Descent, ICLR 2023
4.  Q. Jiang, Mirror Langevin Monte Carlo: the Case Under Isoperimetry, NeurIPS 2021
5.  J. Shi et al., Sampling with Mirrored Stein Operators, ICLR 2022
6.  M. Girolami, B. Calderhead, Riemann manifold Langevin and Hamiltonian Monte Carlo methods, J. Royal Stat. Soc. 2011
7.  G. Roberts, R. Tweedie, Exponential convergence of Langevin distributions and their discrete approximations, Bernoulli 1996
8.  S. Ishida, H. Lavenant, Quantitative Convergence of a Discretization of Dynamic Optimal Transport Using the Dual Formulation, Found. Comp. Math 2024
9.  T. Lelièvre et al, Unbiasing Hamiltonian Monte Carlo Algorithms for a General Hamiltonian Function, Found. Comp. Math 2024
10.  A. Wibisono, Sampling as optimization in the space of measures: The Langevin dynamics as a composite optimization problem, PMLR 2018