Skip to content

kesai-labs/lead

Repository files navigation

KE:SAI     University of Tübingen     NVIDIA

LEAD: Minimizing Learner-Expert Asymmetry in End-to-End Driving

Website  ·  Docs  ·  Dataset  ·  Model  ·  NAVSIM Model  ·  Supplementary  ·  Paper

An open-source end-to-end driving stack for CARLA.
▶ State-of-the-art performance on all major Leaderboard 2.0 benchmarks ◀

Bench2Drive Longest6 V2 Town13 Fail2Drive

Long Nguyen1,3  ·   Micha Fauth1  ·   Bernhard Jaeger1,3  ·   Daniel Dauner1,3
Maximilian Igl2  ·   Andreas Geiger1,3  ·   Kashyap Chitta2

1University of Tübingen, Tübingen AI Center  ·  2NVIDIA Research  ·  3KE:SAI

Table of Contents

Updates

Date Content
26.04.29 PlanT 2.0 privileged planner integration added, see instructions.
26.04.25 LEAD360 dataset published, features six cameras and metas to train PlanT 2.0.
26.04.19 LEAD123D dataset published: five hours driving logs in 123D format.
26.04.17 NAVSIM LTFv6 checkpoint predicts in CARLA's left-handed frame. When using inference code in HuggingFace, convert waypoints/headings back to ISO 8855 before evaluation.
26.04.14 Bug: transfuser_token_dim's default value is 256.
26.04.13 Bug: Occasionally, we notice training instability. See instructions, if you face similar problem.
26.04.11 Fail2Drive benchmark support added, see instructions.
26.03.21 CaRL reinforcement-learning planner evaluation support added, see instructions.
26.03.18 Update: Removed Kalman Filter and post-processing heuristics. See performance report here.
26.02.25 🎉🎉 LEAD is accepted to CVPR 2026 🎉🎉
26.02.25 NAVSIM extension released. Code and instructions available. Supplementary data coming soon.
26.02.02 123D data collection format preliminary support added, see instructions.
26.01.13 LEAD dataset and training documentation released.
25.12.24 Initial release - paper, checkpoints, expert driver, and inference code.

1. Quick Start

Get LEAD running locally: from cloning the repo and installing dependencies, to downloading a pretrained checkpoint and driving a CARLA Leaderboard 2.0 route end-to-end. We tested the instructions on the following configurations:

OS GPU CUDA Driver Inference Training
Ubuntu 22.04 L40S 13.0 580
Ubuntu 22.04 A100 13.0 580
Ubuntu 24.04 RTX 5090 13.1 590
Ubuntu 22.04 RTX A4000 13.0 580
Ubuntu 22.04 GTX 1080ti 13.0 580

1.1. Environment initialization

Clone the repository and register the project root:

git clone https://github.com/kesai-labs/lead.git
cd lead

# Set project's environment variables
echo -e "export LEAD_PROJECT_ROOT=$(pwd)" >> ~/.bashrc
echo "source $(pwd)/scripts/main.sh" >> ~/.bashrc

# Reload shell config
source ~/.bashrc

Verify that ~/.bashrc reflects these paths correctly.

1.2. Install dependencies

Set up CARLA. Beside simulation, we will need its Python API at 3rd_party/CARLA_0915/PythonAPI/carla in PYTHONPATH:

# Download and setup CARLA at 3rd_party/CARLA_0915
bash scripts/setup_carla.sh

# Or symlink your pre-installed CARLA
ln -s /your/carla/path 3rd_party/CARLA_0915

We use Miniconda as container and uv for Python dependencies. Runtime and dev dependencies are declared entirely in pyproject.toml.

# (Optional, needed in some cases) Accept terms and services
conda tos accept --override-channels --channel \
  https://repo.anaconda.com/pkgs/main
conda tos accept --override-channels --channel \
  https://repo.anaconda.com/pkgs/r

# Create a conda environment
conda create -n lead python=3.10 -y
conda activate lead

# Install system-level tools
conda install -c conda-forge ffmpeg parallel tree gcc zip unzip git-lfs uv

# Tell uv to use conda environment
mkdir -p $CONDA_PREFIX/etc/conda/activate.d $CONDA_PREFIX/etc/conda/deactivate.d
echo 'export VIRTUAL_ENV=$CONDA_PREFIX' > $CONDA_PREFIX/etc/conda/activate.d/uv.sh
echo 'unset VIRTUAL_ENV' > $CONDA_PREFIX/etc/conda/deactivate.d/uv.sh
conda activate lead

# Install dependencies
uv sync --active --extra dev

# Optional: activate git hooks
pre-commit install

Tip

  1. Blackwell and newer GPUs (RTX 5090, etc.) require PyTorch 2.7+ with CUDA 12.8. This setup is not tested for training and might lead to instability.
pip install torch==2.7.0 torchvision --index-url https://download.pytorch.org/whl/cu128
  1. uv cheatsheet:
# Add dependency
uv add --active <pkg>

# Add dev dependency
uv add --active --optional dev <pkg>

# Update uv.lock
uv lock

1.3. Download checkpoints

Pre-trained checkpoints are hosted on HuggingFace. The following table depicts the results from our paper. Evaluations of some checkpoints on Town13 are omitted because of resource constraints. Some numbers do not have full precision.

Variant Bench2Drive Longest6 v2 Town13 Checkpoint
Full TransFuser V6 95.28 62.92 5.24 Link
ResNet34 (60M params) 94.72 57.74 5.01 Link
  + Rear camera 95.04 54.16 -- Link
  − Radar 94.70 52.00 -- Link
  Vision only 91.60 43.00 -- Link
  Town13 held out 93.00 52.00 3.52 Link

To reproduce these results, enable the Kalman filter, stop-sign, and creeping heuristics by setting sensor_agent_creeping=True use_kalman_filter=True slower_for_stop_sign=True in config_closed_loop. Without these heuristics, the performance changes minimally, with a boost observed on Town13.

Variant Bench2Drive Longest6 v2 Town13
Full E2E TransFuser V6 95 ⟶ 94 62 ⟶ 62 5 ⟶ 10

Download the checkpoints:

# Download one checkpoint for testing
bash scripts/download_one_checkpoint.sh

# Download all checkpoints
git clone https://huggingface.co/ln2697/tfv6 outputs/checkpoints
cd outputs/checkpoints
git lfs pull

1.4. Evaluate model

Verify your setup with a single route:

# Start driving environment
bash scripts/start_carla.sh

# Turn on images and videos output
export LEAD_CLOSED_LOOP_CONFIG="produce_demo_image=true \
  produce_demo_video=true \
  produce_debug_image=true \
  produce_debug_video=true \
  produce_input_image=true \
  produce_input_video=true"

# Run policy on one route
python -m lead \
  --checkpoint outputs/checkpoints/tfv6_resnet34 \
  --routes data/benchmark_routes/bench2drive/23687.xml \
  --bench2drive

Driving logs are saved to outputs/local_evaluation/<route_id>/:

outputs/local_evaluation/<route_id>/
├── <route_id>_debug.mp4  # Debug visualization video
├── <route_id>_demo.mp4   # Demo video
├── <route_id>_grid.mp4   # Grid visualization video
├── <route_id>_input.mp4  # Raw input video
├── infractions.json      # Detected infractions, needed for the webapp
├── metric_info.json      # Evaluation metrics, needed for final scoring
├── debug_images/         # Per-frame debug visualizations
├── demo_images/          # Per-frame demo images
├── grid_images/          # Per-frame grid visualizations
├── input_images/         # Per-frame raw inputs
└── input_log/            # Input log data, useful for debugging

Tip

If run into OOM issue, there are few options:

  1. Run those commands block the later 2 seeds from being loaded into memory for ensembling:
mv outputs/checkpoints/tfv6_resnet34/model_0030_1.pth \
outputs/checkpoints/tfv6_resnet34/_model_0030_1.pth

mv outputs/checkpoints/tfv6_resnet34/model_0030_2.pth \
outputs/checkpoints/tfv6_resnet34/_model_0030_2.pth
  1. For local computer, start CARLA with -quality-level=Poor. This reduces the rendering quality, however will introduce distribution shift and should not be used for official evaluation.

1.5. Start Webapp

Launch the interactive dashboard to analyze driving failures - especially useful for Longest6 v2 or Town13 where iterating over evaluation logs is time-consuming:

