REVIEW 3 major objections 6 minor 24 references
Matching student and reference trajectories from shared on-policy states beats KL distillation for flow models.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · grok-4.5
2026-07-31 12:27 UTC pith:22VFYA3T
load-bearing objection Real win is unweighted on-policy velocity match vs SDE-KL weighting; the trajectory-bound story is mostly unused at the headline K=1 result. the 3 major comments →
FlowCTS: On-policy Continuous Trajectory Supervision of Flow Models
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
On-policy distillation for flow models is more effective when formulated as continuous trajectory supervision from shared student-visited states than as reverse-KL on auxiliary SDE transition kernels. The trajectory discrepancy admits a temporally weighted velocity-matching upper bound whose Euler discretization yields a family of objectives parameterized by supervision horizon K; K=1 already outperforms vanilla KL-OPD with faster convergence on GenEval, OCR, and PickScore, while longer K trades richer path information against harder optimization.
What carries the argument
Flow Continuous Trajectory Supervision (FlowCTS): the integrated squared gap between student and reference trajectories started from the same state, bounded by a trajectory-derived weighted velocity MSE (the velocity upper bound) and discretized into a K-step weighted velocity-matching loss.
Load-bearing premise
Optimizing the Cauchy–Schwarz velocity upper bound and its short Euler discretization is assumed to be a faithful enough stand-in for true path matching, rather than gains coming mainly from plain on-policy velocity regression without that bound.
What would settle it
A controlled ablation that replaces the trajectory-derived weights with uniform or late-heavy weights (or plain unweighted velocity MSE at the same on-policy states) and checks whether the reported gains over KL-OPD and the early-timestep advantage disappear.
If this is right
- Dense OPD for continuous generators should target early denoising stages, where student–reference velocity gaps and structural decisions are largest.
- Supervision horizon K becomes an explicit scaling knob for flow post-training, balancing path information against optimization difficulty.
- The same trajectory objective applies off-policy: multi-step FlowCTS-SFT can beat single-state velocity SFT, especially on text rendering.
- Multi-reference capability consolidation can be done by routing student states to task experts under one trajectory loss instead of mixed sparse rewards.
Where Pith is reading between the lines
- If early-path supervision is the real driver, any OPD or RL method whose effective weight peaks late (SDE kernels, terminal rewards) is systematically misallocated for flow and diffusion models.
- Concurrent direct velocity matching at student states may capture much of the K=1 gain; the distinctive claim to pressure-test is whether the derived multi-step weights and path rollouts add value beyond local field regression.
- Treating short trajectory segments as the supervision atom could transfer to video or other long-horizon continuous generators where exposure bias compounds over many steps.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes FlowCTS, an on-policy distillation objective for flow models that supervises the discrepancy between student and reference ODE trajectories launched from a shared student-visited state. Starting from the integral identity between trajectory and velocity fields, the authors derive a Cauchy–Schwarz velocity upper bound (VUB, Eq. 8) with a trajectory-derived weight w(r), and discretize it over K Euler steps into a weighted velocity-matching objective with coefficients α_i = [K(K+1)−i(i+1)]/2 (Eq. 10, App. B). In a multi-reference setup on SD3.5-Medium with task-specific Flow-GRPO experts as references, K=1 FlowCTS-OPD reportedly improves over a transition-kernel KL baseline (Vanilla OPD, following Flow-OPD) on GenEval (0.90→0.93), OCR (0.90→0.92), and PickScore (22.75→23.06), with faster convergence (Fig. 3) and better OOD compositional and preference metrics (Tables 2–3). The authors diagnose the KL baseline's weakness as a temporal supervision mismatch: the SDE-induced weight w_KL(t) concentrates supervision late in denoising while student–reference velocity gaps peak early (Fig. 4). An off-policy SFT variant and a study of supervision horizon K are also reported.
Significance. If the results hold, the paper makes a useful contribution to an active area: a controlled multi-reference OPD comparison in which prompt mixture, rollouts, reference routing, and optimization are held fixed between objectives (§4.1, App. C.1), a clean derivation of a trajectory-discrepancy upper bound with trajectory-derived weights (Apps. A–B), a training-budget control for the K=2 comparison (App. E), and a temporal-misallocation analysis of KL-based Flow-OPD (§4.4, Fig. 4) that is consistent with, and adds evidence to, the TempFlow-GRPO timing story. The off-policy extension and OOD evaluations on T2I-CompBench and DrawBench broaden the evidence. However, the significance is currently capped by two factors: the objective at the headline operating point (K=1) coincides with the concurrent DanceOPD field-regression objective, so the distinctive "trajectory" machinery is not what delivers the reported gains; and the multi-step variants that do exercise the machinery give non-monotonic or negative effects (PickScore 22.60 at K=3). The honest, still-valuable version of this paper is "unweighted velocity regression at student-visited states beats KL-weighted OPD, and here is a bound/
major comments (3)
- [§3.3, Eq. (10); §4.2, Table 1; §4.4, Eq. (11)] The headline comparison is at K=1, where α_0=1 and Eq. (10) is exactly unweighted mean-squared velocity matching between student and reference at a single student-visited state — by the authors' own Eq. (11), the only difference from Vanilla OPD is the removal of the SDE-induced weight w_KL(t). The VUB derivation (Eq. 8, App. A) and the α-weight profile (Eq. 33, App. B) therefore do no work in the headline experiment, and the K=2,3 rows that would exercise the trajectory machinery show inconsistent or negative effects (PickScore 23.06→22.60 from K=1 to K=3; OCR flat). The abstract and §1 attribute the gains to the 'trajectory-derived velocity-matching objective,' but no experiment separates that attribution from the simpler hypothesis 'uniform velocity regression beats KL-weighted velocity regression.' The clean counterfactual — KL-weighted vs. unweighted velocity regression at identical
- [§4.2, Table 1; §4.5, Tables 2–3] All headline deltas (GenEval 0.90→0.93, OCR 0.90→0.92, PickScore 22.75→23.06) are reported from what appears to be a single training run per configuration, with no seed variance, confidence intervals, or significance testing. GenEval and OCR are known to have non-trivial evaluation variance, and a 0.02–0.03 gap is within plausible run-to-run fluctuation for LoRA fine-tuning. The same concern applies to Table 3 (e.g., ImageReward 1.21→1.27) and is acute in Table 2, where Vanilla OPD actually wins Color (0.8350 vs. 0.8295), Texture, and Non-Spatial, undercutting the claim of consistent OOD improvement. At minimum, the main Table 1 comparison and Fig. 3 curves should be replicated over ≥3 seeds (or the evaluation variance quantified by repeated decoding of fixed checkpoints), and claims of 'consistent' improvement should be restricted to metrics where the gap exceeds measured noise.
- [§3.2, Eq. (8); Appendix A, Eq. (17)] The VUB is obtained by one application of Cauchy–Schwarz followed by an integral reordering (App. A, Eq. 17); its tightness depends on the velocity mismatch being roughly aligned along the segment, which fails as the student and reference paths diverge. The paper never quantifies how loose L_VUB is relative to L_CTS, nor does it establish that minimizing the bound improves the true trajectory objective. This matters most exactly where the bound is stressed — K>1 — and the non-monotonic K results in Table 1 are consistent with the bound becoming uninformative as paths diverge. A diagnostic (e.g., reporting the empirical ratio L_CTS/L_VUB or the trajectory discrepancy at the end of training for K=1,2,3) would either substantiate or honestly bound the mechanism claim. Relatedly, the discrete derivation (App. B) applies Cauchy–Schwarz per step and then discards the h^3 factor; a brief note o
minor comments (6)
- [Abstract] Abstract/§1: 'Beyond on-policy setting,FlowCTS' is missing a space; similar spacing artifacts ('from0.90to0.93', 'Vanilla SFT ,') appear throughout, likely from PDF extraction but worth fixing in source.
- [§4.2, Table 1 (Part II)] The off-policy gains over Vanilla SFT are marginal and partly negative: at K=2 PickScore drops 21.68→21.66 and GenEval moves only 0.71→0.72; the 'consistently outperforms vanilla SFT' claim in the abstract rests almost entirely on OCR (+0.03–0.05). The abstract should reflect this metric-selective pattern.
- [Appendix E, Table 4] Table 4 uses a 'reduced training configuration' and its K=2 numbers (GenEval 0.93, PickScore 22.80) do not match the Table 1 K=2 row (0.94, 22.96). Please state explicitly what differs between the two configurations and caution that Table 4 is not comparable to Table 1.
- [§1, contributions list] The phrase 'a new scaling law dimension' (§1, contributions) overstates what is shown; three values of K with non-monotonic outcomes is a hyperparameter study, not a scaling law. Suggest 'a new design dimension' or similar.
- [§4.4, Fig. 4] Fig. 4 would be more convincing with the w_KL(t) weight itself plotted alongside panels (a) and (b), so the reader can see directly that the concentration of the KL loss is weight-driven rather than loss-driven. Also specify over how many states/prompts the curves are averaged.
- [§4.1; Appendix C.1] §4.1/Appendix C: the timestep fraction 0.99 yielding 'nine denoising transitions' and the 1:3:1 prompt-cycle ratio are load-bearing hyperparameters with no sensitivity analysis; a brief ablation or at least a justification for these choices would help reproducibility. Code/checkpoint release plans are not stated.
Circularity Check
No significant circularity: trajectory bound and weights are derived from ODE error accumulation, and empirical gains are measured against external baselines under a controlled shared pipeline.
full rationale
The paper’s load-bearing derivation is self-contained and does not reduce to its inputs by construction. L_CTS is defined as the integrated squared trajectory discrepancy from a shared state (Eq. 5); the velocity upper bound (Eq. 8 / App. A) follows from the integral relation between states and velocities plus Cauchy–Schwarz, yielding trajectory-derived weights w(r) and discrete α_i that are not fitted to GenEval/OCR/PickScore. Discretization (Eq. 10 / App. B) is a standard Euler reduction of that bound. At K=1 the objective collapses to unweighted velocity MSE—which the paper states explicitly—but that is a special case of the derivation, not a circular fit or a self-definitional loop. Empirical claims compare FlowCTS-OPD to Vanilla OPD (and RL/SFT baselines) under identical prompts, rollouts, references, and hyperparameters on held evaluation protocols; references are external Flow-GRPO checkpoints, not author-fitted uniqueness theorems. Concurrent DanceOPD and the incomplete isolation of “bound/weights vs plain MSE” are novelty/ablation concerns, not circularity. No self-citation is load-bearing for the central math or the metric lifts. Score 0.
Axiom & Free-Parameter Ledger
free parameters (4)
- supervision steps K =
K=1 primary; K=2 best overall on-policy in Table 1
- trajectory segment endpoints (t_e, t_s) / step fraction 0.99 =
timestep fraction 0.99; 10 sampling steps
- prompt mixture cycle OCR:GenEval:PickScore =
1:3:1
- LoRA rank/alpha, LR, EMA decay =
rank 32, alpha 64, LR 1e-4, EMA 0.9
axioms (4)
- domain assumption Student and reference states obey the same reverse-time ODE/Euler integration from a shared x_ts, so path difference equals integrated velocity mismatch (Eqs. 6–7).
- ad hoc to paper Cauchy–Schwarz bound on the time integral yields a valid training surrogate (L_CTS ≤ L_VUB) whose minimizers transfer to better generations.
- domain assumption SDE student rollouts (Flow-GRPO style) define the correct on-policy state distribution for distillation comparisons.
- domain assumption Task-specific Flow-GRPO checkpoints are appropriate fixed references for multi-reference OPD without extra teachers.
invented entities (2)
-
FlowCTS / L_CTS continuous trajectory supervision objective
no independent evidence
-
Trajectory-derived velocity weights w(r) and discrete α_i
independent evidence
read the original abstract
While on-policy distillation (OPD) effectively addresses sparse rewards and exposure bias in large language model post-training, its extension to flow models remains underexplored. To this end, we propose Flow Continuous Trajectory Supervision (FlowCTS), which matches subsequent student and reference trajectories initialized from the same student-visited state. Using the integral relation between trajectories and velocity fields, we derive a temporally weighted velocity-matching upper bound and discretize it into practical objectives parameterized by the number of supervision steps. Under a multi-reference setup, single-state FlowCTS-OPD outperforms vanilla KL-based OPD with faster convergence. FlowCTS-OPD improves GenEval from 0.90 to 0.93, OCR from 0.90 to 0.92, and PickScore from 22.75 to 23.06, while outperforming a mixed-reward RL baseline across all target metrics. Further analysis reveals a clear temporal supervision mismatch in vanilla KL-based OPD arising from its auxiliary SDE transition kernels. Beyond on-policy setting,FlowCTS also consistently outperforms vanilla SFT , particularly on OCR, while increasing supervision steps exhibit a trade-off between richer trajectory information and greater optimization difficulty.
Figures
Reference graph
Works this paper leans on
-
[1]
On-policy distillation of language models: Learning from self- generated mistakes
Rishabh Agarwal, Nino Vieillard, Yongchao Zhou, Piotr Stanczyk, Sabela Ramos Garea, Matthieu Geist, and Olivier Bachem. On-policy distillation of language models: Learning from self- generated mistakes. InInternational Conference on Learning Representations, volume 2024, pp. 21246–21263,
2024
-
[2]
Models marked with‡in Table 3 are evaluated at1024×1024
Method Training steps GenEval↑OCR↑PickScore↑ FLOWCTS-OPD (K= 1)N0.91 0.89 22.93 FLOWCTS-OPD (K= 1)2N0.92 0.91 23.01 FLOWCTS-OPD (K= 2)N0.93 0.91 22.80 prompts, generation settings, and evaluator implementations. Models marked with‡in Table 3 are evaluated at1024×1024. Following Flow-GRPO (Liu et al., 2026a), we evaluate GenEval, OCR, and PickScore on thei...
2023
-
[3]
Nicholas M Boffi, Michael S Albergo, and Eric Vanden-Eijnden. Flow map matching with stochastic interpolants: A mathematical framework for consistency models.arXiv preprint arXiv:2406.07507,
-
[4]
Directly fine-tuning diffusion models on differentiable rewards
Kevin Clark, Paul Vicol, Kevin Swersky, and David Fleet. Directly fine-tuning diffusion models on differentiable rewards. InInternational Conference on Learning Representations, volume 2024, pp. 4793–4822,
2024
-
[7]
Minillm: Knowledge distillation of large language models
Yuxian Gu, Li Dong, Furu Wei, and Minlie Huang. Minillm: Knowledge distillation of large language models. InInternational Conference on Learning Representations, volume 2024, pp. 32694–32717,
2024
-
[8]
Tempflow-grpo: When timing matters for grpo in flow models.arXiv preprint arXiv:2508.04324,
Xiaoxuan He, Siming Fu, Yuke Zhao, Wanli Li, Jian Yang, Dacheng Yin, Fengyun Rao, and Bo Zhang. Tempflow-grpo: When timing matters for grpo in flow models.arXiv preprint arXiv:2508.04324,
-
[10]
Consistency trajectory models: Learning proba- bility flow ode trajectory of diffusion
10 Preprint Dongjun Kim, Chieh-Hsin Lai, WeiHsiang Liao, Naoki Murata, Yuhta Takida, Toshimitsu Uesaka, Yutong He, Yuki Mitsufuji, and Stefano Ermon. Consistency trajectory models: Learning proba- bility flow ode trajectory of diffusion. InInternational Conference on Learning Representations, volume 2024, pp. 44493–44525,
2024
-
[11]
Nikita Kornilov, Petr Mokrov, Alexander Gasnikov, and Alexander Korotin. Optimal flow matching: Learning straight trajectories in just one step.arXiv preprint arXiv:2403.13117,
-
[12]
Kat-coder-v2 technical report.arXiv preprint arXiv:2603.27703, 2026a
Fengxiang Li, Han Zhang, Haoyang Huang, Jinghui Wang, Jinhua Hao, Kun Yuan, Mengtong Li, Minglei Zhang, Pengcheng Xu, Wenhao Zhuang, et al. Kat-coder-v2 technical report.arXiv preprint arXiv:2603.27703, 2026a. Quanhao Li, Junqiu Yu, Kaixun Jiang, Yujie Wei, Zhen Xing, Pandeng Li, Ruihang Chu, Shiwei Zhang, Yu Liu, and Zuxuan Wu. Diffusionopd: A unified pe...
-
[14]
https://thinkingmachines.ai/blog/on-policy-distillation
doi: 10.64434/tml.20251026. https://thinkingmachines.ai/blog/on-policy-distillation. Wenhan Ma, Jianyu Wei, Liang Zhao, Hailin Zhang, Bangjun Xiao, Lei Li, Qibin Yang, Bofei Gao, Yudong Wang, Rang Li, et al. Mopd: Multi-teacher on-policy distillation for capability integration in llm post-training.arXiv preprint arXiv:2606.30406, 2026a. Xu Ma, Yitian Zhan...
-
[15]
Progressive distillation for fast sampling of diffusion models.arXiv preprint arXiv:2202.00512,
Tim Salimans and Jonathan Ho. Progressive distillation for fast sampling of diffusion models.arXiv preprint arXiv:2202.00512,
-
[16]
Alexander Tong, Kilian Fatras, Nikolay Malkin, Guillaume Huguet, Yanlei Zhang, Jarrid Rector- Brooks, Guy Wolf, and Yoshua Bengio. Improving and generalizing flow-based generative models with minibatch optimal transport.arXiv preprint arXiv:2302.00482,
-
[17]
Mimo-v2-flash technical report.arXiv preprint arXiv:2601.02780,
Bangjun Xiao, Bingquan Xia, Bo Yang, Bofei Gao, Bowen Shen, Chen Zhang, Chenhong He, Chiheng Lou, Fuli Luo, Gang Wang, et al. Mimo-v2-flash technical report.arXiv preprint arXiv:2601.02780,
-
[18]
Dancegrpo: Unleashing grpo on visual generation.arXiv preprint arXiv:2505.07818,
Zeyue Xue, Jie Wu, Yu Gao, Fangyuan Kong, Lingting Zhu, Mengzhao Chen, Zhiheng Liu, Wei Liu, Qiushan Guo, Weilin Huang, et al. Dancegrpo: Unleashing grpo on visual generation.arXiv preprint arXiv:2505.07818,
-
[19]
Qwen3 technical report.arXiv preprint arXiv:2505.09388,
An Yang, Anfeng Li, Baosong Yang, Beichen Zhang, Binyuan Hui, Bo Zheng, Bowen Yu, Chang Gao, Chengen Huang, Chenxu Lv, et al. Qwen3 technical report.arXiv preprint arXiv:2505.09388,
-
[20]
Wenkai Yang, Weijie Liu, Ruobing Xie, Kai Yang, Saiyong Yang, and Yankai Lin. Learning beyond teacher: Generalized on-policy distillation with reward extrapolation.arXiv preprint arXiv:2602.12125,
-
[21]
Tianwei Yin, Micha ¨el Gharbi, Taesung Park, Richard Zhang, Eli Shechtman, Fredo Durand, and William T Freeman. Improved distribution matching distillation for fast image synthesis.Ad- vances in neural information processing systems, 37:47455–47487, 2024a. Tianwei Yin, Micha¨el Gharbi, Richard Zhang, Eli Shechtman, Fredo Durand, William T Freeman, and Tae...
-
[22]
Diffusionnft: Online diffusion reinforcement with forward process.arXiv preprint arXiv:2509.16117,
Kaiwen Zheng, Huayu Chen, Haotian Ye, Haoxiang Wang, Qinsheng Zhang, Kai Jiang, Hang Su, Stefano Ermon, Jun Zhu, and Ming-Yu Liu. Diffusionnft: Online diffusion reinforcement with forward process.arXiv preprint arXiv:2509.16117,
-
[23]
Danceopd: On-policy generative field distillation.arXiv preprint arXiv:2606.27377,
Wei Zhou, Xiongwei Zhu, Zelin Xu, Bo Dong, Lixue Gong, Yongyuan Liang, Meng Chu, Leigang Qu, Lingdong Kong, Wei Liu, et al. Danceopd: On-policy generative field distillation.arXiv preprint arXiv:2606.27377,
-
[2022]
Jie Liu, Gongye Liu, Jiajun Liang, Yangguang Li, Jiaheng Liu, Xintao Wang, Pengfei Wan, Di Zhang, and Wanli Ouyang. Flow-grpo: Training flow matching models via online rl.Ad- vances in neural information processing systems, 38:40783–40818, 2026a. Jie Liu, Gongye Liu, Jiajun Liang, Ziyang Yuan, Xiaokun Liu, Mingwu Zheng, Xiele Wu, Qiulin Wang, Menghan Xia,...
-
[2023]
Flow-opd: On-policy distillation for flow matching models.arXiv preprint arXiv:2605.08063,
Zhen Fang, Wenxuan Huang, Yu Zeng, Yiming Zhao, Shuang Chen, Kaituo Feng, Yunlong Lin, Lin Chen, Zehui Chen, Shaosheng Cao, et al. Flow-opd: On-policy distillation for flow matching models.arXiv preprint arXiv:2605.08063,
-
[2024]
Training diffusion models with reinforcement learning
Kevin Black, Michael Janner, Yilun Du, Ilya Kostrikov, and Sergey Levine. Training diffusion models with reinforcement learning. InInternational Conference on Learning Representations, volume 2024, pp. 4965–4987,
2024
-
[2025]
Wenjin Hou, Shangpin Peng, Weinong Wang, Zheng Ruan, Yue Zhang, Zhenglin Zhou, Mingqi Gao, Yifei Chen, Kaiqi Wang, Hongming Yang, et al. Uni-opd: Unifying on-policy distillation with a dual-perspective recipe.arXiv preprint arXiv:2605.03677,
-
[2026]
One step diffusion via shortcut models
Kevin Frans, Danijar Hafner, Sergey Levine, and Pieter Abbeel. One step diffusion via shortcut models. InInternational Conference on Learning Representations, volume 2025, pp. 34668– 34684,
2025
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.