stat.MLAug 25, 2026

Rényi Tracking Bounds for Langevin Dynamics with Moving Targets

Authors: Yuchen Xin, Jingxin Zhan, Zhihua Zhang

Organizations: School of Mathematical Sciences, Peking University

Abstract

We study Langevin diffusion and Langevin Monte Carlo (LMC) when the target distribution changes over time. Under a log-Sobolev inequality (LSI), we derive non-asymptotic Rényi-divergence guarantees for tracking the current target. The framework covers continuous-time Langevin diffusion and its discretizations. We then apply the results to nonsmooth sampling based on successive Moreau envelopes. For this scheme, we give explicit choices of the smoothing parameters and step sizes, together with corresponding complexity bounds. To our knowledge, these are the first non-asymptotic Rényi-divergence tracking bounds for Langevin dynamics with discrete target updates.

Explore similar work

Oct 7, 2026

Reflected Anchored Langevin Algorithms

First order Langevin algorithms for constrained sampling in machine learning, such as projected Langevin Monte Carlo which are based on discretizations of reflected Langevin dynamics, require differentiable log densities that limits their applicability. This paper introduces reflected anchored Langevin dynamics (RALD), a reflected diffusion that converges to non-differentiable targets on constrained domains. The method uses a smooth anchored reference potential and multiplies the drift and noise covariance of its reflected Langevin dynamics by the same state dependent scaling factor. Its Euler-Maruyama discretization with projection gives reflected anchored Langevin Monte Carlo (RALMC) algorithm. We prove explicit convergence bounds and iteration complexity for RALMC in the 2-Wasserstein distance to the target distribution. Numerical experiments are provided to illustrate the theoretical predictions and the empirical performance of the method.
Aug 13, 2026

Active-Trace Complexity Bounds for Moreau--Yosida Unadjusted Langevin Sampling

We study the Moreau--Yosida unadjusted Langevin algorithm (MYULA) for the nonsmooth composite target π(dx)∝exp⁡{−f(x)−g(x)} dx,x∈Rd,π(dx)\propto \exp\{-f(x)-g(x)\}\,dx, \qquad x\in\mathbb R^d, where ff is mm-strongly convex with LfL_f-Lipschitz gradient and gg is convex and GG-Lipschitz. Let gλg_λ be the Moreau envelope of gg, πλπ_λ the corresponding smoothed target, and aλ=tr⁡Hλa_λ=\operatorname{tr}H_λ, where HλH_λ is the a.e./weak Hessian of gλg_λ. We show that the leading MYULA discretization error is controlled by the reference active trace BrefB_{\mathrm{ref}}, the average of aλa_λ along the heat substep of one MYULA update started from πλπ_λ, rather than by the global curvature bound d/λd/λ. If MλM_λ is an a.e. upper bound for aλa_λ, then, up to logarithmic factors, N≲1m[Lf+τf+G2+Brefεalg2+Mλεalg],τf:=sup⁡xtr⁡∇2f(x),N \lesssim \frac{1}{m} \left[ L_f + \frac{ τ_f+G^2+B_{\mathrm{ref}} }{ \varepsilon_{\mathrm{alg}}^2 } + \frac{M_λ}{\varepsilon_{\mathrm{alg}}} \right], \qquad τ_f:= \sup_x\operatorname{tr}\nabla^2 f(x), iterations suffice to ensure m W2(μN,πλ)≤εalg\sqrt m\,W_2(μ_N,π_λ)\leq\varepsilon_{\mathrm{alg}}, where μNμ_N is the law of the NN-th iterate and W2W_2 is the quadratic Wasserstein distance. We also prove the Moreau-bias bound m W2(πλ,π)≤G2λ4.\sqrt m\,W_2(π_λ,π) \leq \frac{G^2λ}{4}. Thus, choosing λ≍ε/G2λ\asymp\varepsilon/G^2 gives an end-to-end guarantee for ππ. The universal estimate Bref≤d/λB_{\mathrm{ref}}\leq d/λ yields O~(ε−3)\widetilde O(\varepsilon^{-3}) accuracy dependence. For the structured piecewise-linear, lasso-type, group, and total-variation penalties considered here, curvature--tube estimates make BrefB_{\mathrm{ref}} independent of λλ, yielding O~(ε−2)\widetilde O(\varepsilon^{-2}) for the same classical MYULA kernel.
May 29, 2026

Improved Guarantees for Langevin Monte Carlo with Average Smoothness

We establish improved nonasymptotic bounds for Langevin Monte Carlo in the strongly log-concave setting, when the error is measured by the Wasserstein distance. The main result shows that the discretization error is governed by an average coordinate-wise smoothness constant, rather than by the usual global smoothness constant. The proof is short and probabilistic, and relies on a refined use of the synchronous coupling. We further show that the same ideas lead to improved bounds for variable step sizes, for potentials whose Laplacian is Lipschitz-continuous, and for finite-sum problems sampled by stochastic-gradient Langevin dynamics with fixed point control variates. In the Laplacian-smooth case, the usual Hessian-Lipschitz contribution is replaced by a weaker trace-type third-order smoothness quantity. In the finite-sum setting, the resulting SGLD bound improves the dependence on the root mean square smoothness of the component functions. Applications to generalized linear models with Gaussian design show that these refinements can yield substantial, dimension-dependent improvements over previously known bounds, especially for correlated covariates.