用强化学习提升少步文本生成多视角图像的质量与一致性
Refining Few-Step Text-to-Multiview Diffusion via Reinforcement Learning
- 设计联合视图奖励的马尔可夫决策模型,协调多视角生成
- 通过自精炼采样提升学习信号,实现更有效的强化学习微调
- 自适应双优化框架平衡单视角与整体一致性,适合实时多视角生成
文本到多视角(T2MV)扩散模型能从单一文本提示生成场景的多个视角。尽管少步骨干网络支持实时生成,但常牺牲生成质量,如单视角保真度和跨视角一致性。强化学习(RL)微调有潜力解决此问题,但现有针对单图扩散的方法难以直接适用于少步T2MV场景,因其忽略跨视角协同,且在少步条件下学习信号较弱。为此,我们提出MVC-ZigAL,一种专为少步T2MV扩散模型设计的强化学习微调框架。核心包括:(1) 新的马尔可夫决策过程(MDP)形式化,联合建模所有生成视角,并通过联合视图奖励评估其整体质量;(2) 创新的优势学习策略,利用自精炼采样相比标准采样的性能提升,获得更强的学习信号;(3) 统一的强化学习框架,结合拉格朗日对偶形式进行多视角约束优化,通过自适应原-对偶更新与自适应课程阈值,协调探索与约束执行。该设计实现稳健均衡的少步T2MV扩散模型强化学习微调,在单视角保真度和跨视角一致性上均有显著提升。代码已公开于https://github.com/ZiyiZhang27/MVC-ZigAL。
原文摘要 · Abstract (English)
Text-to-multiview (T2MV) diffusion models have shown great promise in generating multiple views of a scene from a single text prompt. While few-step backbones enable real-time T2MV generation, they often compromise key aspects of generation quality, such as per-view fidelity and cross-view consistency. Reinforcement learning (RL) finetuning offers a potential solution, yet existing approaches designed for single-image diffusion do not readily extend to the few-step T2MV setting, as they neglect cross-view coordination and suffer from weak learning signals in few-step regimes. To address this, we propose MVC-ZigAL, a tailored RL finetuning framework for few-step T2MV diffusion models. Specifically, its core insights are: (1) a new MDP formulation that jointly models all generated views and assesses their collective quality via a joint-view reward; (2) a novel advantage learning strategy that exploits the performance gains of a self-refinement sampling scheme over standard sampling, yielding stronger learning signals for effective RL finetuning; and (3) a unified RL framework that extends advantage learning with a Lagrangian dual formulation for multiview-constrained optimization, balancing single-view and joint-view objectives through adaptive primal-dual updates under a self-paced threshold curriculum that harmonizes exploration and constraint enforcement. Collectively, these designs enable robust and balanced RL finetuning for few-step T2MV diffusion models, yielding substantial gains in both per-view fidelity and cross-view consistency. Code is available at https://github.com/ZiyiZhang27/MVC-ZigAL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。