用奖励模型训练更优负向嵌入,提升文生图质量。
ReNeg: Learning Negative Embedding with Reward Guidance
- 通过奖励反馈学习框架,端到端优化负向嵌入。
- 相比零文本和人工设计的负向提示,人类偏好匹配度显著提升。
- 学得的负向嵌入可跨模型通用,支持文生图与文生视频任务。
在文本生成图像(T2I)应用中,负向嵌入已被证明是提升生成质量的简单而有效方法。传统方法依赖用户定义的负向提示,虽可工作但未必最优。本文提出ReNeg,一种基于奖励模型引导的端到端负向嵌入学习方法。我们采用奖励反馈学习框架,并将无分类器引导(CFG)引入训练过程,此前该技术仅用于推理阶段,现可实现负向嵌入的有效学习。我们还提出两种策略,分别学习全局与样本级负向嵌入。大量实验表明,所学负向嵌入显著优于零文本和手工设计的版本,在人类偏好对齐上取得明显进步。此外,同一文本嵌入空间内学习的负向嵌入具备强泛化能力:例如,使用CLIP文本编码器在SD1.5上学习的负向嵌入,可无缝迁移至ControlNet、ZeroScope、VideoCrafter2等文生图及文生视频模型,带来一致的性能提升。
原文摘要 · Abstract (English)
In text-to-image (T2I) generation applications, negative embeddings have proven to be a simple yet effective approach for enhancing generation quality. Typically, these negative embeddings are derived from user-defined negative prompts, which, while being functional, are not necessarily optimal. In this paper, we introduce ReNeg, an end-to-end method designed to learn improved Negative embeddings guided by a Reward model. We employ a reward feedback learning framework and integrate classifier-free guidance (CFG) into the training process, which was previously utilized only during inference, thus enabling the effective learning of negative embeddings. We also propose two strategies for learning both global and per-sample negative embeddings. Extensive experiments show that the learned negative embedding significantly outperforms null-text and handcrafted counterparts, achieving substantial improvements in human preference alignment. Additionally, the negative embedding learned within the same text embedding space exhibits strong generalization capabilities. For example, using the same CLIP text encoder, the negative embedding learned on SD1.5 can be seamlessly transferred to text-to-image or even text-to-video models such as ControlNet, ZeroScope, and VideoCrafter2, resulting in consistent performance improvements across the board.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。