mlsanityData to 7 Oct 2026

Z-Reward: Beyond Scalar Rewards by Internalizing Reasoning into Score Distributions

arXiv 2606.09076 ↗First posted 8 Jun 2026cs.CVcited by 2

On the map: LLM alignment / RLHF / preference optimisation → Diffusion and multimodal model alignment

Abstract (arXiv)

Reward models are central to text-to-image post-training, but visual preference is subjective and better represented as a distribution over rubric scores than as a deterministic scalar. Existing scalar, score-token, and pairwise reward models over-compress uncertainty and fine-grained score differences, while reasoning-based generative rewards provide stronger judgments but are costly to deploy and difficult to use as direct optimization signals. We propose Z-Reward, a teacher-student reward modeling framework that decouples reasoning-heavy judgment from efficient reward deployment. The teacher is a large VLM that uses reasoning to infer rubric-aligned score distributions, and is trained with Group-wise Direct Score Optimization (GDSO), which combines policy-gradient rewards from distribution expectations with direct pointwise and pairwise supervision on score distributions and score gaps. The student is trained with Reasoning-Internalized Score Distillation (RISD), which transfers the teacher's reasoning-conditioned score distribution into a compact VLM without requiring explicit reasoning chains at inference time. On our internally annotated evaluation set, the 27B GDSO teacher reaches 89.6% human preference accuracy, outperforming SFT, RewardDance, and GRPO, while the 9B RISD student reaches 88.6%, outperforming the OPD baseline and closely matching the larger teacher. We further show that Z-Reward can serve as a differentiable reward signal for text-to-image optimization, yielding a 41.3% net human-preference improvement over the SFT baseline.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Diffusion and multimodal model alignment, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 55 of its 65 references that are in the map. First posted on arXiv 8 Jun 2026.

Builds on

55 of 65 references.

Closest ancestors · 12

Same direction, newest first

  1. 2026Leveraging Verifier-Based Reinforcement Learning in Image Editing30 Apr
  2. 2026Joint Reward Modeling: Internalizing Chain-of-Thought for Efficient Visual Reward Models7 Feb
  3. 2025RewardDance: Reward Scaling in Visual Generation10 Sep
  4. 2025HPSv3: Towards Wide-Spectrum Human Preference Score5 Aug
  5. 2025Unified Reward Model for Multimodal Understanding and Generation7 Mar
  6. 2025Improving Video Generation with Human Feedback23 Jan
  7. 2024VisionReward: Fine-Grained Multi-Dimensional Human Preference Learning for Image and Video Generation30 Dec
  8. 2024MJ-Bench: Is Your Multimodal Reward Model Really a Good Judge for Text-to-Image Generation?5 Jul
  9. 2024A Dense Reward View on Aligning Text-to-Image Diffusion with Preference13 Feb
  10. 2023Rich Human Feedback for Text-to-Image Generation15 Dec
  11. 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
  12. 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewards29 Sep

Other references · 43

  1. 2022Training language models to follow instructions with human feedbackcited by 15,712
  2. 2015Distilling the Knowledge in a Neural Networkcited by 10,076
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  4. 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,393
  5. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
  6. 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
  7. 2017Deep reinforcement learning from human preferencescited by 3,846
  8. 2021CLIPScore: A Reference-free Evaluation Metric for Image Captioningcited by 2,388
  9. 2023VBench: Comprehensive Benchmark Suite for Video Generative Modelscited by 1,755
  10. 2024A Survey on LLM-as-a-Judgecited by 1,384
  11. 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
  12. 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,010
31 more
  1. 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 950
  2. 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 946
  3. 2023Training Diffusion Models with Reinforcement Learningcited by 939
  4. 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 847
  5. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
  6. 2016Sequence-Level Knowledge Distillationcited by 761
  7. 2023Distilling Step-by-Step! Outperforming Larger Language Models with Less Training Data and Smaller Model Sizescited by 710
  8. 2023Q-Align: Teaching LMMs for Visual Scoring via Discrete Text-Defined Levelscited by 489
  9. 2024Evaluating Text-to-Visual Generation with Image-to-Text Generationcited by 465
  10. 2024Generative Verifiers: Reward Modeling as Next-Token Predictioncited by 465
  11. 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Modelscited by 431
  12. 2023TIFA: Accurate and Interpretable Text-to-Image Faithfulness Evaluation with Question Answeringcited by 416
  13. 2017NIMA: Neural Image Assessmentcited by 387
  14. 2024MLLM-as-a-Judge: Assessing Multimodal LLM-as-a-Judge with Vision-Language Benchmarkcited by 362
  15. 2021Dealing with Disagreements: Looking Beyond the Majority Vote in Subjective Annotationscited by 306
  16. 2023Aligning Text-to-Image Diffusion Models with Reward Backpropagationcited by 284
  17. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
  18. 2025Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformercited by 247
  19. 2025Inference-Time Scaling for Generalist Reward Modelingcited by 233
  20. 2026A Survey of On-Policy Distillation for Large Language Modelscited by 148
  21. 2024Scaling Laws for Reward Model Overoptimization in Direct Alignment Algorithmscited by 139
  22. 2025Teaching Large Language Models to Regress Accurate Image Quality Scores using Score Distributioncited by 139
  23. 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixescited by 122
  24. 2024Learning Multi-dimensional Human Preference for Text-to-Image Generationcited by 102
  25. 2023Toward Verifiable and Reproducible Human Evaluation for Text-to-Image Generationcited by 66
  26. 2026Self-Distillation Zero: Self-Revision Turns Binary Rewards into Dense Supervisioncited by 58
  27. 2026CRISP: Compressed Reasoning via Iterative Self-Policy Distillationcited by 49
  28. 2026Stable On-Policy Distillation through Adaptive Target Reformulationcited by 47
  29. 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixescited by 37
  30. 2025PairJudge RM: Perform Best-of-N Sampling with Knockout Tournamentcited by 21
  31. 2026A Brief Overview: On-Policy Self-Distillation In Large Language Modelscited by 5

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Think Before You Score: Thinking Reward Model for Visual Generation
  2. 2026RationalRewards: Reasoning Rewards Scale Visual Generation Both Training and Test Time
  3. 2026Think, then Score: Decoupled Reasoning and Scoring for Video Reward Modeling
  4. 2025GRAM-R2: Self-Training Generative Foundation Reward Models for Reward Reasoning
  5. 2025InternLM-XComposer2.5-Reward: A Simple Yet Effective Multi-Modal Reward Model
  6. 2026Reward Design for Physical Reasoning in Vision-Language Models
  7. 2025VLRMBench: A Comprehensive and Challenging Benchmark for Vision-Language Reward Models
  8. 2025Listener-Rewarded Thinking in VLMs for Image Preferences