arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.03211cs.LGcs.AI
收藏

前瞻样本奖励引导用于扩散模型的测试时缩放

Lookahead Sample Reward Guidance for Test-Time Scaling of Diffusion Models

KAIST(韩国科学技术院)

Yeongmin Kim, Donghyeok Shin, Byeonghu Na, Minsang Park, Richard Lee Kim, Il-Chul Moon

AI总结:

提出一种高效测试时缩放方法LiDAR采样,通过前瞻几步采样和精确求解器引导粒子向高奖励区域移动,无需反向传播,在GenEval上达到与最新梯度引导方法相同性能且加速9.5倍。

摘要:

扩散模型已展现出强大的生成性能;然而,生成的样本往往未能完全符合人类意图。本文研究了一种高效的测试时缩放方法,用于从具有更高人类对齐奖励值的区域进行采样。现有的计算期望未来奖励(EFR)方法面临重要限制:反向展开导致采样成本过高,而基于Tweedie的方法(包括顺序蒙特卡洛和梯度引导)则存在偏差和固有的采样问题。我们证明,任何$\mathbf{x}_t$处的EFR仅需使用预训练扩散模型的边际样本即可计算,从而无需神经反向传播即可实现闭式奖励引导。为了进一步提高效率,我们引入了少步前瞻采样和一个精确求解器,引导粒子向高奖励的前瞻样本移动。我们将这种采样方案称为LiDAR采样。LiDAR在SDXL上达到了与最新梯度引导方法相同的GenEval性能,并实现了9.5倍的加速。我们在https://github.com/aailab-kaist/Diffusion-LiDAR-Sampling 上发布了代码。

英文摘要:

Diffusion models have demonstrated strong generative performance; however, generated samples often fail to fully align with human intent. This paper studies an efficient test-time scaling method for sampling from regions with higher human-aligned reward values. Existing methods for computing the expected future reward (EFR) face important limitations: backward rollout incurs prohibitively high sampling costs, while Tweedie-based approaches, including Sequential Monte Carlo and gradient guidance, suffer from bias and inherent sampling issues. We show that the EFR at any $\mathbf{x}_t$ can be computed using only marginal samples from a pre-trained diffusion model, enabling closed-form reward guidance without neural backpropagation. To further improve efficiency, we introduce a few-step lookahead sampling and an accurate solver that guides particles toward high-reward lookahead samples. We refer to this sampling scheme as LiDAR sampling. LiDAR achieves the same GenEval performance as the latest gradient guidance method for SDXL with a 9.5x speedup. We release the code at https://github.com/aailab-kaist/Diffusion-LiDAR-Sampling.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