Z-Reward: Beyond Scalar Rewards by Internalizing Reasoning into Score Distributions
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
- 2026Leveraging Verifier-Based Reinforcement Learning in Image Editing30 Apr
- 2026Joint Reward Modeling: Internalizing Chain-of-Thought for Efficient Visual Reward Models7 Feb
- 2025RewardDance: Reward Scaling in Visual Generation10 Sep
- 2025HPSv3: Towards Wide-Spectrum Human Preference Score5 Aug
- 2025Unified Reward Model for Multimodal Understanding and Generation7 Mar
- 2025Improving Video Generation with Human Feedback23 Jan
- 2024VisionReward: Fine-Grained Multi-Dimensional Human Preference Learning for Image and Video Generation30 Dec
- 2024MJ-Bench: Is Your Multimodal Reward Model Really a Good Judge for Text-to-Image Generation?5 Jul
- 2024A Dense Reward View on Aligning Text-to-Image Diffusion with Preference13 Feb
- 2023Rich Human Feedback for Text-to-Image Generation15 Dec
- 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
- 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewards29 Sep
Other references · 43
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,393
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
- 2017Deep reinforcement learning from human preferencescited by 3,846
- 2021CLIPScore: A Reference-free Evaluation Metric for Image Captioningcited by 2,388
- 2023VBench: Comprehensive Benchmark Suite for Video Generative Modelscited by 1,755
- 2024A Survey on LLM-as-a-Judgecited by 1,384
- 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,010
31 more
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 950
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 946
- 2023Training Diffusion Models with Reinforcement Learningcited by 939
- 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 847
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
- 2016Sequence-Level Knowledge Distillationcited by 761
- 2023Distilling Step-by-Step! Outperforming Larger Language Models with Less Training Data and Smaller Model Sizescited by 710
- 2023Q-Align: Teaching LMMs for Visual Scoring via Discrete Text-Defined Levelscited by 489
- 2024Evaluating Text-to-Visual Generation with Image-to-Text Generationcited by 465
- 2024Generative Verifiers: Reward Modeling as Next-Token Predictioncited by 465
- 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Modelscited by 431
- 2023TIFA: Accurate and Interpretable Text-to-Image Faithfulness Evaluation with Question Answeringcited by 416
- 2017NIMA: Neural Image Assessmentcited by 387
- 2024MLLM-as-a-Judge: Assessing Multimodal LLM-as-a-Judge with Vision-Language Benchmarkcited by 362
- 2021Dealing with Disagreements: Looking Beyond the Majority Vote in Subjective Annotationscited by 306
- 2023Aligning Text-to-Image Diffusion Models with Reward Backpropagationcited by 284
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
- 2025Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformercited by 247
- 2025Inference-Time Scaling for Generalist Reward Modelingcited by 233
- 2026A Survey of On-Policy Distillation for Large Language Modelscited by 148
- 2024Scaling Laws for Reward Model Overoptimization in Direct Alignment Algorithmscited by 139
- 2025Teaching Large Language Models to Regress Accurate Image Quality Scores using Score Distributioncited by 139
- 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixescited by 122
- 2024Learning Multi-dimensional Human Preference for Text-to-Image Generationcited by 102
- 2023Toward Verifiable and Reproducible Human Evaluation for Text-to-Image Generationcited by 66
- 2026Self-Distillation Zero: Self-Revision Turns Binary Rewards into Dense Supervisioncited by 58
- 2026CRISP: Compressed Reasoning via Iterative Self-Policy Distillationcited by 49
- 2026Stable On-Policy Distillation through Adaptive Target Reformulationcited by 47
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixescited by 37
- 2025PairJudge RM: Perform Best-of-N Sampling with Knockout Tournamentcited by 21
- 2026A Brief Overview: On-Policy Self-Distillation In Large Language Modelscited by 5
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Think Before You Score: Thinking Reward Model for Visual Generation
- 2026RationalRewards: Reasoning Rewards Scale Visual Generation Both Training and Test Time
- 2026Think, then Score: Decoupled Reasoning and Scoring for Video Reward Modeling
- 2025GRAM-R2: Self-Training Generative Foundation Reward Models for Reward Reasoning
- 2025InternLM-XComposer2.5-Reward: A Simple Yet Effective Multi-Modal Reward Model
- 2026Reward Design for Physical Reasoning in Vision-Language Models
- 2025VLRMBench: A Comprehensive and Challenging Benchmark for Vision-Language Reward Models
- 2025Listener-Rewarded Thinking in VLMs for Image Preferences