FinPercep-RM:面向基于强化学习的真实世界超分辨率的细粒度奖励模型与协同进化课程
FinPercep-RM: A Fine-grained Reward Model and Co-evolutionary Curriculum for RL-based Real-world Super-Resolution
提出细粒度感知奖励模型(全局分数+感知退化图)和协同进化课程学习策略,可复用于基于强化学习的图像复原任务中,缓解奖励黑客问题并提升局部真实感。
原文依据 we propose FinPercep-RM, a fine-grained perceptual reward model built on an encoder-decoder architecture that predicts both a global quality score and a Perceptual Degradation Map
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论受人类反馈强化学习(RLHF)在图像生成中取得成功的启发,近期工作通过使用图像质量评估(IQA)模型作为奖励,将基于奖励的学习应用于图像超分辨率(ISR)。然而,现有的IQA模型通常仅输出单一全局分数,对局部细粒度失真不敏感,使得感知上不理想的伪影获得虚高奖励,导致奖励黑客问题。为解决该问题,我们提出FinPercep-RM,一种基于编码器-解码器架构的细粒度感知奖励模型,可同时预测全局质量分数和用于空间定位与量化局部缺陷的感知退化图。我们进一步引入FGR-30k数据集,包含由真实世界超分辨率模型产生的多样且细微的失真,用于训练奖励模型。尽管FinPercep-RM提供了更强的监督,其增加的复杂度也使生成器策略学习不稳定。因此,我们开发了协同进化课程学习(CCL)策略,其中奖励模型与ISR模型同步进化:奖励信号的复杂度逐渐增加,而ISR模型从简单的全局监督开始以快速收敛,并逐渐过渡到细粒度奖励。这种由易到难的设计稳定了训练并抑制了奖励黑客。在多个ISR模型上的大量实验表明,全局质量和局部真实感均得到提升。代码将发布于 https://github.com/lyd-2022/FinPercep-RM。
英文原文摘要 1598 字符
Inspired by the success of Reinforcement Learning with Human Feedback (RLHF) in image generation, recent work has adapted reward-based learning to image super-resolution (ISR) by using Image Quality Assessment (IQA) models as rewards. However, existing IQA models typically output only a single global score and are insensitive to local, fine-grained distortions, allowing perceptually undesirable artifacts to obtain spuriously high rewards and leading to reward hacking. To address this issue, we propose FinPercep-RM, a fine-grained perceptual reward model built on an encoder-decoder architecture that predicts both a global quality score and a Perceptual Degradation Map for spatially localizing and quantifying local defects. We further introduce FGR-30k, a dataset containing diverse and subtle distortions produced by real-world super-resolution models, to train the reward model. While FinPercep-RM provides stronger supervision, its increased complexity also makes generator policy learning unstable. We therefore develop a Co-evolutionary Curriculum Learning (CCL) strategy, in which the reward model and the ISR model evolve synchronously: the reward signal progressively increases in complexity, while the ISR model starts with simple global supervision for fast convergence and gradually transitions to fine-grained rewards. This easy-to-hard design stabilizes training and suppresses reward hacking. Extensive experiments across multiple ISR models demonstrate improvements in both global quality and local realism. Code will be available at https://github.com/lyd-2022/FinPercep-RM.