Finite-Time Analysis of Projected Langevin Monte Carlo Ronen Eldan Microsoft Research
–Neural Information Processing Systems
We analyze the projected Langevin Monte Carlo (LMC) algorithm, a close cousin of projected Stochastic Gradient Descent (SGD). We show that LMC allows to sample in polynomial time from a posterior distribution restricted to a convex body and with concave log-likelihood. This gives the first Markov chain to sample from a log-concave distribution with a first-order oracle, as the existing chains with provable guarantees (lattice walk, ball walk and hit-and-run) require a zerothorder oracle. Our proof uses elementary concepts from stochastic calculus which could be useful more generally to understand SGD and its variants.
Neural Information Processing Systems
Mar-13-2024, 03:14:28 GMT
- Technology: