Explorerβ€ΊData Scienceβ€ΊMachine Learning
Research PaperResearchia:202610.12088

KL Before and After Mixing: A Round-Complexity Separation in Parallel Sampling

Tianyi Wu

Abstract

A parallel sampler can choose at random which coordinates to generate together. For a target $p$ and the output law $q_U$ of a schedule $U$ drawn afresh for each output, we compare the schedule-averaged divergence $\mathbb{E}_U D_{\mathrm{KL}}(p\|q_U)$ with the output-law divergence $D_{\mathrm{KL}}(p\|\mathbb{E}_U q_U)$, in nats. Rounds commit independent draws from exact conditional marginals frozen at the round's start, each output's schedule is drawn before sampling from a finite mixture of ...

Submitted: October 12, 2026Subjects: Machine Learning; Data Science

Description / Details

A parallel sampler can choose at random which coordinates to generate together. For a target pp and the output law qUq_U of a schedule UU drawn afresh for each output, we compare the schedule-averaged divergence EUDKL(pβˆ₯qU)\mathbb{E}_U D_{\mathrm{KL}}(p\|q_U) with the output-law divergence DKL(pβˆ₯EUqU)D_{\mathrm{KL}}(p\|\mathbb{E}_U q_U), in nats. Rounds commit independent draws from exact conditional marginals frozen at the round's start, each output's schedule is drawn before sampling from a finite mixture of schedules, and every scheduled round is charged. For the uniform distribution on binary strings of even length nn with n/2n/2 ones, the least expected number of rounds at a fixed tolerance Ξ΅>0\varepsilon>0 is ΘΡ((log⁑n)2)Θ_\varepsilon((\log n)^2) under the first criterion but (eβˆ’Ξ΅+o(1))log⁑2ln⁑n(e^{-\varepsilon}+o(1))\log_2\ln n under the second, and the double-logarithmic order is attainable with the same round count on every run. For this target, a count-state recursion gives the exact least output divergence at every round cap over all causal randomized policies. The separation extends to weighted fixed-count targets with bounded weight ratios and quota fractions bounded away from zero and one, without permutation symmetry, and to the full-support Beta(1,1)(1,1)--Bernoulli target, where the sharp expected cost under the output-law criterion is (eβˆ’Ξ΅/2+o(1))log⁑2log⁑2n(e^{-\varepsilon}/2+o(1))\log_2\log_2n, attained with rare nn-round runs. Prior work implies the order of the schedule-averaged cost on the balanced target and its OΞ΅((log⁑n)2)O_\varepsilon((\log n)^2) upper bounds on both extensions.


Source: arXiv:2610.13924v1 - http://arxiv.org/abs/2610.13924v1 PDF: https://arxiv.org/pdf/2610.13924v1 Original Link: http://arxiv.org/abs/2610.13924v1

Please sign in to join the discussion.

No comments yet. Be the first to share your thoughts!

Access Paper
View Source PDF
Submission Info
Date:
Oct 12, 2026
Topic:
Data Science
Area:
Machine Learning
Comments:
0
Bookmark