arXiv:2502.01692v5 宣布类型: replace-cross 摘要: 引导扩散模型生成是定制预训练扩散模型以解决特定下游任务生成过程的一个有前景的方向。现有的引导扩散模型要么依赖于使用预先收集的数据集训练引导模型，要么需要使目标函数可微。然而，对于大多数现实世界任务，离线数据集往往不可用，其目标函数往往不可微，例如具有人类偏好的图像生成、药物发现中的分子生成和材料设计。因此，我们需要一种能够在运行时收集数据且支持黑盒目标函数的在线算法。此外，算法的查询效率也至关重要，因为在实际场景中，查询的目标评估往往很昂贵。在本文中，我们提出了一种新颖且简单的算法——Fast Direct，用于高效的在线黑盒目标生成。Fast Direct在数据流形上构建一个伪目标，以通用方向更新扩散模型的噪声序列，这有望实现高效的引导生成。在十二项高分辨率（1024×1024）图像目标生成任务和六项3D分子目标生成任务上的广泛实验表明，查询效率分别提高了6倍至10倍和11倍至44倍。我们的实现已公开发布在：https://github.com/kimyong95/guide-stable-diffusion/tree/fast-direct