stat.MLOct 7, 2026

Reflected Anchored Langevin Algorithms

Authors: Changwei Tu, Xiaoyu Wang, Yingli Wang, Xicheng Zhang, Lingjiong Zhu

Organizations: Hong Kong University of Science and Technology (Guangzhou), Guangzhou, Guangdong Province, People’s Republic of China · School of Mathematical Sciences, Fudan University, Shanghai, People’s Republic of China · School of Mathematics and Statistics, Beijing Institute of Technology, Beijing 100081, People’s Republic of China · Department of Mathematics, Florida State University, Tallahassee, Florida, United States of America

Abstract

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.

Figures & tables

Explore similar work

Sep 21, 2026stat.ML

Penalized Nonreversible Langevin for Constrained Sampling

We propose penalized nonreversible Langevin algorithms for sampling from π(x)∝e−f(x)1C(x)π(x)\propto e^{-f(x)}\mathbf 1_{\mathcal C}(x), where C⊂Rd\mathcal C\subset\mathbb R^d is a compact convex set. The algorithms combine a squared distance penalty with constant or compatible state dependent skew symmetric perturbations that preserve the penalized Gibbs distribution. For smooth, possibly nonconvex ff, we derive nonasymptotic total variation bounds for the full gradient algorithm under a log Sobolev inequality. When unbiased stochastic gradients are available, we establish 22-Wasserstein bounds under global contraction and Lipschitz conditions on the full drift in an adapted quadratic metric. For a fixed penalty parameter, the error relative to the penalized Gibbs distribution decays exponentially to an O(η)\mathcal{O}(\sqrtη) neighborhood, where ηη is the stepsize. We also bound the discrepancy between the penalized Gibbs distribution and the constrained target. In a two dimensional quadratic model, we establish nonreversible acceleration by tuning the skew perturbation to the curvature imbalance induced by penalization. With the target accuracy and smaller curvature fixed and initial Wasserstein distances uniformly bounded, tuning the skew perturbation improves the sufficient Euler iteration bound from linear to logarithmic in the curvature ratio. Numerical experiments evaluate the algorithms on constrained Bayesian regression, classification, neural networks, and truncated sampling, and examine the acceleration mechanism in a stochastic quadratic model.
May 1, 2026stat.ML

Decentralized Proximal Stochastic Gradient Langevin Dynamics

We propose Decentralized Proximal Stochastic Gradient Langevin Dynamics (DE-PSGLD), a decentralized Markov chain Monte Carlo (MCMC) algorithm for sampling from a log-concave probability distribution constrained to a convex domain. Constraints are enforced through a shared proximal regularization based on the Moreau-Yosida envelope, enabling unconstrained updates while preserving consistency with the target constrained posterior. We establish non-asymptotic convergence guarantees in the 2-Wasserstein distance for both individual agent iterates and their network averages. Our analysis shows that DE-PSGLD converges to a regularized Gibbs distribution and quantifies the bias introduced by the proximal approximation. We evaluate DE-PSGLD for different sampling problems on synthetic and real datasets. As the first decentralized approach for constrained domains, our algorithm exhibits fast posterior concentration and high predictive accuracy.
Sep 30, 2026cs.LG

Near-Linear Accuracy Bounds for Moreau--Yosida Unadjusted Langevin Sampling

We establish near-linear accuracy bounds for the classical Moreau--Yosida unadjusted Langevin algorithm (MYULA). The target is π∝e−f−gπ\propto e^{-f-g}, where f∈C2(Rd)f\in C^2(\mathbb{R}^d) is mm-strongly convex with Lipschitz gradient and gg is convex and globally Lipschitz. Under an explicit parameter-dependent step-size condition, we bound the invariant-measure bias relative to the Moreau-smoothed target by O~(h)\widetilde O(h), with only logarithmic dependence on the inverse smoothing parameter in the error coefficient. Combining this estimate with the Moreau approximation bias and Wasserstein contraction gives O~(ε−1)\widetilde O(\varepsilon^{-1}) iterations to make the NNth-iterate law μNμ_N satisfy m W2(μN,π)≤ε\sqrt m\,W_2(μ_N,π)\le\varepsilon, for fixed model parameters and initialization. We bound the stationary error directly, without assuming third derivatives or a Lipschitz Hessian. Each iteration uses one gradient evaluation and one exact proximal evaluation. The key idea in our analysis is to convert a second-order stationary residual into a Wasserstein bound using a Poisson-based estimate.