REVIEW 5 major objections 5 minor 5 cited by
By aligning reasoning representations with discriminative features in latent space, chain-of-thought reasoning can be delivered through a fast scalar reward signal.
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 · deepseek-v4-flash
2026-08-03 03:33 UTC pith:QAC6KKUD
load-bearing objection A genuinely promising training recipe for discriminative visual reward models, with SOTA claims that hinge on undisclosed train/eval split details and a mechanism story backed by indirect evidence. the 5 major comments →
Joint Reward Modeling: Internalizing Chain-of-Thought for Efficient Visual Reward 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
JRM's central claim is that generative-style reasoning can be internalized into the latent space of a discriminative reward model. A shared representation h = E(x, c) is jointly constrained by an uncertainty-aware ranking loss over human preferences and a cross-entropy language loss over structured evaluation text. During inference the language head is removed, and the scalar score head reads directly from h; the paper calls this implicit process Latent Chain-of-Thought. The supporting evidence is a 85.1% overall score on EditReward-Bench and 69.3% on MMRB2, a near-doubling of effective feature-space rank versus the ranking-only baseline (91.77 vs 46.86), and downstream reinforcement-learnin
What carries the argument
The load-bearing mechanism is the shared representation h produced by the vision-language backbone for an image pair and instruction. Two heads read from h: a RankNet-style discriminative score head trained with an uncertainty-aware Gaussian ranking loss, and a conditional language head trained to generate structured, region-referencing evaluation text. The language loss pushes h to encode global semantic structure, while the ranking loss keeps h ordered by human preference. At test time only the score head runs, so the claimed reasoning gain is obtained without generating text.
Load-bearing premise
The argument depends on the generated explanations used as language supervision carrying semantic information beyond the pre-assigned score they were prompted with; if the text mostly restates the score, the language loss adds no new reasoning signal and the claimed internalization reduces to extra regularization.
What would settle it
Train JRM exactly as described but replace each explanation with a length-matched, score-matched explanation from a different sample, keeping the ranking loss and data volume fixed. If benchmark accuracy stays at JRM's reported level, the language objective is contributing regularization or compute rather than reasoning; if accuracy drops, the semantic content of the explanations is doing the work. A complementary check is to evaluate JRM on instructions and image pairs with no overlap with its training distribution.
If this is right
- Reward models for image editing no longer need to generate reasoning at evaluation time: JRM reports 85.1% on EditReward-Bench and 69.3% on MMRB2 while using the same discriminative inference path as a plain ranking scorer.
- Using JRM as the reward signal in online reinforcement learning improves an image-editing model by +1.00 on GEdit-Bench and +0.50 on ImageEdit-Bench, larger than the gains reported for the compared closed-source judges.
- Joint language supervision stabilizes and accelerates ranking-loss convergence and roughly doubles the effective rank of the shared representation, which the paper interprets as preventing collapse to shallow visual cues.
- In the tested range, larger language-supervision weight α consistently improves benchmark accuracy, suggesting the language objective contributes semantic structure rather than acting only as a regularizer.
- The language head can be re-enabled as a diagnostic to produce textual evaluations whose stated problems match the scalar scores, enabling JRM-guided self-correction of editing models.
Where Pith is reading between the lines
- If the mechanism transfers, the same joint ranking-plus-language objective could produce fast reasoning-like discriminators in other expensive evaluation domains, such as video generation or multi-turn dialogue, where full chain-of-thought generation is too costly at test time.
- The decisive untested control is to train JRM with the same generated explanations but with their content shuffled relative to the scores; if accuracy survives the shuffle, the gain comes from extra data or compute rather than from the semantics of the reasoning text.
- The sharp rise in effective rank suggests a practical extension the paper does not run: after joint training, prune or distill the language head entirely and check how much of the benchmark advantage remains in the score head alone.
- The region-tagged language supervision hints that the shared representation could be used not only to score an edit but to localize what went wrong, potentially supporting targeted repair rather than reward assignment alone.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes Joint Reward Modeling (JRM), a method that trains a shared vision-language backbone with both a preference-ranking loss and a language-modeling loss on image-editing data, then uses only the discriminative scoring head at inference. The central claim is that joint training internalizes chain-of-thought-style reasoning into the shared representation, so that a discriminative reward model can match or exceed generative reward models in semantic evaluation without paying generation cost at inference. The authors report state-of-the-art accuracy on EditReward-Bench (85.1% vs GPT-5's 75.5%) and MMRB2 (69.3% vs 61.9%), representation-space evidence (effective rank 91.77 vs 46.86), and downstream online RL gains with JRM as the reward signal on GEdit-Bench (+1.00) and ImageEdit-Bench (+0.50).
Significance. If the empirical claims hold, JRM would be a practically valuable result: it would show that language-supervision can be folded into a discriminative reward model, combining reasoning-like semantic understanding with efficient inference suitable for online RL. The paper has clear strengths: the method is simple and reproducible in principle, the authors provide a code link, they evaluate on multiple benchmarks, ablate the language-supervision weight, analyze internal representations, and test the model in an actual Flow-GRPO loop. However, the current evidence is not yet sufficient to support the central SOTA and mechanism claims, because of a likely train/evaluation overlap and the lack of a control isolating the information added by the generated language targets.
major comments (5)
- [§3.1 + Table 1] The training data is built 'upon existing image editing reward datasets (Wu et al., 2025d)', which is the same source that introduces EditReward-Bench, the benchmark in Table 1. The paper never discloses whether EditReward-Bench evaluation pairs were excluded from training, whether duplicates or near-duplicates were removed, or what the exact train/eval split is. If any evaluation pairs or images from the same editing tasks appear in the training corpus, the reported 85.1% Overall and 85.4% Prompt-Following numbers could be inflated by memorization rather than by internalized reasoning. The same corpus is also used to train the reward signal for the downstream RL experiments, so contamination could affect the GEdit +1.00 / ImageEdit +0.50 gains. This is a load-bearing disclosure that must be provided.
- [Appendix A.1 + §3.2] The language-supervision targets are generated by prompting a VLM with the pre-assigned score as part of the input ('generate a structured evaluation entry for an image editing pair based on a pre-assigned score'). The LM loss may therefore simply learn to verbalize the scalar score, adding no semantic information beyond what the ranking loss already provides. The central mechanism claim — that reasoning is 'internalized' into the representation — requires a control that isolates the information content of the language targets. For example, training with score-scrambled or paraphrased language targets, or with language targets produced without conditioning on the score, would show whether the effect comes from the language content or from additional regularization/compute. Without this control, the 'internalized reasoning' claim is not established.
- [§4.2, Figure 4] The note under Figure 4 states that except for α=0.7, the optimal checkpoint differs between the two benchmarks, and results shown are the best performance achieved on each benchmark. This makes the α-ablation an upper envelope, not a fair comparison of a single training run, and it means α=0.7 is selected post hoc for each benchmark. The monotonic improvement trend claimed in the text is therefore not supported as stated. Reporting a single held-out checkpoint, or multiple seeds with error bars, would be needed to support the claim that language supervision has a monotonic positive effect.
- [§4.1, Tables 1–2, Table 4] No error bars, confidence intervals, or significance tests are reported for any of the central numbers. Several headline comparisons are small in absolute terms (e.g., MMRB2 Overall 69.3% vs EditReward's 65.7%; EditReward-Bench Overall 85.1% vs PaCo-Reward's 75.1% is larger but still a single run). Since model training is stochastic, the claim of 'significantly outperforms' (Section 1) requires at least multiple seeds or a statement about evaluation/test-set determinism. This is particularly important for the downstream RL results in Table 4, where +0.19 vs +0.23 vs +0.50 are within the range of typical RL-run variance.
- [§4.3 + §3.3] The representation analysis shows that joint training changes the singular-value spectrum and effective rank, but it does not establish that the additional dimensions encode reasoning semantics. The comparison is between α=0 and α=0.7; any additional training signal (including the extra language-modeling objective and the associated compute) could produce a more dispersed representation without instantiating a 'Latent Chain-of-Thought'. The paper needs a control (e.g., training with an equally weighted but non-semantic auxiliary loss) to support the mechanistic interpretation in Eq. (8)–(9).
minor comments (5)
- [Table 1] The 'Cons.' column has dashes for EditReward and JRM, but the text discusses only PF and Overall. Please clarify what 'Cons.' is, why it is omitted for these two methods, and whether it is a meaningful comparison dimension.
- [References] Several references are incomplete: 'Yao et al.' has no year or venue, 'Ankner et al.' has no year, and Zhang et al. appears as 'Zhang et al.;' with no citation details in the Related Work section.
- [§3.3, Eq. (9)] The notation 'rank Cov(h) ↑' is informal and not a well-defined equation; please state the intended mathematical claim (e.g., effective rank of the covariance matrix of a batch of representations).
- [Figure 5 / Appendix B] The text says 'For training details, see Figure 5 and Appendix B,' but Figure 5 shows loss dynamics, not hyperparameter details. The actual training hyperparameters are in Appendix B; please fix the pointer.
- [§3.2, Appendix B] The shared representation h is first defined in Eq. (1) as the output of E(x,c), but the concrete implementation (a learnable special token whose final hidden state is h) appears only in Appendix B. Moving this detail into Section 3.2 would make the architecture clearer.
Circularity Check
No significant circularity; the central JRM claim is empirically testable and supported by an external benchmark, with two non-circular data-quality caveats.
full rationale
Walking the derivation chain: Eq. (7) defines the joint objective L_total = (1-alpha)L_rank + alpha L_LM, and inference (Eq. 10) outputs r = f_theta(E(x,c)) from the shared representation only. No predicted quantity is computed from a parameter that was fit to that same quantity; the score head is not a renamed version of the language head or of the language targets. The alpha=0.7 choice is a hyperparameter selected from benchmark curves (Fig. 4) and is normal model selection, not a fitted input renamed as a prediction. The central SOTA claim is not exclusively self-referential: MMRB2 (Hu et al., 2025) is an external benchmark, and JRM beats prior methods there too. The representation-rank analysis (91.77 vs 46.86) is an additional empirical observation, not a tautology. Two caveats are worth flagging but neither is a circular reduction. First, Appendix A.1 generates language supervision by prompting a VLM 'based on a pre-assigned score,' so the L_LM target is partly a verbalization of the same preference label used in L_rank; this is a supervision-redundancy / mechanism-identification concern, not a case where the derivation equals its input by construction. Second, Section 3.1 says JRM is built 'upon existing image editing reward datasets (Wu et al., 2025d),' while Table 1 evaluates EditReward-Bench from the same citation, with no disclosed train/eval split; this is a potential contamination risk that the authors can settle empirically, but absent evidence of overlap it cannot be called circular. Self-citations to R1-Reward / Keye-VL appear as background related work and are not load-bearing for the main result.
Axiom & Free-Parameter Ledger
free parameters (1)
- alpha (language supervision weight) =
0.7
axioms (5)
- domain assumption Human preference comparisons in EditReward/MMRB2 are reliable ground truth for reward quality.
- domain assumption A VLM prompted with the assigned score produces faithful, informative reasoning text (Appendix A).
- domain assumption Effective rank, isotropy, and spectral entropy of hidden representations measure 'reasoning capability'.
- standard math The uncertainty-aware ranking formula from HPSv3 (Equations 4-5) models reward noise adequately.
- standard math Flow-GRPO (Liu et al., 2025a) provides a valid online RL alignment procedure.
invented entities (1)
-
Latent Chain-of-Thought (Latent CoT)
no independent evidence
read the original abstract
Reward models are critical for reinforcement learning from human feedback, as they determine the alignment quality and reliability of generative models. For complex tasks such as image editing, reward models are required to capture global semantic consistency and implicit logical constraints beyond local similarity. Existing reward modeling approaches have clear limitations. Discriminative reward models align well with human preferences but struggle with complex semantics due to limited reasoning supervision. Generative reward models offer stronger semantic understanding and reasoning, but they are costly at inference time and difficult to align directly with human preferences. To this end, we propose Joint Reward Modeling (JRM), which jointly optimizes preference learning and language modeling on a shared vision-language backbone. This approach internalizes the semantic and reasoning capabilities of generative models into efficient discriminative representations, enabling fast and accurate evaluation. JRM achieves state-of-the-art results on MMRB2 and EditReward-Bench, and significantly improves stability and performance in downstream online reinforcement learning. These results show that joint training effectively bridges efficiency and semantic understanding in reward modeling.
Figures
Forward citations
Cited by 5 Pith papers
-
Will It Go Viral? Grounding Micro-Video Popularity Prediction on the Open Web
WEBSHORTS dataset and SHORTS-CAST framework ground micro-video popularity prediction in structured open-web context collected at upload time and enable selective online adaptation using delayed labels.
-
SpatialFlow-GRPO: Where Spatial Credit Drives Image Editing
SpatialFlow-GRPO improves image editing quality by converting region-aware rewards into semantic-region-level optimization signals aligned with latent positions during policy updates.
-
SpatialFlow-GRPO: Where Spatial Credit Drives Image Editing
SpatialFlow-GRPO adds region-level reward feedback and spatial alignment to Flow-GRPO-style RL for image editing, reporting gains on GEdit-Bench, ImgEdit-Bench, and a new MultiEditBench.
-
Z-Reward: Beyond Scalar Rewards by Internalizing Reasoning into Score Distributions
Z-Reward trains a 27B reasoning teacher VLM on score distributions via GDSO and distills it via RISD into a 9B student, reaching 89.6% and 88.6% human preference accuracy with 41.3% optimization gain over SFT baseline.
-
Z-Reward: Beyond Scalar Rewards by Internalizing Reasoning into Score Distributions
A teacher-student reward model learns reasoning-conditioned score distributions for text-to-image images, yielding ~89% preference accuracy and a 41% net human-preference gain when used for generator optimization.
Reference graph
Works this paper leans on
-
[4]
Emerging proper- ties in unified multimodal pretraining.arXiv preprint arXiv:2505.14683,
Deng, C., Zhu, D., Li, K., Gou, C., Li, F., Wang, Z., Zhong, S., Yu, W., Nie, X., Song, Z., et al. Emerging proper- ties in unified multimodal pretraining.arXiv preprint arXiv:2505.14683,
-
[5]
Gong, Y ., Wang, X., Wu, J., Wang, S., Wang, Y ., and Wu, X. Onereward: Unified mask-guided image generation via multi-task human preference learning.arXiv preprint arXiv:2508.21066,
-
[6]
Guo, D., Yang, D., Zhang, H., Song, J., Zhang, R., Xu, R., Zhu, Q., Ma, S., Wang, P., Bi, X., et al. Deepseek-r1: In- centivizing reasoning capability in llms via reinforcement learning.arXiv preprint arXiv:2501.12948,
-
[7]
Tempflow-grpo: When timing matters for grpo in flow models.arXiv preprint arXiv:2508.04324,
He, X., Fu, S., Zhao, Y ., Li, W., Yang, J., Yin, D., Rao, F., and Zhang, B. Tempflow-grpo: When timing matters for grpo in flow models.arXiv preprint arXiv:2508.04324,
-
[8]
Hu, Y ., Askari-Hemmat, R., Hall, M., Dinan, E., Zettle- moyer, L., and Ghazvininejad, M. Multimodal reward- bench 2: Evaluating omni reward models for interleaved text and image.arXiv preprint arXiv:2512.16899,
-
[10]
Flow-grpo: Training flow matching models via online rl.arXiv preprint arXiv:2505.05470, 2025a
Liu, J., Liu, G., Liang, J., Li, Y ., Liu, J., Wang, X., Wan, P., Zhang, D., and Ouyang, W. Flow-grpo: Training flow matching models via online rl.arXiv preprint arXiv:2505.05470, 2025a. Liu, S., Han, Y ., Xing, P., Yin, F., Wang, R., Cheng, W., Liao, J., Wang, Y ., Fu, H., Han, C., et al. Step1x-edit: A practical framework for general image editing.arXiv...
-
[11]
Editscore: Unlocking online rl for image editing via high-fidelity reward modeling
Luo, X., Wang, J., Wu, C., Xiao, S., Jiang, X., Lian, D., Zhang, J., Liu, D., et al. Editscore: Unlocking online rl for image editing via high-fidelity reward modeling. arXiv preprint arXiv:2509.23909,
-
[13]
Proximal policy optimization algorithms
Schulman, J., Wolski, F., Dhariwal, P., Radford, A., and Klimov, O. Proximal policy optimization algorithms. arXiv preprint arXiv:1707.06347,
-
[14]
See- dream 4.0: Toward next-generation multimodal image generation.arXiv preprint arXiv:2509.20427,
Seedream, T., Chen, Y ., Gao, Y ., Gong, L., Guo, M., Guo, Q., Guo, Z., Hou, X., Huang, W., Huang, Y ., et al. See- dream 4.0: Toward next-generation multimodal image generation.arXiv preprint arXiv:2509.20427,
-
[15]
Wang, Y ., Li, Z., Zang, Y ., Wang, C., Lu, Q., Jin, C., and Wang, J. Unified multimodal chain-of-thought reward model through reinforcement fine-tuning.arXiv preprint arXiv:2505.03318, 2025a. Wang, Y ., Zang, Y ., Li, H., Jin, C., and Wang, J. Unified re- ward model for multimodal understanding and generation. CoRR, 2025b. Wei, H., Xu, B., Liu, H., Wu, C...
-
[16]
Qwen-image technical report.arXiv preprint arXiv:2508.02324, 2025a
Wu, C., Li, J., Zhou, J., Lin, J., Gao, K., Yan, K., Yin, S.-m., Bai, S., Xu, X., Chen, Y ., et al. Qwen-image technical report.arXiv preprint arXiv:2508.02324, 2025a. Wu, C., Zheng, P., Yan, R., Xiao, S., Luo, X., Wang, Y ., Li, W., Jiang, X., Liu, Y ., Zhou, J., et al. Omnigen2: Exploration to advanced multimodal generation.arXiv preprint arXiv:2506.188...
-
[17]
Dancegrpo: Unleashing grpo on visual generation.arXiv preprint arXiv:2505.07818,
Xue, Z., Wu, J., Gao, Y ., Kong, F., Zhu, L., Chen, M., Liu, Z., Liu, W., Guo, Q., Huang, W., et al. Dancegrpo: Unleashing grpo on visual generation.arXiv preprint arXiv:2505.07818,
-
[18]
Yao, S., Yu, D., Zhao, J., Shafran, I., Griffiths, T
URL https://arxiv.org/abs/2509.01563. Yao, S., Yu, D., Zhao, J., Shafran, I., Griffiths, T. L., Cao, Y ., and Narasimhan, K. R. Tree of thoughts: Deliberate problem solving with large language models. InThirty- seventh Conference on Neural Information Processing Systems. Ye, Y ., He, X., Li, Z., Lin, B., Yuan, S., Yan, Z., Hou, B., and Yuan, L. Imgedit: A...
-
[19]
Zhang, P., Dong, X., Wang, B., Cao, Y ., Xu, C., Ouyang, L., Zhao, Z., Duan, H., Zhang, S., Ding, S., et al. Internlm-xcomposer: A vision-language large model for advanced text-image comprehension and composition. arXiv preprint arXiv:2309.15112,
-
[20]
Mm-rlhf: The next step forward in multimodal llm alignment
Zhang, Y ., Yu, T., Tian, H., Fu, C., Li, P., Zeng, J., Xie, W., Shi, Y ., Zhang, H., Wu, J., et al. Mm-rlhf: The next step forward in multimodal llm alignment. InF orty-second International Conference on Machine Learning. Zhang, Y .-F., Lu, X., Hu, X., Fu, C., Wen, B., Zhang, T., Liu, C., Jiang, K., Chen, K., Tang, K., et al. R1-reward: 11 Joint Reward M...
Pith/arXiv arXiv 1909
-
[2017]
Comanici, G., Bieber, E., Schaekermann, M., Pasupat, I., Sachdeva, N., Dhillon, I., Blistein, M., Ram, O., Zhang, D., Rosen, E., et al. Gemini 2.5: Pushing the frontier with advanced reasoning, multimodality, long context, and next generation agentic capabilities.arXiv preprint arXiv:2507.06261,
-
[2022]
10 Joint Reward Modeling: Internalizing Chain-of-Thought for Efficient Visual Reward Models Ping, B., Jia, C., Luo, M., Xia, C., Shen, X., Dang, Z., and Qian, H. Paco-rl: Advancing reinforcement learning for consistent image generation with pairwise reward modeling.arXiv preprint arXiv:2512.04784,
-
[2023]
URL https: //arxiv.org/abs/2308.12966. Bai, S., Cai, Y ., Chen, R., Chen, K., Chen, X., Cheng, Z., Deng, L., Ding, W., Gao, C., Ge, C., Ge, W., Guo, Z., Huang, Q., Huang, J., Huang, F., Hui, B., Jiang, S., Li, Z., Li, M., Li, M., Li, K., Lin, Z., Lin, J., Liu, X., Liu, J., Liu, C., Liu, Y ., Liu, D., Liu, S., Lu, D., Luo, R., Lv, C., Men, R., Meng, L., Re...
-
[2024]
Labs, B. F., Batifol, S., Blattmann, A., Boesel, F., Con- sul, S., Diagne, C., Dockhorn, T., English, J., English, Z., Esser, P., et al. Flux. 1 kontext: Flow matching for in-context image generation and editing in latent space. arXiv preprint arXiv:2506.15742,
-
[2025]
URLhttps://arxiv.org/abs/2511.21631. Christiano, P. F., Leike, J., Brown, T. B., Martic, M., Legg, S., and Amodei, D. Deep reinforcement learning from hu- man preferences. InProceedings of the 31st International Conference on Neural Information Processing Systems, pp. 4302–4310,
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.