RNE: plug-and-play diffusion inference-time control and energy-based training
AI总结 本文提出了一种名为RNE的插件式扩散模型方法,用于在推理阶段实现对生成过程的控制,并支持基于能量的训练。RNE基于路径分布之间的密度比概念,建立了边缘密度与转移核之间的基本联系,从而统一了扩散密度估计、推理控制和能量训练等多个任务。实验表明,RNE在推理控制任务中表现出色,同时为能量型扩散模型提供了简单高效的正则化方法,并适用于连续和离散扩散模型。
Comments Accepted at ICLR 2026