WWorldModel GymResearch Benchmark

Concepts

Tracks, environments & metrics

The benchmark vocabulary: tracks, environments, agents, and the metrics that rank them — including confidence intervals, model fidelity, and planning cost.

Tracks

A track selects which seed suite a run is evaluated against. Seed suites are fixed in code so results are reproducible.

TrackWhat it measures
trainProcedural train seeds — the in-distribution baseline.
testHeld-out procedural test seeds — the headline generalization number.
continualNonstationary dynamics that shift difficulty every few episodes.

Generalization gap

The gap is train_success_rate − test_success_rate. A small gap means the agent generalizes; a large gap means it overfit the train seeds.

Environments

EnvironmentTests
memory_mazePartial observability + memory: navigate a maze where the goal is out of view.
switch_questCausal sub-goals: toggle switches in the right order to unlock a reward.
craft_liteCompositional achievements: gather and combine resources to complete a craft tree.

Agents

Baselines span scripted oracles to learned and planning agents. The registry ships: random, greedy_oracle, planner_oracle, imagination_mpc, search_mcts, and ppo.

Custom agents

An agent implements reset(seed), act(obs, info), and observe(transition), with an optional get_trace() for planner debug output. Register it in agents/worldmodel_agents/registry.py.

Metrics

Runs report scalar metrics plus structured sub-objects. The leaderboard ranks on success rate but surfaces uncertainty and cost alongside it.

MetricMeaning
success_rateFraction of episodes solved — the primary ranking metric.
mean_returnAverage episode return.
success_rate_ciA [low, high] confidence interval, rendered as a whisker on the leaderboard.
median_steps_to_successSample efficiency — how quickly the agent reaches the goal.
model_fidelityk-step reward error of the world model at k1 / k5 / k20 horizons.
planning_costCompute cost, including wall_clock_ms_per_step and imagined transitions.
generalization_gapTrain vs. test success-rate delta.

Why confidence intervals matter

Two agents with the same point success rate are not equal if one has a far wider interval. The leaderboard always renders the CI so rankings stay honest about sample size and variance.

The continual track adds transfer metrics: forward_transfer, backward_transfer, and forgetting.