用强化学习提升人像生成视频的身份一致性。
Identity-Preserving Image-to-Video Generation via Reward-Guided Optimization
- 通过奖励引导优化,在不改架构下增强身份保真度。
- 在小人脸场景下,身份相似度提升18.7%,生成更稳定。
- 适合需要高保真人像视频的应用,如数字人、虚拟主播。
图像到视频(I2V)生成近年取得显著进展,尤其在人物中心视频生成中应用广泛。然而,现有模型在输入人像与生成视频间难以保持身份一致性,尤其当人物表情变化大或动作剧烈时,且人脸仅占图像小部分时问题更严重。由于人类对身份差异极为敏感,该挑战尚未被充分解决。本文提出身份保真奖励引导优化(IPRO),一种基于强化学习的视频扩散框架,无需引入额外模块或修改模型结构,而是通过人脸身份评分器直接优化扩散模型。为提升性能并加速收敛,方法将奖励信号反向传播至采样链最后几步,实现更丰富的梯度反馈。同时提出新型面部评分机制,将真实视频中的人脸作为特征池,提供多角度信息以增强泛化能力。进一步引入KL散度正则项稳定训练,防止对奖励信号过拟合。在Wan 2.2 I2V模型及自研I2V模型上的大量实验验证了方法有效性。项目与代码已开源:https://ipro-alimama.github.io/。
原文摘要 · Abstract (English)
Recent advances in image-to-video (I2V) generation have achieved remarkable progress in synthesizing high-quality, temporally coherent videos from static images. Among all the applications of I2V, human-centric video generation includes a large portion. However, existing I2V models encounter difficulties in maintaining identity consistency between the input human image and the generated video, especially when the person in the video exhibits significant expression changes and movements. This issue becomes critical when the human face occupies merely a small fraction of the image. Since humans are highly sensitive to identity variations, this poses a critical yet under-explored challenge in I2V generation. In this paper, we propose Identity-Preserving Reward-guided Optimization (IPRO), a novel video diffusion framework based on reinforcement learning to enhance identity preservation. Instead of introducing auxiliary modules or altering model architectures, our approach introduces a direct and effective tuning algorithm that optimizes diffusion models using a face identity scorer. To improve performance and accelerate convergence, our method backpropagates the reward signal through the last steps of the sampling chain, enabling richer gradient feedback. We also propose a novel facial scoring mechanism that treats faces in ground-truth videos as facial feature pools, providing multi-angle facial information to enhance generalization. A KL-divergence regularization is further incorporated to stabilize training and prevent overfitting to the reward signal. Extensive experiments on Wan 2.2 I2V model and our in-house I2V model demonstrate the effectiveness of our method. Our project and code are available at https://ipro-alimama.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。