mlsanityData to 14 Oct 2026

Joint stochastic localization and applications

arXiv 2505.13410 ↗First posted 19 May 2025math.ST math.PR stat.MLcited by 2

Not on the map. Why?

Abstract (arXiv)

Stochastic localization is a pathwise analysis technique that has emerged as a powerful tool in high-dimensional probability and sampling. In this work, we extend stochastic localization to a joint framework for coupling probability measures and explore its applications in distributional data analysis. We first unify existing stochastic localization processes under Eldan's α-scheme and characterize their localization rates. Building on this, we introduce a joint scheme to couple probability measures via concurrent α-schemes driven by a shared Brownian motion. This construction is canonical and induces a family of metrics on the space of probability measures, which we call Eldan's α-distance. Alternative variants that extrapolate optimal Gaussian couplings to log-concave measures are also discussed. We study the theoretical properties of Eldan's α-distance, including its restriction to Gaussian measures and its behavior under affine transformations. For α = 0, we show it is topologically equivalent to the 2-Wasserstein distance for measures supported on a common compact set; we also relate its weighted variants to linearized optimal transport in Wiener space and to score-matching objectives in training diffusion models. Computationally, we develop efficient estimators for Eldan's α-distance in the cases α=0 and α=1/2, with rigorous error guarantees for log-concave and finitely supported measures in the former setting and Gaussian measures in the latter. Finally, we apply Eldan's α-distance as a scalable surrogate for the 2-Wasserstein distance to enable fast pairwise distance estimation and approximate computation of Wasserstein barycenters.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 13 of its 54 references that are in the map. First posted on arXiv 19 May 2025.

Builds on

13 of 54 references.

References · 13

  1. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  2. 2017Learning Generative Models with Sinkhorn Divergencescited by 314
  3. 2023Nearly d-Linear Convergence Bounds for Diffusion Models via Stochastic Localizationcited by 235
  4. 2018Improving GANs Using Optimal Transportcited by 123
  5. 2024Score-based Diffusion Models via Stochastic Differential Equations -- a Technical Tutorialcited by 60
  6. 2022Sampling from the Sherrington-Kirkpatrick Gibbs measure via algorithmic stochastic localizationcited by 48
  7. 2019Asymptotic Guarantees for Learning Generative Models with the Sliced-Wasserstein Distancecited by 43
  8. 2023Sampling, Diffusions, and Stochastic Localizationcited by 40
  9. 2024Stochastic Localization via Iterative Posterior Samplingcited by 28
  10. 2022Localization Schemes: A Framework for Proving Mixing Bounds for Markov Chainscited by 16
  11. 2016Eldan's Stochastic Localization and the KLS Conjecture: Isoperimetry, Concentration and Mixingcited by 13
  12. 2022A Slightly Improved Bound for the KLS Constantcited by 10
1 more
  1. 2025Generative diffusion for perceptron problems: statistical physics analysis and efficient algorithmscited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Discrete Stochastic Localization for Non-autoregressive Generation
  2. 2026Discrete Stochastic Localization for Non-autoregressive Generation
  3. 2023Structured Approximations of Measures
  4. 2025On the Contractivity of Stochastic Interpolation Flow
  5. 2024Large Deviation Upper Bounds and Improved MSE Rates of Nonlinear SGD: Heavy-tailed Noise and Power of Symmetry
  6. 2024Denoising Diffusions with Optimal Transport: Localization, Curvature, and Multi-Scale Complexity
  7. 2021Improving Approximate Optimal Transport Distances using Quantization
  8. 2026Local Fokker--Planck Geometry for Score Estimation: Heat-Ball Mean-Value Representations and Exact High-Dimensional Sampling