Stable World Model

Stable World Model is a research platform for collecting data, training world models, and evaluating policies with model-predictive control across standardized environments. The LanceDB integration is built into Stable World Model’s data format registry. Lance is the default backend for collected datasets, so a path ending in .lance gives you an append-friendly LanceDB table with episode-contiguous rows and fast indexed reads. That matters for world model research because the training loop repeatedly samples temporal windows from high-dimensional observations, actions, and rewards. Faster random access means more GPU time is spent training the model instead of waiting on the data loader.

Install

pip install stable-worldmodel

Datasets and checkpoints are stored under $STABLEWM_HOME, which defaults to ~/.stable_worldmodel/.

Collect data into Lance

Stable World Model uses Lance by default when you collect to a .lance path. Replace your_expert_policy with the expert or scripted policy you use to collect demonstrations. Every writer accepts a mode argument such as append, overwrite, or error. The default is append, so re-running collection extends the existing dataset.

Load a Lance dataset for training

The dataset loader autodetects the Lance format from the path. This keeps model code focused on the world model objective while LanceDB handles the storage layout and read path.

Evaluate with model-predictive control

After training a world model on the Lance-backed dataset, Stable World Model can evaluate it with planning solvers such as CEM. Replace world_model with the trained model object from your training loop.

Convert between formats

Stable World Model can also convert between registered dataset formats. For example, you can collect in Lance for fast training reads, then export to the video layout for compact inspection artifacts.

Throughput

The Stable World Model README reports the following PushT benchmark results from scripts/benchmark/compare_h5_lance.py:

Format	Source	Cache	samples/s	ms/step
HDF5	local	no-cache	1,416.1	45.2
HDF5	local	cached	1,474.0	43.4
LanceDB	local	no-cache	4,814.8	13.3
LanceDB	local	cached	4,431.3	14.4
Video	local	-	1,330.6	48.1
LanceDB	s3	no-cache	3,183.7	20.1
LanceDB	s3	cached	3,253.2	19.7
HDF5	s3	no-cache	9.1	7,032.5
HDF5	s3	cached	756.5	84.6

In that benchmark, local LanceDB reached about 3.4x the no-cache throughput of local HDF5, while S3-backed LanceDB reached about 350x the no-cache throughput of S3-backed HDF5. Even with cache enabled, S3-backed LanceDB was about 4.3x faster than S3-backed HDF5. These numbers are reported by the Stable World Model project for its benchmark setup. Treat them as a reproducible directional baseline, not a universal guarantee for every environment, model, or storage configuration.

Storage

The same README reports these local storage sizes for the benchmark dataset:

Format	Local size
HDF5	43.12 GB
LanceDB	13.31 GB
Video	496.29 MB

LanceDB used about 69% less local storage than HDF5 in the reported benchmark, while preserving a table interface that is designed for fast training reads and append-heavy dataset collection.

More resources