论文精选

λ-Controlled GRPO:用路径方差预算解决流匹配模型RL训练不稳定

$λ$-Controlled GRPO: Turning Flow-Matching Ratio Instability into a Budgeted Resource

精选理由

图像生成做 RL 训练后期老崩?这篇把不稳归到一个能算的量"路径方差",按预算分配梯度,不用再手调补丁。

Flow-GRPO 把去噪采样器当作可由奖励反馈优化的随机策略,但多步去噪中重要性比率漂移到 1 以下、各步裁剪率不一、训练后期可用样本减少。论文把这些症状归因于单个量——路径方差,它由采样器的高斯转移核精确决定,训练中可低成本估计。λ-Controlled GRPO 依据这条预测规律校准重要性比率,并按各去噪步的预测成本分配梯度,两个控制尺度由标准策略选择固定,不新增调参项。在文本到图像模型的两种奖励设定下——OCR 打分的难目标文本渲染与偏好模型打分的人类偏好——该方法在文本准确率和偏好奖励上都超过最强经验稳定器,并把后期去噪步的路径方差控制在预算内,而基线在这一阶段系统性超支。

原文 · arXiv cs.LG

$λ$-Controlled GRPO: Turning Flow-Matching Ratio Instability into a Budgeted Resource

Reinforcement learning is increasingly used to align image generators with reward signals, and Flow-GRPO recently extended this paradigm to flow-matching models by treating the denoising sampler as a stochastic policy that can be optimized from reward feedback. Training in this setting is unstable in a way specific to multi-step denoising: the policy update changes systematically across denoising steps, with importance ratios drifting below one, becoming increasingly dispersed, clipping at different rates, and leaving fewer usable samples late in training. Prior work treats these effects as separate failure modes and addresses each with a hand-tuned stabilizer. We show instead that they arise from a single per-step quantity, which we call path variance. This quantity is determined exactly by the sampler's Gaussian transition kernel and can be estimated cheaply during training. This reframes instability as a resource that can be measured and budgeted rather than a collection of symptoms to repair. Our method, $λ$-Controlled GRPO, calibrates importance-ratio behavior from this predicted law rather than from noisy empirical statistics, and allocates gradient effort across denoising steps according to their predicted cost. The two scales governing the update are fixed by standard policy choices rather than introduced as free tuning parameters. On a text-to-image model under two reward settings, rendering difficult target text scored by optical character recognition and matching human preferences scored by a preference model, $λ$-Controlled GRPO improves both text accuracy and preference reward over the strongest empirical stabilizer. It also keeps late-step path variance within its intended budget, precisely where the baseline systematically overshoots. The result is a Flow-GRPO update calibrated by its own transition law rather than stabilized after instability appears.