跳到正文
arXiv cs.AI· Yijia Fan, Ziqi Huang, Zhongang Cai, Yan Li, Zimo Wen, Wanqi Yin, Haiwen Diao, Ziwei Liu·· 11 小时前AI 评分38

UMM-Reflection:用交错强化学习让统一模型学会原生反思

Learning Native Reflection in Unified Models with Interleaved Reinforcement Learning

AI 导读

UMM-Reflection 提出用强化学习在统一多模态模型内部训练完整反思轨迹,让反思 token 与基于流的图像修订共享轨迹级优势、跨轮次联合更新,推理时无需外部验证器。在 BAGEL 上,该方法使 GenEval 较 SFT 提升 12.05 分,并迁移至 WISE(+10.97)、OneIG-Bench(+3.48)和 T2I-CompBench++(+4.63),这三个基准均未用于训练。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:统一多模态模型既能观察图像也能渲染图像,因此原则上它们可以修复自己生成的图像:诊断图像哪里出错、修改它、观察结果,然后再次诊断。一次修改是否有帮助只有在渲染之后才能知道,因此反思文本和图像生成必须在整个循环中联合学习。对反思轨迹进行监督微调(SFT)可以提供一个冷启动,但无法找到高成功率的修复路径,而只优化渲染器或只优化一个头的朴素强化学习(RL)会留下大部分收益未被利用。我们提出 UMM-Reflection,它在单个统一模型内对完整的反思轨迹应用强化学习(RL):兄弟轨迹共享同一张初始图像,因此组相对优势可以比较不同的反思策略,而一个轨迹级优势同时更新反思 token 和基于流的修改,避免了逐轮信用分配的组合爆炸。与单轮编辑或带有外部评论器的流水线不同,信用可以跨轮次流动,并流向同一模型的两种角色,而且推理时不需要验证器。在 BAGEL 上,UMM-Reflection 相比 SFT 将 GenEval 提升了 12.05 分,且这些增益可以迁移到 WISE(+10.97)、OneIG-Bench(+3.48)和 T2I-CompBench++(+4.63),而这些基准均未用于训练。
主题: 计算机视觉与模式识别(cs.CV);人工智能(cs.AI)
引用为: arXiv:2609.35767 [cs.CV]
  (或此版本的 arXiv:2609.35767v2 [cs.CV])
  https://doi.org/10.48550/arXiv.2609.35767

由 DataCite 颁发的 arXiv DOI

提交历史

来自:Yijia Fan [查看电子邮件]
[v1] 2026年9月28日星期一 17:59:36 UTC(20,973 KB)
[v2] 2026年10月20日星期二 07:26:16 UTC(21,824 KB)

来源:arXiv cs.AI · arxiv.org