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