arXiv · cs.LG· Samar Hadou, Yigit Berkay Uslu, Alejandro Ribeiro·· 4 小时前
约束扩散模型与原始-对偶推理(PDI):从平均约束优化分布中采样
Constrained Diffusion Models with Primal-Dual Inference
AI 导读
研究提出约束扩散模型与原始-对偶推理(PDI),用于从带平均约束的熵正则化优化问题的最优分布中采样。PDI 在 Lagrangian 对偶域中联合推断最优原始分布及其对偶变量,每个反向扩散步用当前乘子对应的 score 场去噪,再依据去噪样本的约束违反程度做对偶上升更新。作者训练单一对偶条件 score 网络,并证明对偶迭代收敛到对偶最优邻域、生成分布逼近已知最优乘子的 oracle 采样器。
来源:arXiv · cs.LG · arxiv.org