Skip to content

phai-lab/OpenLongTail

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

OpenLongTail

Generative Scaling of Long-Tail Driving Data

Lulin Liu* · Nuo Chen* · Yan Wang · Bangya Liu · Wenyan Cong · Hezhen Hu
Boris Ivanovic · Hao Wang · Ziyao Zeng · Xinyu Gong · Yang Zhou · Zixiang Xiong
Dilin Wang · Zhangyang Wang · Weisong Shi · Ruohan Zhang · Marco Pavone · Zhiwen Fan

Texas A&M University · NVIDIA · University of Wisconsin–Madison · UT Austin · Yale University · Adobe · Meta · University of Delaware · Stanford University

*Equal contribution Corresponding author

TL;DR — From a single front-camera driving video, OpenLongTail synthesizes the five missing rig cameras at the same timestamps — turning monocular long-tail clips into pose-grounded, multi-view training data for robust VLA driving policies.

TODO

  • Checkpointwill be released in July 2026
  • Inference code
  • Preprocessing & training code

Table of Contents

Installation

# 1. Clone
git clone https://github.com/lulinliu/longt.git openlongtail
cd openlongtail

# 2. Create the environment (Python 3.10, CUDA, bf16)
conda create -n openlongtail python=3.10 -y
conda activate openlongtail

# 3. Install a CUDA-matched PyTorch, then the rest of the deps
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
pip install -r requirements.txt

Download the base backbone and the OpenLongTail checkpoint from Hugging Face in one step:

bash scripts/download.sh

Inference

Generate the 5 synchronized non-front views from a single front-camera clip. Each clip produces the 5 generated views, the front input, and a preview grid.

# raw dash-cam clips (front video only)
bash scripts/infer.sh --input path/to/front_clips --output outputs/openlongtail_infer

# already-preprocessed clips
bash scripts/infer.sh --input path/to/cache --output outputs/openlongtail_infer --cached

The rear views reuse the front camera's earlier view of the same scene and are steered by an automatically generated scene caption, so the synthesized surround stays consistent with the input. Model and sampler settings can be overridden via environment variables (see the top of scripts/infer.sh).

Building Training Data

Preprocessing turns raw front-camera clips into a ready-to-train multi-view cache in one step: it recovers a metric ego-trajectory for each clip and reprojects the front-view evidence into every target camera (same-time for the side views, temporal look-back for the rear views).

bash scripts/preprocess.sh --input path/to/raw_clips --output data/openlongtail_cache

# shard across GPUs (run i = 0..N-1 in parallel)
bash scripts/preprocess.sh --input path/to/raw_clips --output data/openlongtail_cache --shards 8 --shard 0

# if your source already provides camera poses, skip pose recovery
bash scripts/preprocess.sh --input path/to/raw_clips --output data/openlongtail_cache --have-poses

The resulting cache is consumed directly by training.

Training

# Single node (torchrun), 8 GPUs → outputs/openlongtail_train
LATENT_CACHE_ROOT=data/openlongtail_cache OUTPUT_DIR=outputs/openlongtail_train bash scripts/train.sh

# short run on fewer GPUs
NUM_GPUS=4 NUM_STEPS=5000 OUTPUT_DIR=outputs/smoke bash scripts/train.sh

# multi-node (SLURM)
sbatch scripts/train.slurm

train.sh auto-resumes from the latest checkpoint under the output directory.

Citation

@misc{liu2026openlongtailgenerativescalinglongtail,
      title={OpenLongTail: Generative Scaling of Long-Tail Driving Data}, 
      author={Lulin Liu and Nuo Chen and Yan Wang and Bangya Liu and Wenyan Cong and Hezhen Hu and Boris Ivanovic and Hao Wang and Ziyao Zeng and Xinyu Gong and Yang Zhou and Zixiang Xiong and Dilin Wang and Zhangyang Wang and Weisong Shi and Ruohan Zhang and Marco Pavone and Zhiwen Fan},
      year={2026},
      eprint={2607.09655},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2607.09655}, 
}

Acknowledgements

OpenLongTail builds on the following open-source projects:

We thank the authors of these projects for releasing their code and models. Please refer to the corresponding files under third_party/ for upstream licenses and notices.

License

Released under the MIT License; base models remain under their respective licenses.

About

OpenLongTail: Generative Scaling of Long-Tail Driving Data

Resources

License

Stars

27 stars

Watchers

1 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors