JumpStart Your Policy Learning with Lessons from 160,000 Training Runs
Organizations: University of Washington · Microsoft
Abstract
Reliable progress in offline policy learning depends on careful reporting, well-tuned baselines, and evaluation across diverse conditions. Prior work has shown that results can be sensitive to reporting choices, hyperparameter tuning, and dataset properties, but these sources of variability have not been systematically investigated together at the scale needed to understand how they shape conclusions. To address this gap, we present a large-scale empirical study of offline reinforcement and imitation learning, training over 160,000 policies across 114 datasets. At this scale, no algorithm dominates: aggregate performance among the strongest methods is often close, but the leaders differ substantially across environments. We find that proper hyperparameter tuning frequently reshuffles perceived algorithm rankings and that benchmark composition can produce conflicting conclusions. We also study hyperparameter sensitivity and transfer across environments, identifying a simple strategy for deriving strong default configurations. We use our findings to develop a dataset-conditioned recommender that provides task-specific algorithm recommendations for practitioners. Finally, we release JumpStart: a resource suite containing every trained policy, per-model scores and hyperparameters, strong baselines across all environments, training and evaluation code, and an extensible website for retrieving, analyzing, and contributing results. Together, these resources aim to make offline policy-learning research more reliable and enable future work beyond the scope of this study.
Figures & tables
| Study | Datasets or tasks | Methods / variants | Policies counted | Continuous | Discrete | Trained models | Policy metadata | Training code | Extensible leaderboard |
|---|---|---|---|---|---|---|---|---|---|
| agarwal2020optimistic (Atari) | 60 games | 6 | – | ✓ | – | – | ✓ | – | |
| paine2020hyperparameterselectionofflinereinforcement | 10 tasks | 3 | ✓ | – | – | – | – | – | |
| mandlekar2021matterslearningofflinehuman | 8 tasks | 6 | NR | ✓ | – | ✓ | – | ||
| schweighofer2021dataset | 150 (6 envs.) | 9 | 6,750 | – | ✓ | – | – | ✓ | – |
| d3rlpy | 17 datasets/tasks | 12 | ✓ | ✓ | – | ✓ | ✓ | – | |
| tarasov2023revisiting | 30 datasets | 10 | ✓ | – | – | ✓ | ✓ | – |
| BC | BCP | BCQ | CQL | DT | IQL | ReBRAC | ACT | VQ-BeT | Diffusion | Total | |
|---|---|---|---|---|---|---|---|---|---|---|---|
| Continuous | 11,400 | 11,400 | 11,400 | 11,400 | 11,400 | 11,302 | 11,401 | 10,830 | 8,833 | 10,590 | 109,956 |
| Discrete | 11,400 | 11,400 | 8,687 | 10,021 | 11,400 | – | – | – | – | – | 52,908 |
| Total | 22,800 | 22,800 | 20,087 | 21,421 | 22,800 | 11,302 | 11,401 | 10,830 | 8,833 | 10,590 | 162,864 |
| Algorithm | Top 10% | All trials | Enrichment |
|---|---|---|---|
| ACT | 32.8% | 35.3% | pp |
| BC | 33.5% | 33.9% | pp |
| BCP | 41.6% | 40.3% | pp |
| BCQ | 85.4% | 85.2% | pp |
| CQL | 83.3% | 78.5% | pp |
| DT | 70.1% | 69.9% | pp |
| Parameter | Range |
|---|---|
| Learning rate | |
| Weight decay | |
| Gradient clip | |
| Dropout | |
| Hidden units | |
| Policy type | {det., stoch.} |
| Parameter | Range |
|---|---|
| Learning rate | |
| Weight decay | |
| Gradient clip | |
| Dropout | |
| Hidden units | |
| Regularization |
| Hyperparameter | Share (%) | |
|---|---|---|
| Hidden units | ||
| Learning rate | ||
| Dropout | ||
| Weight decay | ||
| Policy type | ||
| Grad-clip norm |
| Hyperparameter | Share (%) | |
|---|---|---|
| Learning rate | ||
| Dropout | ||
| Hidden units | ||
| Reg. factor | ||
| Weight decay | ||
| Grad-clip norm |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
Offline RL
Offline reinforcement learning learns decision-making policies from fixed datasets without interacting with the environment during training.
See all 290 papers