摘要
arXiv:2502.03686v1 类型: cross
摘要: 扩散模型在样本质量方面表现出色,但现有的引导方法往往需要额外的模型训练,或者仅限于特定任务。我们从变分推理和控制的角度回顾了扩散模型中的引导方法,引入了扩散轨迹匹配(DTM),以使预训练的扩散轨迹满足终端成本。DTM 统一了广泛类别的引导方法,并允许新的实现。我们在此框架内提出了一种新方法,在几种线性和(盲)非线性逆问题上取得了最先进的结果,无需额外的模型训练或修改。例如,在 ImageNet 非线性去模糊任务中,我们的模型实现了 34.31 的 FID 分数,显著优于最佳预训练方法基线(FID 78.07)。未来我们将提供代码。