English

Fast and Efficient Parallel Sampling Using Higher Order Langevin Dynamics

Statistics Theory 2026-05-11 v2 Methodology Machine Learning Statistics Theory

Abstract

We study parallel sampling from high-dimensional strongly log-concave distributions. Langevin-based samplers converge rapidly in continuous time, but their discretizations are typically sequential and often require polynomially many steps in the dimension dd, the target accuracy ε1\varepsilon^{-1}, or both. Picard-based parallel sampling methods reduce this sequential depth to polylogarithmic scale by solving for many time-discretization points in parallel; however, existing guarantees often require a polynomial number of processors, leading to substantial memory and gradient-evaluation costs in high dimensions. We show that higher-order Langevin structure can reduce this parallel resource burden while preserving polylogarithmic sequential depth. Our method combines arbitrary-order Langevin dynamics with blockwise Lagrange polynomial interpolation. This sharper discretization reduces the number of parallel points required to achieve a target accuracy. Our results cover both higher-order smooth potentials and ridge-separable potentials, including models such as Bayesian logistic regression and two-layer neural networks, and improve upon the space complexity of the current literature on parallel log-concave sampling.

Keywords

Cite

@article{arxiv.2510.18242,
  title  = {Fast and Efficient Parallel Sampling Using Higher Order Langevin Dynamics},
  author = {Jaideep Mahajan and Kaihong Zhang and Feng Liang and Jingbo Liu},
  journal= {arXiv preprint arXiv:2510.18242},
  year   = {2026}
}
R2 v1 2026-07-01T06:56:58.978Z