Decentralized Proximal Stochastic Gradient Langevin Dynamics

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Islam, Mohammad Rafiqul, Zhu, Lingjiong
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918477439172608
author Islam, Mohammad Rafiqul
Zhu, Lingjiong
author_facet Islam, Mohammad Rafiqul
Zhu, Lingjiong
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2605_00723
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Decentralized Proximal Stochastic Gradient Langevin Dynamics
Islam, Mohammad Rafiqul
Zhu, Lingjiong
Machine Learning
Probability
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.
title Decentralized Proximal Stochastic Gradient Langevin Dynamics
topic Machine Learning
Probability
url https://arxiv.org/abs/2605.00723