让图像生成模型学会个性化偏好,用少量例子就能匹配用户独特审美。
Personalized Preference Fine-tuning of Diffusion Models
- 用视觉语言模型提取用户偏好嵌入,通过交叉注意力融入扩散模型。
- 仅需4个示例,就能在新用户上达到76%胜率,超越Stable Cascade。
- 支持多用户偏好共优化,推理时可灵活插值,适合个性化生成场景。
基于RLHF的DPO等技术能显著提升文本到图像扩散模型的生成质量,但这些方法仅优化单一奖励函数,以对齐群体偏好,忽视了个体用户的信念或价值观差异。这种缺乏个性化限制了模型的实际应用效果。为此,我们提出PPD——一种多奖励优化目标,使扩散模型能够学习并适应个体用户偏好。该方法(1)利用视觉语言模型(VLM)从少量成对偏好样本中提取个人偏好嵌入;(2)通过交叉注意力将这些嵌入注入扩散模型。在文本到图像模型中,以用户嵌入为条件,使用DPO目标进行微调,同时优化多个用户的偏好对齐。实验证明,本方法能有效优化多个奖励函数,并在推理阶段实现偏好插值。在真实用户场景中,仅需4个新用户偏好示例,平均胜率可达76%,生成结果更准确反映特定用户偏好。
原文摘要 · Abstract (English)
RLHF techniques like DPO can significantly improve the generation quality of text-to-image diffusion models. However, these methods optimize for a single reward that aligns model generation with population-level preferences, neglecting the nuances of individual users' beliefs or values. This lack of personalization limits the efficacy of these models. To bridge this gap, we introduce PPD, a multi-reward optimization objective that aligns diffusion models with personalized preferences. With PPD, a diffusion model learns the individual preferences of a population of users in a few-shot way, enabling generalization to unseen users. Specifically, our approach (1) leverages a vision-language model (VLM) to extract personal preference embeddings from a small set of pairwise preference examples, and then (2) incorporates the embeddings into diffusion models through cross attention. Conditioning on user embeddings, the text-to-image models are fine-tuned with the DPO objective, simultaneously optimizing for alignment with the preferences of multiple users. Empirical results demonstrate that our method effectively optimizes for multiple reward functions and can interpolate between them during inference. In real-world user scenarios, with as few as four preference examples from a new user, our approach achieves an average win rate of 76\% over Stable Cascade, generating images that more accurately reflect specific user preferences.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。