Research path

Trustworthy Visual Generation Post-Training

Studying how reward validity, uncertainty, and credit assignment determine the reliability of post-training for visual generative models.

中文

Core question

How should a visual generative model be updated when the available reward is incomplete, uncertain, or too coarse to identify which parts of the generation process should change?

Visual post-training can optimize perceptual quality, semantic alignment, motion, and preference objectives that are not fully specified by pre-training losses. Its reliability depends on three questions: whether the reward measures the intended property, how uncertainty in that reward affects the update, and how credit is assigned across samples, regions, timesteps, and objectives.

Pre-training optimizes a model on a fixed dataset with likelihood, reconstruction, or related surrogate objectives. Visual post-training instead samples outputs or denoising trajectories, evaluates them with reward models or preference signals, and updates the generator toward higher-scoring behavior. This changes the optimization target, but it does not by itself create an agent that learns from an external environment.

The central difficulty is that reward models are imperfect measurements. A scalar score may combine several objectives, may be unreliable for an unfamiliar prompt, and may provide no direct evidence about which region or denoising step caused the result. My recent work studies these failure modes separately: reward granularity, uncertainty-aware weighting, adaptation of multiple objectives, temporal credit assignment, efficient generation, and integration into a complete training pipeline.

Distinguish the training signal from the target capability

VrR-VG, ELP, MHEM, and IcoCap address different training settings, but they establish a common diagnostic principle. Benchmark accuracy can depend on dataset shortcuts, classifier confidence can differ from feature discriminability, hard-example losses can promote memorization, and labels can become inconsistent after augmentation. None of these signals should be interpreted without checking how it was produced.

Define the objective and its domain of validity

Visual generation is evaluated by several properties that are only partially captured by likelihood or reconstruction loss, including text alignment, perceptual quality, motion consistency, and human preference. Post-training introduces rewards for these properties, but every reward has a domain of validity defined by its training data, model assumptions, and calibration. Reward design and reward evaluation therefore remain part of the optimization problem.

Represent spatial credit and reward uncertainty

ViPO and BPGO address two distinct limitations of sample-level GRPO rewards. ViPO uses perceptual features to distribute relative advantages over spatial or spatiotemporal regions. BPGO uses a semantic prior to reweight comparisons according to reward confidence. ViPO changes where the update is applied; BPGO changes the contribution of uncertain comparisons.

Adapt multiple rewards and assign temporal credit

TaRoS adjusts the influence of multiple reward components when they disagree, provide weak group separation, or saturate during training. OTCA decomposes a final reward across denoising timesteps and allocates different objectives across the trajectory. The first addresses how a composite reward changes during optimization; the second addresses how an outcome-level reward is assigned to intermediate decisions.

Extend reward-based training to efficient models and full pipelines

RATS uses relative reward quality to determine when a few-step student should follow a multi-step teacher, connecting preference optimization with trajectory distillation. TeleBoost addresses a different level of the problem: it organizes supervised policy shaping, reward-driven reinforcement learning, and preference refinement into a staged video post-training pipeline, with diagnostics and infrastructure supporting training stability.

What trustworthy means here

Here, trustworthy does not mean that a reward model is correct for every prompt or that an attributed region or timestep is the true causal source of quality. The methods specify how available rewards are weighted or assigned within particular image and video training settings. Their conclusions remain conditional on reward-model coverage, prompt distribution, generator architecture, and whether evaluation uses evidence independent of the training rewards.

Open questions

Open questions include how to calibrate reward uncertainty under distribution shift, detect reward exploitation even when aggregate metrics increase, and compare process-level credit across generator architectures. For deployment, another question is how verified outcomes should be used to update the reward model, generator, and evaluation protocol without allowing all three to reinforce the same bias.

Read and cite the papers

This page connects collaborative research contributions; it does not replace the individual papers. Use each canonical paper record for evidence, source links, and citation downloads.