AIAny
Icon for item

Beyond Scalar Rewards by Internalizing Reasoning into Score Distributions

Models visual preference as distributions over rubric scores and introduces Z-Reward, a teacher–student framework that decouples reasoning-heavy judgment (teacher trained with GDSO) from efficient deployment (student via RISD). Demonstrates higher human-preference accuracy and works as a differentiable reward for text-to-image optimization.

Introduction

Visual preference is subjective and often ill-captured by a single scalar score. This paper's core insight is to represent preferences as full score distributions produced by a reasoning-capable teacher VLM, then distill those reasoning-conditioned distributions into a compact, deployable student — preserving nuanced judgments without requiring reasoning at inference time. The approach yields large human-preference gains when used as a reward for text-to-image optimization.

Key Findings
  • Teacher-student decomposition: a large VLM teacher performs reasoning to infer rubric-aligned score distributions; a compact student learns those distributions for fast inference — so what? you keep rich, uncertainty-aware judgments in production-friendly models.
  • Group-wise Direct Score Optimization (GDSO): trains the teacher by combining expectation-based policy-gradient rewards with pointwise and pairwise supervision on score distributions and gaps — so what? it encourages both distributional fidelity and decision-consistent ranking.
  • Reasoning-Internalized Score Distillation (RISD): transfers the teacher's reasoning-conditioned score distribution into the student without exposing reasoning chains at runtime — so what? you get near-teacher accuracy with a much smaller model and lower inference cost.
  • Empirical gains: the 27B GDSO teacher achieves 89.6% human preference accuracy and the 9B RISD student reaches 88.6%, and using Z-Reward as a differentiable reward yields a 41.3% net human-preference improvement over an SFT baseline — so what? distributional rewards materially improve perceived image quality.
Who It's For and Trade-offs

Great fit if you research or build text-to-image reward models, want uncertainty-aware preference signals, or need a deployable reward model that encodes reasoning without runtime chains. Look elsewhere if you cannot access or fine-tune large VLMs, need fully transparent reasoning at inference, or require publicly reproducible benchmarks (the evaluation uses an internally annotated set). Computational and annotation costs for training the teacher are nontrivial.

Method Overview

The teacher is a large vision-language model optimized with GDSO to output score distributions aligned to a human rubric; GDSO blends distribution-expectation policy gradients with direct supervision on distribution shapes and inter-score gaps. The student is trained with RISD to internalize those distributions into a compact VLM, enabling efficient deployment and use as a differentiable reward for downstream text-to-image optimization.

Information

  • Websitearxiv.org
  • AuthorsXin Jin, Huanqia Cai, Zhen Li, Zechao Zhan, Dengyang Jiang, Aiming Hao, Yuming Jiang, Chunle Guo, Peng Gao, Ming-Ming Cheng
  • Published date2026/06/08

More Items

Evaluates spatial cognition of image-generation models by eliciting protocol-constrained visual answers and parsing pixel outputs into structured predictions compatible with existing metrics. Introduces the ProVisE framework and SpatialGen-Bench (470 samples) to compare image-generation models and text-output VLMs on unified spatial tasks.

Selects a referred target from candidate bounding boxes, then decodes tracking waypoints for single-camera embodied visual tracking. Injects past selected-bbox geometry via sliding-window TVBI tokens and is co-trained on a Refer‑QA dataset; achieves SOTA on EVT‑Bench and demonstrates sim-to-real on legged and humanoid robots.

Converts image-content removal into a contrastive on-policy self-distillation signal: the EMA teacher produces next-token distributions with and without image content, uses their log-probability differences to sharpen visual-grounded candidates, and distills that full-distribution target into the student—no external teacher or extra inference cost.