python lead/webapp/app.py

Navigate to http://localhost:5000 and point it at your evaluation output directory (e.g. outputs/evaluation). The app discovers the three-level hierarchy automatically:

outputs/evaluation/
└── <experiment>/                # experiment
    └── <benchmark + seed>/      # benchmark + seed
        └── <timestamp> /        # timestamp
            ├── <route_id> /
            │   ├── infractions.json
            │   ├── <route_id>_debug.mp4
            │   ├── <route_id>_demo.mp4
            │   └── <route_id>_grid.mp4
            └── checkpoint_endpoint.json

Each route folder contains an infractions.json with per-step violation records and optional video files (_debug, _demo, _grid). The webapp reads these and provides:

  • Video playback with speed control (0.5x-5x) and frame-accurate seeking by time, frame number, or CARLA step.
  • Click-to-jump: click any infraction in the sidebar to seek to that moment, with a configurable offset (-1s, -3s, -5s) so you see the lead-up.
  • Clip extraction: cut a short clip around any infraction via FFmpeg, useful for sharing or reporting.
  • Filters: show only routes with infractions, or search by infraction type (collision, red light, off-road, etc.).

Tip

The app supports browser bookmarking - URLs encode the directory, route, video timestamp, playback speed, and offset, so you can share a link that jumps directly to a specific failure.

2. CARLA Research Cycle

The primary focus of this repository is solving the original CARLA Leaderboard 2.0. This section walks through the full research loop - collecting expert demonstrations, training a TFv6 policy, and benchmarking it closed-loop.

2.1. (Optional) Extending Data Routes

Before collecting, you may want to enlarge or diversify the route set under data/data_routes/. This step is optional - the shipped routes are enough to reproduce the paper results. However, if you want to improve the performance of the model, in particular for Longest6 v2 or Fail2Drive, introducing more routes is the easiest way to achieve this.

Two ways to do it:

  • Automatic - sample routes programmatically from a CARLA town (e.g. random start/goal pairs with scenario annotations). Useful when you want large-scale coverage without hand-authoring.
  • Manual - use the bundled carla_route_generator, a GUI tool for clicking waypoints on a map and exporting routes. Launch it via the hotkey script:
cd 3rd_party/carla_route_generator
python3 scripts/window.py

Generated XML files can be dropped directly into data/data_routes/ and picked up by the expert during data collection.

Tip

Out of the box, carla_route_generator is purely mouse-driven (left-click to add/remove waypoints, right-click for scenarios, wheel to pan/zoom). Annotating hundreds of routes this way is slow. There are few tricks to accelerate the process:

  1. We strongly recommend extending scripts/window.py with Qt QShortcut / keyPressEvent bindings for the actions you repeat most - e.g. add new route. Even one or two of keys cuts manual annotation time substantially.
  2. Only annotate route manually, add scenarios automatically via Python.

2.2. Obtaining Expert Demonstrations

Option A - Download the pre-collected dataset. Sufficient to reproduce the paper results:

# Download all routes
git clone https://huggingface.co/datasets/ln2697/lead data/carla_leaderboard2/zip
cd data/carla_leaderboard2/zip
git lfs pull

# Or download a single route for testing
bash scripts/download_one_route.sh

# Unzip the routes
bash scripts/unzip_routes.sh

Option B - Run the rule-based expert driver. Use this if you extended the route set, modified the expert, or changed the data format. With CARLA running, collect data for a single route via Python (recommended for debugging):

python -m lead \
  --expert \
  --routes data/data_routes/lead/noScenarios/short_route.xml

Or via bash (recommended for flexibility):

bash scripts/eval_expert.sh

Collected data is saved to outputs/expert_evaluation/ with the following structure:

├── bboxes/                  # Per-frame 3D bounding boxes for all actors
├── depth/                   # Compressed and quantized depth maps
├── depth_perturbated        # Depth from a perturbated ego state
├── hdmap/                   # Ego-centric rasterized HD map
├── hdmap_perturbated        # HD map aligned to perturbated ego pose
├── lidar/                   # LiDAR point clouds
├── metas/                   # Per-frame metadata and ego state
├── radar/                   # Radar detections
├── radar_perturbated        # Radar detections from perturbated ego state
├── rgb/                     # Front-facing RGB images
├── rgb_perturbated          # RGB images from perturbated ego state
├── semantics/               # Semantic segmentation maps
├── semantics_perturbated    # Semantics from perturbated ego state
└── results.json             # Route-level summary and evaluation metadata

On a SLURM Cluster of 92 GTX 1080ti, the data collection is often finished after 2 days.

Tip

  1. To configure camera/lidar/radar calibration, see config_base.py and config_expert.py.
  2. For large-scale collection on SLURM, see the data collection docs.
  3. The Jupyter notebooks provide visualization examples.
  4. The expert performs full 3D raycasting, so data collection speed scales inversely with the number and resolution of video cameras.

2.3. Training

Before training, build the data cache. This preprocesses the raw routes into an optimized format for the data loader, significantly speeding up training:

python scripts/build_cache.py

Perception pretraining. Logs and checkpoints are saved to outputs/local_training/pretrain:

# Single GPU
python3 lead/training/train.py \
  logdir=outputs/local_training/pretrain

# Distributed Data Parallel
bash scripts/pretrain_ddp.sh

Planning post-training. Logs and checkpoints are saved to outputs/local_training/posttrain:

# Single GPU
python3 lead/training/train.py \
  logdir=outputs/local_training/posttrain \
  load_file=outputs/local_training/pretrain/model_0030.pth \
  use_planning_decoder=true

# Distributed Data Parallel
bash scripts/posttrain_ddp.sh

Tip

  1. For distributed training on SLURM, see the SLURM training docs.
  2. For a complete workflow (pretrain → posttrain → eval), see this example.
  3. For detailed documentation, see the training guide.

2.4. Benchmarking

With CARLA running, evaluate on any benchmark via Python:

python -m lead \
  --checkpoint outputs/checkpoints/tfv6_resnet34 \
  --routes <ROUTE_FILE> \
  [--bench2drive]
Benchmark Route file Extra flag
Bench2Drive data/benchmark_routes/bench2drive/23687.xml --bench2drive
Longest6 v2 data/benchmark_routes/longest6/00.xml -
Town13 data/benchmark_routes/Town13/0.xml -
Fail2Drive data/benchmark_routes/fail2drive/Base_Animals_0075.xml --fail2drive

Or via bash:

bash scripts/eval_bench2drive.sh   # Bench2Drive
bash scripts/eval_longest6.sh      # Longest6 v2
bash scripts/eval_town13.sh        # Town13
bash scripts/eval_fail2drive.sh    # Fail2Drive (requires CARLA_F2D)

Results are saved to outputs/local_evaluation/ with videos, infractions, and metrics.

Tip

  1. See the evaluation docs for details.
  2. For distributed evaluation, see the SLURM evaluation docs.
  3. Our SLURM wrapper supports WandB for reproducible benchmarking.

3. Extensions

Beyond the core Leaderboard 2.0 workflow, LEAD also supports additional benchmarks (Fail2Drive, NAVSIM), an alternative RL policy (CaRL), and an alternative data format (123D). Each extension plugs into the code base with minimal changes.

3.1. Fail2Drive Evaluation

Fail2Drive (Gerstenecker et al., 2026) is a CARLA Leaderboard 2 benchmark for testing closed-loop generalization on unseen long-tail scenarios.

  • 200 short routes (avg. 219 m) across Town 13
  • 17 novel scenario classes in four generalization categories (see table below)
  • Each generalization route is paired with an in-distribution counterpart (same road geometry and traffic), isolating the effect of the distribution shift
  • Reports Driving Score (DS), Success Rate (SR), and their harmonic mean (HM)

Setup. Download the Fail2Drive simulator (custom CARLA build with novel assets):

mkdir -p 3rd_party/CARLA_F2D
curl -L https://hf.co/datasets/SimonGer/Fail2Drive/resolve/main/fail2drive_simulator.tar.gz \
  | tar -xz -C 3rd_party/CARLA_F2D

Evaluate. With CARLA_F2D running, evaluate on a single route:

# Start the Fail2Drive CARLA simulator
bash 3rd_party/CARLA_F2D/CarlaUE4.sh

# Evaluate model on one route
LEAD_CLOSED_LOOP_CONFIG="sensor_agent_creeping=True \
  use_kalman_filter=True \
  slower_for_stop_sign=True" \
python -m lead \
  --checkpoint outputs/checkpoints/tfv6_regnety \
  --routes data/benchmark_routes/fail2drive/Generalization_Animals_1075.xml \
  --fail2drive

Route files follow the naming convention {Base,Generalization}_{ScenarioClass}_{id}.xml and live in data/benchmark_routes/fail2drive/. Base_* routes are in-distribution; Generalization_* routes introduce the targeted shift.

Tip

For SLURM evaluation, see this example

Results.

Method Bench2Drive Fail2Drive In-Distribution Fail2Drive Generalization
DS ↑ DS ↑ SR(%) ↑ HM ↑ DS ↑ SR(%) ↑ HM ↑
TCP 59.9 24.7 39.1 30.3 24.5 (-0.8%) 31.4 (-19.7%) 27.5 (-9.1%)
UniAD 45.8 47.5 36.3 41.2 44.0 (-7.4%) 27.6 (-24.0%) 33.9 (-17.6%)
Orion 77.8 53.0 52.0 52.5 51.2 (-3.4%) 46.0 (-11.5%) 48.5 (-7.7%)
HiP-AD 86.8 74.1 70.7 72.4 67.1 (-9.4%) 56.7 (-19.8%) 61.5 (-15.1%)
SimLingo 85.1 82.6 79.3 80.9 71.7 (-13.2%) 55.0 (-30.6%) 62.2 (-23.1%)
TFv5 84.2 83.3 78.5 80.8 75.4 (-9.5%) 61.1 (-22.2%) 67.5 (-16.5%)
TFv6 (Ours) 95.2 90.2 93.3 91.7 79.5 (-11.9%) 70.7 (-24.2%) 74.8 (-18.4%)

3.2. CaRL Agent Evaluation

CaRL (Jaeger et al., 2025) is an RL-based privileged planner trained with PPO using a simple route-completion reward. It operates on a bird's-eye-view semantic segmentation input and outputs vehicle actions via a small CNN.

LEAD ships an inference-only port of CaRL, allowing it to be evaluated on any Leaderboard 2.0 benchmark alongside TFv6.

Note

CaRL was trained exclusively on CARLA towns 01-10, so Longest6 v2 is its natural benchmark. Performance on Bench2Drive (towns 01-15) and Town13 is mostly zero-shot.

Setup. The CaRL checkpoint is included in the model repository. If you followed the download instructions, it is already at outputs/checkpoints/CaRL/.

Evaluate. With CARLA running, evaluate the CaRL agent via Python:

CUBLAS_WORKSPACE_CONFIG=:4096:8 \
python -m lead \
  --checkpoint outputs/checkpoints/CaRL \
  --routes data/benchmark_routes/bench2drive/24240.xml \
  --carl \
  --bench2drive \
  --timeout 900

Or via bash:

bash scripts/eval_carl.sh

The results are in outputs/local_evaluation/<route_id>/.

Tip

  1. With small code changes, you can also integrate CaRL into LEAD's expert-driving pipeline as a hybrid expert policy.
  2. For large scale evaluation on SLURM, see this directory.

3.3. PlanT 2.0 Training and Evaluation

PlanT 2.0 (Gerstenecker et al., 2025) is a privileged object-centric planning transformer. Instead of processing raw sensor data, it operates directly on ground-truth bounding boxes from CARLA, isolating the planning problem from perception. This makes it a powerful tool for accelerating the research cycle:

  • Fast iteration - trains in a fraction of the time of sensor-based models (no image/LiDAR processing), enabling rapid prototyping of planning ideas, loss functions, and data strategies.
  • Controlled experiments - object-level inputs can be easily perturbed (e.g. shifting vehicles, adding pedestrians, removing obstacles) to systematically study planning behavior and failure modes.
  • Upper-bound analysis - by removing perception noise, PlanT reveals the ceiling of planning performance for a given dataset and training setup, helping diagnose whether failures stem from perception or planning.

