arXiv:2601.03213cs.LG2026-01被引 1

用强化学习让文生图模型精准删除特定概念,效果更好且图像质量不降。

Critic-Guided Reinforcement Unlearning in Text-to-Image Diffusion

  • 将去噪过程视为序列决策,每步都用批判器评估动作好坏
  • 在噪声潜空间上训练奖励预测器,实现更稳定有效的梯度更新
  • 适合需要精确擦除特定内容又不破坏整体生成能力的研究者

文生图扩散模型中的机器遗忘旨在移除特定概念的同时保持整体可用性。以往方法多依赖监督权重修改或全局惩罚;强化学习方法虽灵活,但常依赖稀疏的终局奖励,导致更新方差大、信用分配弱。本文提出一种通用的强化学习框架,将去噪建模为序列决策过程,并引入带噪声步骤奖励的时序感知批判器。具体而言,在噪声潜空间上训练基于CLIP的奖励预测器,利用其每步信号计算策略梯度更新的优势估计,优化反向扩散核。该方法实现简单,支持离线重用,可嵌入标准文生图骨干网络。在多个概念上,本方法遗忘效果优于或相当于强基线,同时保持图像质量和无害提示保真度;消融实验表明:(i) 每步批判器和 (ii) 噪声条件奖励是稳定性和有效性的关键。代码与评估脚本已公开,以促进可复现性和未来研究。

原文摘要 · Abstract (English)

Machine unlearning in text-to-image diffusion models aims to remove targeted concepts while preserving overall utility. Prior diffusion unlearning methods typically rely on supervised weight edits or global penalties; reinforcement-learning (RL) approaches, while flexible, often optimize sparse end-of-trajectory rewards, yielding high-variance updates and weak credit assignment. We present a general RL framework for diffusion unlearning that treats denoising as a sequential decision process and introduces a timestep-aware critic with noisy-step rewards. Concretely, we train a CLIP-based reward predictor on noisy latents and use its per-step signal to compute advantage estimates for policy-gradient updates of the reverse diffusion kernel. Our algorithm is simple to implement, supports off-policy reuse, and plugs into standard text-to-image backbones. Across multiple concepts, the method achieves better or comparable forgetting to strong baselines while maintaining image quality and benign prompt fidelity; ablations show that (i) per-step critics and (ii) noisy-conditioned rewards are key to stability and effectiveness. We release code and evaluation scripts to facilitate reproducibility and future research on RL-based diffusion unlearning.

文生图强化学习遗忘学习扩散模型

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