通过采样时重排去噪时间,一次训练即可模拟不同正则强度下的扩散模型效果。
DeRaDiff: Denoising Time Realignment of Diffusion Models
- 采样阶段动态调整正则化强度,用几何混合分布替代逆过程参考分布。
- 仅需一次对齐训练,即可在推理时近似多种正则强度的模型表现。
- 适用于追求高效调参、避免昂贵超参数搜索的研究者和开发者。
近期研究致力于将扩散模型与人类偏好对齐,以提升美学质量并减少伪影与偏见。这类方法旨在最大化与高奖励对齐的条件输出分布,同时不远离预训练先验,通常通过KL(Kullback-Leibler)正则化实现。然而核心问题仍存:如何选择合适的正则化强度?过强导致对齐不足,过弱则引发“奖励劫持”。现有方法需在多个正则强度下分别对齐预训练模型,再择优,但代价高昂。本文提出DeRaDiff,一种去噪时间重排方法:在一次对齐后,通过采样时调节正则化强度,无需额外训练即可模拟不同强度下的模型表现。该方法扩展了语言模型中的解码时重排至扩散模型,通过对连续潜在变量的迭代预测,用对齐后验与参考后验的几何混合替换逆向步骤的参考分布,在常见调度器下可导出闭式更新,仅需一个可调参数λ实现即时控制。实验表明,无论在文本图像对齐或图像质量指标上,本方法均能稳定逼近从头训练于不同正则强度下的模型表现,为最优正则强度搜索提供高效方案,显著降低计算成本。
原文摘要 · Abstract (English)
Recent advances align diffusion models with human preferences to increase aesthetic appeal and mitigate artifacts and biases. Such methods aim to maximize a conditional output distribution aligned with higher rewards whilst not drifting far from a pretrained prior. This is commonly enforced by KL (Kullback Leibler) regularization. As such, a central issue still remains: how does one choose the right regularization strength? Too high of a strength leads to limited alignment and too low of a strength leads to "reward hacking". This renders the task of choosing the correct regularization strength highly non-trivial. Existing approaches sweep over this hyperparameter by aligning a pretrained model at multiple regularization strengths and then choose the best strength. Unfortunately, this is prohibitively expensive. We introduce DeRaDiff, a denoising time realignment procedure that, after aligning a pretrained model once, modulates the regularization strength during sampling to emulate models trained at other regularization strengths without any additional training or finetuning. Extending decoding-time realignment from language to diffusion models, DeRaDiff operates over iterative predictions of continuous latents by replacing the reverse step reference distribution by a geometric mixture of an aligned and reference posterior, thus giving rise to a closed form update under common schedulers and a single tunable parameter, lambda, for on the fly control. Our experiments show that across multiple text image alignment and image-quality metrics, our method consistently provides a strong approximation for models aligned entirely from scratch at different regularization strengths. Thus, our method yields an efficient way to search for the optimal strength, eliminating the need for expensive alignment sweeps and thereby substantially reducing computational costs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。