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