Models / TD-MPC

TD-MPC2

UC San Diego · Oct 2023 · MIT

Single 317M agent trained on 104 continuous-control tasks across 4 domains, planning in a learned implicit world model. Scales predictably with model and data size.

Success rate
85%
Downloads
27K
Control rate
50 Hz
Training episodes
—

Benchmarks

Numbers as reported by the authors on the named evaluation suite.

EvaluationSuiteScore

104-task multitask score

TD-MPC2 multitask70.6normalized score

DMControl average

DMControl85normalized score

Meta-World average

Meta-World MT5075%success

Humanoid / dog locomotion

DMControl hard68normalized score

Training

Architecture

Model-based RL with latent-space planning

Parameters

317M

Hardware

8x A100 40GB

Compute

3.4K accelerator-hours

Recipe

Joint multitask training on 545M transitions with decoupled latent dynamics, reward and Q ensembles; no per-task tuning.

Fine-tuning notes

Same hyperparameters across all 104 tasks; capacity scaling from 1M to 317M improves the multitask score monotonically.

Linked datasets, robots & papers

Resolving related models, datasets and papers…