PlanT replaces TFv6's image+LiDAR backbone with a BERT-based encoder that processes object tokens (vehicles, pedestrians, traffic lights, etc.), while reusing LEAD's planning decoder, training loop, and evaluation infrastructure. Our implementation closely follows the original PlanT, but since we use a different expert, some hyperparameter tuning may be required to fully recover its performance.

Important

PlanT is a privileged planner - it uses ground-truth bounding boxes from CARLA's MAP track, not sensor observations. It is designed for research analysis (ablations, debugging, upper-bound estimation), not as a deployable sensor-based driving system.

Prepare data. Since PlanT requires some additional data, it can be trained only on LEAD360 or newer dataset versions. Download the data:

# Download all routes
git clone https://huggingface.co/datasets/ln2697/lead360 data/carla_leaderboard2_360/zip
cd data/carla_leaderboard2_360/zip
git lfs pull

After that, unzip the zip files as similar to scripts/unzip_routes.sh.

Training. PlanT uses the same CARLA data as TFv6. Set model_type=plant to switch the model before training:

# Single GPU
python3 lead/training/train.py \
  model_type=plant \
  logdir=outputs/local_training/plant

# Distributed Data Parallel
bash scripts/train_plant.sh

Evaluate. With CARLA running, evaluate the PlanT agent on MAP track:

python -m lead \
  --checkpoint outputs/local_training/plant \
  --routes data/benchmark_routes/bench2drive/23687.xml \
  --plant \
  --bench2drive

Tip

For large scale experiments on SLURM, see this directory.

3.4. NAVSIM Training and Evaluation

NAVSIM (Dauner et al., 2024) is a non-reactive simulation benchmark that evaluates vision-based driving policies on real-world sensor data with simulation-based metrics (collisions, drivable-area compliance, progress).

LEAD provides a training pipeline for this benchmark.

Stage Owner What happens
Data setup NAVSIM Raw sensor logs, scene caching, metric caching
Training LEAD Loads cached features, runs optimization and checkpointing
Evaluation NAVSIM Scores predictions via its own harnesses

Data setup. Install the three data splits:

NAVSIM handles all raw data processing; only the resulting caches are needed for training.

Training. Once the NAVSIM caches exist, LEAD loads the cached transfuser_feature.gz / transfuser_target.gz pairs through lead/data_loader/navsim_dataset.py and trains in two stages:

  1. Perception pretraining (script), one seed
  2. Planning post-training (script), three seeds to estimate variance

All optimization, checkpointing, and logging runs entirely within LEAD.

Evaluation. Evaluation runs through NAVSIM's own harnesses on navtest and navhard.

The bridge between the two codebases is CarlaTransfuserAgent (navsimv1.1, navsimv2.2). It implements NAVSIM's AbstractAgent interface but internally wraps LEAD's OpenLoopInference, reloading the trained checkpoint and translating NAVSIM's feature dict into LEAD's inputs. The agent is inference-only.

3.5. CARLA 123D Data Collection

123D is an open-source library that unifies diverse driving datasets into a single, lightweight framework.

  • Storage: Apache Arrow IPC files - one file per modality, each an independent timestamped event stream (Driving Log)
  • Modalities: cameras, LiDAR, 3D annotations, HD maps
  • Coordinate conventions: ISO 8855 (vehicle/body frames), OpenCV (camera frames)
  • Sensor model: synchronous and asynchronous sensors handled uniformly

LEAD integrates 123D as its data collection format for CARLA. With CARLA running, collect data in 123D format via Python:

export LEAD_EXPERT_CONFIG="target_dataset=6 \
  py123d_data_format=true \
  use_radars=false \
  lidar_stack_size=2 \
  save_only_non_ground_lidar=false \
  save_lidar_only_inside_bev=false"

python -m lead \
    --expert \
    --py123d \
    --routes data/data_routes/50x38_Town12/ParkingCrossingPedestrian/3250_1.xml

Or via bash:

bash scripts/eval_expert_123d.sh

Output in 123D format is saved to data/carla_leaderboard2_py123d/. Each route produces a directory under logs/carla_train/ with one Arrow file per modality:

