arXiv:2606.11437cs.DScs.AI2026-06

提出测试时训练的理论框架,揭示其采样效率边界。

The Power of Test-Time Training for Approximate Sampling

  • 将测试时训练形式化为从已知分布类中近似采样
  • 证明在大分布类下采样查询复杂度有二次下界
  • 发现分布类规模受限时可突破下界,为理论奠基

高效从复杂概率分布中采样是生成式AI中的核心问题。本文将测试时训练(TTT)形式化为:给定一个可近似估计密度的预言机 $\hat \mu$,从已知分布类 ${F}$ 中的某目标分布 $\mu^\star$ 生成样本。该问题与经典文献中“计数转采样”问题紧密相关——当 ${F}$ 为所有分布时,二者完全等价。本文首先证明,在足够大的分布类 ${F}$ 下,采样所需的查询次数存在二次下界,表明 Jerrum & Sinclair(1989)提出的随机游走方法及其后续改进(Hayes & Sinclair, 2010)已达最优。这回答了该领域长期存在的开放问题。随后,我们证明若 ${F}$ 的规模受到合理限制,则此下界可被突破。这一结果可视为对 TTT 的抽象,为构建系统性的理论框架提供了起点。

原文摘要 · Abstract (English)

Efficiently sampling from a complex probability distribution is a fundamental problem which has become increasingly pertinent in recent years with the rise of generative AI, as sophisticated sampling procedures from LLMs have been proposed to solve challenging reasoning problems. The efficacy of such sampling algorithms is limited, however, by the relationship between the LLM and the particular sampling task at hand, which has motivated the framework of test-time training (TTT). TTT works by updating a model's weights in response to partial generations and reward feedback received at inference time, thus adapting to the particular problem. In this work, we propose a formalization for TTT as the problem of producing a sample from a given probability measure $μ^\star$ belonging to a known class ${F}$ of distributions, given an oracle $\hat μ$ which yields approximate density estimates for $μ^\star$. This is closely related to the problem of reducing sampling to approximate counting studied in seminal works of Jerrum, Valiant & Vazirani (1986) and Jerrum & Sinclair (1989): namely, when ${F}$ is the class of all distributions, it coincides exactly with the aforementioned counting-to-sampling reduction. In this paper, we first show a quadratic lower bound on the query complexity of sampling from $μ^\star$ given query access to $\hat μ$ (for sufficiently large classes ${F}$), thus showing that the random walk approach proposed by Jerrum & Sinclair (1989) and refined by Hayes & Sinclair (2010), is optimal. This answers an open question posed by Hayes & Sinclair. We then show that this lower bound can be circumvented if the size of ${F}$ is bounded appropriately. As we discuss, this latter result can be viewed as an abstraction of TTT, and thus represents a starting point for the development of a principled theoretical framework for TTT.

测试时训练采样理论概率推断

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。