arXiv:2411.17310cs.CVcs.LG2024-11被引 1

让文生图模型逐步学会新评分标准,避免遗忘旧能力。

Reward Incremental Learning in Text-to-Image Generation

  • 提出增量奖励学习框架,支持持续新增生成目标
  • 在多阶段任务中保持图像质量和审美一致性
  • 通过轻量级知识蒸馏缓解扩散模型的灾难性遗忘

去噪扩散模型在文生图领域取得显著进展。尽管大规模预训练模型在通用图像生成上表现优异,但下游任务常需微调以满足特定标准(如美学或人类偏好)。基于奖励梯度的方法虽有前景,但现有方法仅适用于单一奖励任务,难以应对现实场景中随时间逐步引入多目标的需求。本文首次定义这一更贴近实际、尚未被探索的问题——奖励增量学习(RIL),即模型需逐步适应多个下游目标。同时,我们发现扩散模型微调过程中存在独特的灾难性遗忘现象,影响图像质量与视觉结构。为此,提出奖励增量蒸馏(RID)方法,以极小计算开销缓解遗忘,实现多任务序列中的稳定生成性能。实验表明,RID在RIL场景下能持续生成高质量图像。代码将在论文接收后公开。

原文摘要 · Abstract (English)

The recent success of denoising diffusion models has significantly advanced text-to-image generation. While these large-scale pretrained models show excellent performance in general image synthesis, downstream objectives often require fine-tuning to meet specific criteria such as aesthetics or human preference. Reward gradient-based strategies are promising in this context, yet existing methods are limited to single-reward tasks, restricting their applicability in real-world scenarios that demand adapting to multiple objectives introduced incrementally over time. In this paper, we first define this more realistic and unexplored problem, termed Reward Incremental Learning (RIL), where models are desired to adapt to multiple downstream objectives incrementally. Additionally, while the models adapt to the ever-emerging new objectives, we observe a unique form of catastrophic forgetting in diffusion model fine-tuning, affecting both metric-wise and visual structure-wise image quality. To address this catastrophic forgetting challenge, we propose Reward Incremental Distillation (RID), a method that mitigates forgetting with minimal computational overhead, enabling stable performance across sequential reward tasks. The experimental results demonstrate the efficacy of RID in achieving consistent, high-quality generation in RIL scenarios. The source code of our work will be publicly available upon acceptance.

文生图增量学习扩散模型奖励学习

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