all projects

Research Paper

Ground-Truth-Free World Model Fine-Tuning with GameWorldScore

Rijul Dahiya1

1New York University, Courant Institute of Mathematical Sciences
Read paper Code

Abstract

World models for video generation face a critical challenge: how to improve generation quality without access to ground-truth future frames during fine-tuning. We propose GameWorldScore, a unified ground-truth-free reward system that enables reinforcement learning fine-tuning of diffusion-based world models. Our approach combines five complementary reward components: Inverse Kinematics Score (RIK) for action fidelity, Temporal Consistency Score (RTC) for motion smoothness, Aesthetic Quality Score (RAQ) for visual appeal, Reality Grounding Score (RRG) for domain anchoring, and Anti-Drift Reward (AD) for preventing degradation. We demonstrate that Group Relative Policy Optimization (GRPO) with GameWorldScore successfully fine-tunes the Oasis world model on Minecraft video generation, achieving improved visual quality, temporal coherence, and action-conditioned generation without requiring paired ground-truth data.

Key Contributions

Method

GameWorldScore combines five reward signals into a unified objective: R_total = w1·RIK + w2·RTC + w3·RAQ + w4·RRG + w5·RAD. The Oasis world model, consisting of a frozen VAE and a fine-tuned Diffusion Transformer (DiT), generates frames autoregressively. GRPO normalises advantages within groups of samples from the same prompt, making it well-suited to diffusion model training without requiring paired ground-truth observations.

Results

The unified model, trained on all five rewards at once, matches each single-reward specialist on that specialist's own metric while holding the rest — action fidelity 0.469 against the RIK-only model's 0.466, anti-drift 0.842 against 0.859. Validation means over the held-out set:

Reward componentUnified meanUnified maxSpecialist mean
RIK — action fidelity0.4690.6140.466
RTC — temporal consistency0.5480.9340.620
RAQ — aesthetic quality0.5910.6060.604
RRG — reality grounding0.6660.7390.663
AD — anti-drift0.8420.9770.859

Ablations

Limitations

Figures

Figure 1. Combined GameWorldScore Performance, all five reward components trained simultaneously on the Oasis world model.
Figure 1. Combined GameWorldScore Performance, all five reward components trained simultaneously on the Oasis world model.
Figure 2. Specialist Models Held-Out GameWorldScore Performance, single-reward specialist models evaluated across all metrics.
Figure 2. Specialist Models Held-Out GameWorldScore Performance, single-reward specialist models evaluated across all metrics.

Full paper