提出直接微调蒸馏扩散模型的新方法,避免模糊和性能下降。
Tuning Timestep-Distilled Diffusion Model Using Pairwise Sample Optimization
- 通过成对样本优化,引入参考图像增强训练图像的相对似然。
- 在仅需少量推理步数下保持生成质量,支持风格迁移与概念定制。
- 适用于离线与在线偏好数据,可直接用于人类偏好的模型调优。
近年来,时间步蒸馏扩散模型实现了与多步非蒸馏模型相当的高质量图像生成,且推理步数显著减少。然而,使用原始扩散目标进行微调会导致输出质量下降和模糊。传统方法需重复蒸馏过程,计算开销巨大。本文提出成对样本优化(PSO),可直接微调任意时间步蒸馏扩散模型。PSO引入当前模型生成的参考图像,增大训练图像与参考图像之间的相对似然差距,使模型在保留少步生成能力的同时,实现输出分布的微调。该方法为通用框架,可灵活扩展至离线与在线采样的成对数据,覆盖多种扩散模型偏好优化目标。实验表明,PSO能有效适应人类偏好生成,支持风格迁移与概念定制,适用于离线与在线生成的成对偏好数据。
原文摘要 · Abstract (English)
Recent advancements in timestep-distilled diffusion models have enabled high-quality image generation that rivals non-distilled multi-step models, but with significantly fewer inference steps. While such models are attractive for applications due to the low inference cost and latency, fine-tuning them with a naive diffusion objective would result in degraded and blurry outputs. An intuitive alternative is to repeat the diffusion distillation process with a fine-tuned teacher model, which produces good results but is cumbersome and computationally intensive; the distillation training usually requires magnitude higher of training compute compared to fine-tuning for specific image styles. In this paper, we present an algorithm named pairwise sample optimization (PSO), which enables the direct fine-tuning of an arbitrary timestep-distilled diffusion model. PSO introduces additional reference images sampled from the current time-step distilled model, and increases the relative likelihood margin between the training images and reference images. This enables the model to retain its few-step generation ability, while allowing for fine-tuning of its output distribution. We also demonstrate that PSO is a generalized formulation which can be flexibly extended to both offline-sampled and online-sampled pairwise data, covering various popular objectives for diffusion model preference optimization. We evaluate PSO in both preference optimization and other fine-tuning tasks, including style transfer and concept customization. We show that PSO can directly adapt distilled models to human-preferred generation with both offline and online-generated pairwise preference image data. PSO also demonstrates effectiveness in style transfer and concept customization by directly tuning timestep-distilled diffusion models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。