混合ODE-SDE采样提升流匹配模型效率,显著缩短训练时间。
MixGRPO: Unlocking Flow-based GRPO Efficiency with Mixed ODE-SDE
- 用滑动窗口结合SDE与ODE采样,只在窗口内优化梯度。
- 相比DanceGRPO减少近50%训练时间,Flash版再减71%。
- 适合追求高效图像生成偏好对齐的研究者与开发者。
尽管GRPO显著提升了图像生成中人类偏好对齐的流匹配模型表现,但如FlowGRPO和DanceGRPO等方法仍因需对马尔可夫决策过程(MDP)中所有去噪步骤进行采样与优化而效率低下。本文提出新颖框架MixGRPO,通过融合随机微分方程(SDE)与常微分方程(ODE)的混合采样策略,简化了MDP内的优化流程,提升效率并增强性能。具体地,MixGRPO引入滑动窗口机制:仅在窗口内使用SDE采样并进行GRPO引导优化,窗口外采用ODE采样。该设计将采样随机性限制在窗口内时间步,降低优化开销,使梯度更新更聚焦,加速收敛。此外,由于窗口外时间步不参与优化,可支持高阶求解器以实现更快采样。因此,我们提出了更高效的变体MixGRPO-Flash,进一步提升训练效率,同时保持相当性能。MixGRPO在多个维度的人类偏好对齐上均有显著提升,相比DanceGRPO在有效性和效率上均占优,训练时间降低近50%;特别地,MixGRPO-Flash再减少71%训练时间。
原文摘要 · Abstract (English)
Although GRPO substantially enhances flow matching models in human preference alignment of image generation, methods such as FlowGRPO and DanceGRPO still exhibit inefficiency due to the necessity of sampling and optimizing over all denoising steps specified by the Markov Decision Process (MDP). In this paper, we propose $\textbf{MixGRPO}$, a novel framework that leverages the flexibility of mixed sampling strategies through the integration of stochastic differential equations (SDE) and ordinary differential equations (ODE). This streamlines the optimization process within the MDP to improve efficiency and boost performance. Specifically, MixGRPO introduces a sliding window mechanism, using SDE sampling and GRPO-guided optimization only within the window, while applying ODE sampling outside. This design confines sampling randomness to the time-steps within the window, thereby reducing the optimization overhead, and allowing for more focused gradient updates to accelerate convergence. Additionally, as time-steps beyond the sliding window are not involved in optimization, higher-order solvers are supported for faster sampling. So we present a faster variant, termed $\textbf{MixGRPO-Flash}$, which further improves training efficiency while achieving comparable performance. MixGRPO exhibits substantial gains across multiple dimensions of human preference alignment, outperforming DanceGRPO in both effectiveness and efficiency, with nearly 50% lower training time. Notably, MixGRPO-Flash further reduces training time by 71%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。