arXiv · cs.LG· Jinyuan Chang, Chenguang Duan, Yuling Jiao, Ruoxuan Li, Yi Xu, Jerry Zhijian Yang·· 16 小时前
基于变分稳定 Doob's Matching 的扩散模型推理时对齐
Inference-Time Alignment for Diffusion Models via Variationally Stable Doob's Matching
AI 导读
研究提出变分稳定 Doob's matching，一种基于 Doob's h-transform 的扩散模型推理时对齐框架，通过梯度正则化回归同时估计 h 函数及其梯度，得到一致的引导估计。理论上给出了估计引导的非渐近收敛速率，并证明生成分布在 Wasserstein-2 距离下的非渐近收敛保证。该方法在低维线性子空间假设下可自适应未知低维结构，缓解维度灾难，数值实验显示优于基线方法。
来源：arXiv · cs.LG · arxiv.org