File Content
ego_state_se3.arrow Ego vehicle pose and motion state
camera.pcam_{f,b,l,r}{0,1}.arrow Camera images (front, back, left, right)
lidar.lidar_top.arrow Top LiDAR point clouds
box_detections_se3.arrow 3D bounding box annotations
traffic_light_detections.arrow Traffic light states
sync.arrow Synchronization timestamps across modalities
maps/carla/*.arrow HD map - lanes, intersections, crosswalks, road edges, road lines in WKB geometry

The collected data can be loaded back via 123D's Scene API and Map API for training, evaluation, or analysis workflows. The Scene API provides declarative access to subsequences, history/future windows, and re-sampling across frequencies, while the Map API supports spatial queries over nearby map objects.

To visualize collected scenes in 3D with Viser:

python scripts/123d_viser.py

Tip

This feature is experimental. Change PY123D_DATA_ROOT in scripts/main.sh to set the output directory.

4. Project Structure

The project is organized into the following top-level directories. See the full documentation for a detailed breakdown.

Directory Purpose
lead/ Main package - model architecture, training, inference, expert driver
3rd_party/ Third-party dependencies (CARLA, benchmarks, evaluation tools)
data/ Route definitions. Sensor data will be stored here, too.
scripts/ Utility scripts for data processing, training, and evaluation
outputs/ Checkpoints, evaluation results, and visualizations
notebooks/ Jupyter notebooks for data inspection and analysis
slurm/ SLURM job scripts for large-scale experiments

We also provide a Claude Code skill to explore the repository interactively. Run /qa inside Claude Code to get a guided walkthrough of the codebase structure, key modules, and how they connect.

5. Common Issues

Issue Fix
Stale or corrupted data errors Delete and rebuild the training cache / buckets
Simulator hangs or is unresponsive Restart the CARLA simulator
Route or evaluation failures Restart the leaderboard
Training instability after PyTorch version update No fix for now. We tried to upgrade Torch several times but failed to achieve stable training on newer Torch versions.
OOM in evaluation Use larger GPU. In our submit_job utility function, we first attempt to use a smaller GPU partition (1080ti/2080ti). After some failures, we switch automatically to a partition with more VRAM.
Training instability in general Turn off mixed-precision training and train in 32bit precision.

Feel free to open a ticket for any problem.

Beyond CARLA: Cross-Benchmark Deployment

The LEAD pipeline and TFv6 models serve as reference implementations across multiple E2E driving platforms:

Platform Model Highlight
Waymo E2E Driving Challenge DiffusionLTF 2nd place in the inaugural vision-based E2E driving challenge
NAVSIM v1 Huggingface Leaderboard LTFv6 +3 PDMS over Latent TransFuser baseline on navtest
NAVSIM v2 Huggingface Leaderboard LTFv6 +6 EPMDS over Latent TransFuser baseline on navhard
NVIDIA AlpaSim TransFuserDriver Official baseline policy for closed-loop simulation

Further Documentation

For a deeper dive, visit the full documentation site:

Data Collection  ·  Training  ·  Evaluation.

The documentation will be updated regularly.

Acknowledgements

This project builds on the shoulders of excellent open-source work. Special thanks to carla_garage for the foundational codebase.

PDM-Lite  ·  Leaderboard  ·  Scenario Runner  ·  NAVSIM  ·  Waymo Open Dataset
SimLingo  ·  PlanT2  ·  Bench2Drive Leaderboard  ·  Bench2Drive  ·  CaRL  ·  Fail2Drive

Long Nguyen led development of the project. Kashyap Chitta, Bernhard Jaeger, and Andreas Geiger contributed through technical discussion and advisory feedback. Daniel Dauner provided guidance with NAVSIM.

Citation

If you find this work useful, please consider giving this repository a star and citing our paper:

@inproceedings{Nguyen2026CVPR,
	author = {Long Nguyen and Micha Fauth and Bernhard Jaeger and Daniel Dauner and Maximilian Igl and Andreas Geiger and Kashyap Chitta},
	title = {LEAD: Minimizing Learner-Expert Asymmetry in End-to-End Driving},
	booktitle = {Conference on Computer Vision and Pattern Recognition (CVPR)},
	year = {2026},
}

About

[CVPR26] LEAD: Minimizing Learner–Expert Asymmetry in End-to-End Driving

Topics

Resources

License

Contributing

Stars

Watchers

Forks

Contributors