The adaptive complexity of parallelized log-concave sampling
Huanjian Zhou, Baoxiang Wang, Masashi Sugiyama
Abstract
In large-data applications, such as the inference process of diffusion models, it is desirable to design sampling algorithms with a high degree of parallelization. In this work, we study the adaptive complexity of sampling, which is the minimum number of sequential rounds required to achieve sampling given polynomially many queries executed in parallel at each round. For unconstrained sampling, we examine distributions that are log-smooth or log-Lipschitz and log strongly or nonstrongly concave. We show that an almost linear iteration algorithm cannot return a sample with a specific exponentially small error under total variation distance. For box-constrained sampling, we show that an almost linear iteration algorithm cannot return a sample with sup-polynomially small error under total variation distance for log-concave distributions. Our proof relies upon novel analysis with the characterization of the output for the hardness potentials based on the chain-like structure with random partition and classical smoothing techniques.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext b5dd19cb-b1ac-4ae3-adcb-999686247c09Cited by top-tier papers2
- Complexity Analysis of Normalizing Constant Estimation: from Jarzynski Equality to Annealed Importance Sampling and beyondWei Guo, Molei Tao, Yongxin ChenICLR 2026 · 12 citations
- The Adaptive Complexity of Minimizing Relative Fisher InformationHuanjian Zhou, Masashi SugiyamaNeurIPS 2025 · 1 citation
Builds on14
- Parallel Sampling of Diffusion ModelsAndy Shih, Suneel Belkhale, Stefano Ermon, Dorsa Sadigh et al.NeurIPS 2023 · 144 citations
- Exponential ergodicity of mirror-Langevin diffusionsSinho Chewi, Thibaut Le Gouic, Chen Lu, Tyler Maunu et al.NeurIPS 2020 · 62 citations
- Accelerating Diffusion Models with Parallel Sampling: Inference at Sub-Linear Time ComplexityHaoxuan Chen, Yinuo Ren, Lexing Ying, Grant M. RotskoffNeurIPS 2024 · 53 citations
- Near-Optimal Lower Bounds For Convex Optimization For All Orders of SmoothnessAnkit Garg, Robin Kothari, Praneeth Netrapalli, Suhail SherifNeurIPS 2021 · 23 citations
- Sampling from Log-Concave Distributions with Infinity-Distance GuaranteesOren Mangoubi, Nisheeth K. VishnoiNeurIPS 2022 · 15 citations
Related papers
- Faster Diffusion Sampling with Randomized Midpoints: Sequential and ParallelShivam Gupta, Linda Cai, Sitan ChenICLR 2025
- High-accuracy sampling for diffusion models and log-concave distributionsFan Chen, Sinho Chewi, Constantinos Daskalakis, Alexander RakhlinICML 2026 · 12 citations
- Parallel Simulation for Log-concave Sampling and Score-based Diffusion ModelsHuanjian Zhou, Masashi SugiyamaICML 2025
- Optimal Underdamped Langevin MCMC MethodZhengmian Hu, Feihu Huang, Heng HuangNeurIPS 2021 · 5 citations
- Improved Convergence Rate for Diffusion Probabilistic ModelsGen Li, Yuchen JiaoICLR 2025
