math.STSep 7, 2026
SaveAccelerated High-Accuracy Sampling from a Warm Start via the Proximal Bouncy Particle Sampler
Organizations: Department of Electrical Engineering and Computer Science, Massachusetts Institute of Technology. · Department of Statistics and Data Science, Yale University. · Department of Mathematics, Duke University. · Department of Mathematics, Massachusetts Institute of Technology.
Abstract
We study the problem of sampling from on , where is -strongly convex and -smooth, and write . We design and analyze the Proximal Bouncy Particle Sampler (Proximal BPS), a new sampler that combines ideas from the proximal sampler and the bouncy particle sampler. From a warm start initialization with Rényi divergence w.r.t. , Proximal BPS returns a sample whose law is -close to in total variation distance using gradient queries in expectation.
Figures & tables
| Symbol | Meaning | Symbol | Meaning |
|---|---|---|---|
| Target and problem parameters | |||
| Ambient dimension. | Potential . | ||
| Target law . | Strong-convexity and smoothness constants of . | ||
| Condition number . | Target total variation accuracy and order-2 Rényi warm start budget. | ||
| Proximal augmentation | |||
| Proximal scale. | Augmented target on . | ||
Table 1: Frequently used notation, grouped by theme.
Figure 1: Geometric overview of Proximal BPS. (a) The initial law is lifted to the augmented distribution. At each iteration, (b) the auxiliary point is reflected through the current position, and (c) a conditional reference point is drawn. With probability , that point becomes the next position; with probability , it determines the equilibrium position of the deterministic harmonic flow for the time- BPS half-turn process in (d), which starts from the current position.
| (3.8) |
Algorithm 1 Simulation of the ideal conditional half-turn process
Explore similar work
We develop a new low-accuracy sampler, called smoothed Picard Hamiltonian Monte Carlo, which combines Gaussian smoothing, Picard iteration, and higher-order discretization. For a log-concave target in dimension satisfying , with condition number , smoothed Picard HMC returns a sample with using gradient queries. We also prove stronger bounds, and then develop an algorithmic framework, the recursive warm start generator, to upgrade these bounds to stronger divergence guarantees. This produces a warm start for the proximal bouncy particle sampler, introduced in a companion work, leading to a high-accuracy log-concave sampler with complexity .
Reflected Proximal Dynamics for Accelerated High-Accuracy Log-Concave Sampling
We study sampling from on , where is -strongly convex and -smooth with condition number . For this problem, the proximal sampler preserves the target exactly and recovers overdamped Langevin dynamics in the small-step limit, but its Gaussian resampling step discards directional information. Underdamped Langevin dynamics retains such information through momentum to accelerate exploration, motivating a proximal construction that combines target preservation with directional persistence. We introduce reflected proximal dynamics, which combines proximal sampling with an exact reflection of the auxiliary variable, conditional kinetic motion, and controlled refreshment. We develop two realizations: a flux-resampled underdamped half-turn and a persistent-momentum quarter-turn. From a warm start with constant divergence, both methods achieve total variation error using expected oracle queries. We further remove the supplied warm-start assumption by combining either sampler with existing posterior samplers in a proximal initialization stage with an optimized coarse step size. Accounting for both initialization and sampling yields expected oracle queries. In both regimes, the dependence on the target accuracy is , making reflected proximal dynamics a promising framework for accelerated high-accuracy log-concave sampling.
Alternating Diffusion for Proximal Sampling with Zeroth Order Queries
This work introduces a new approximate proximal sampler that operates solely with zeroth-order information of the potential function. Prior theoretical analyses have revealed that proximal sampling corresponds to alternating forward and backward iterations of the heat flow. The backward step was originally implemented by rejection sampling, whereas we directly simulate the dynamics. Unlike diffusion-based sampling methods that estimate scores via learned models or by invoking auxiliary samplers, our method treats the intermediate particle distribution as a Gaussian mixture, thereby yielding a Monte Carlo score estimator from directly samplable distributions. Theoretically, when the score estimation error is sufficiently controlled, our method inherits the exponential convergence of proximal sampling under isoperimetric conditions on the target distribution. In practice, the algorithm avoids rejection sampling, permits flexible step sizes, and runs with a deterministic runtime budget. Numerical experiments demonstrate that our approach converges rapidly to the target distribution, driven by interactions among multiple particles and by exploiting parallel computation.