arXiv:2508.19104cs.LGeess.IV2025-08NeurIPS被引 8

让扩散模型同时满足多个生成要求,不互相干扰。

Composition and Alignment of Diffusion Models using Constrained Learning

  • 用约束优化统一处理模型对齐与组合
  • 实验表明能有效满足多目标生成约束
  • 适合需要精准控制生成内容的场景

扩散模型因其从复杂分布中采样的能力,在生成建模中广泛应用。为提升生成样本质量并符合用户需求,常用方法包括:(i) 对齐,即微调扩散模型以匹配奖励函数;(ii) 组合,将多个预训练扩散模型融合,各自强调生成结果中的特定属性。然而,优化多个奖励或组合多个模型时常引发权衡,因各属性可能相互冲突。现有方法无法保证最终模型能忠实生成具备所有期望属性的样本。为此,我们提出一种约束优化框架,通过强制对齐模型满足奖励约束和/或保持与预训练模型的接近度,统一处理对齐与组合问题。我们提供了约束对齐与组合问题解的理论表征,并设计基于拉格朗日的原-对偶训练算法近似求解。在图像生成任务中,我们实证验证了该方法在对齐与组合上的有效性,结果表明所生成模型能有效满足各项约束。代码开源:https://github.com/shervinkhalafi/constrained_comp_align

原文摘要 · Abstract (English)

Diffusion models have become prevalent in generative modeling due to their ability to sample from complex distributions. To improve the quality of generated samples and their compliance with user requirements, two commonly used methods are: (i) Alignment, which involves finetuning a diffusion model to align it with a reward; and (ii) Composition, which combines several pretrained diffusion models together, each emphasizing a desirable attribute in the generated outputs. However, trade-offs often arise when optimizing for multiple rewards or combining multiple models, as they can often represent competing properties. Existing methods cannot guarantee that the resulting model faithfully generates samples with all the desired properties. To address this gap, we propose a constrained optimization framework that unifies alignment and composition of diffusion models by enforcing that the aligned model satisfies reward constraints and/or remains close to each pretrained model. We provide a theoretical characterization of the solutions to the constrained alignment and composition problems and develop a Lagrangian-based primal-dual training algorithm to approximate these solutions. Empirically, we demonstrate our proposed approach in image generation, applying it to alignment and composition, and show that our aligned or composed model satisfies constraints effectively. Our implementation can be found at: \href{https://github.com/shervinkhalafi/constrained_comp_align}{https://github.com/shervinkhalafi/constrained\_comp\_align}

扩散模型生成模型约束优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。