awskhalil commited on
Commit
8d49c88
·
verified ·
1 Parent(s): 939186e

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +125 -2
README.md CHANGED
@@ -2,9 +2,8 @@
2
  license: mit
3
  language:
4
  - en
5
-
6
  library_name: pytorch
7
-
8
  tags:
9
  - teleoperation
10
  - autonomous-driving
@@ -15,4 +14,128 @@ tags:
15
  - pytorch
16
  - predictive-display
17
  - future-action-prediction
 
 
18
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  license: mit
3
  language:
4
  - en
 
5
  library_name: pytorch
6
+ pipeline_tag: image-to-video
7
  tags:
8
  - teleoperation
9
  - autonomous-driving
 
14
  - pytorch
15
  - predictive-display
16
  - future-action-prediction
17
+ datasets:
18
+ - bimilab/TeleopWM-Dataset
19
  ---
20
+
21
+ # TeleopWM
22
+
23
+ TeleopWM is a lightweight predictive latent world model for latency-resilient vision-based teleoperation. Given recent RGB observations and teleoperation control history, it predicts short-horizon future visual observations and future longitudinal/steering trends for predictive display.
24
+
25
+ ## Paper and Project Links
26
+
27
+ - Project page: https://bimilab.github.io/paper-TeleopWM/
28
+ - GitHub repository: https://github.com/bimilab/paper-TeleopWM
29
+ - Paper: TBD
30
+ - Dataset: https://huggingface.co/datasets/bimilab/TeleopWM-Dataset
31
+ - YouTube demo: https://youtu.be/WeKqqZuwBl0
32
+
33
+ ## Model Description
34
+
35
+ TeleopWM predicts 8 future RGB frames and future longitudinal/steering trends from recent visual observations and teleoperation control history. The model uses a SimVP visual backbone together with a TeleopWM latent dynamics branch, and is designed for real-time predictive display under teleoperation latency.
36
+
37
+ The checkpoint was trained and evaluated on CARLA/MILE-style driving rollouts. TeleopWM is intended as a compact research model for short-horizon predictive continuity, not as an open-ended video generation or autonomous-driving foundation model.
38
+
39
+ ## Architecture
40
+
41
+ TeleopWM combines a SimVP visual backbone with a lightweight latent dynamics module and a motion-aware future action prediction head. The model jointly predicts future visual observations and future driving actions within a unified predictive framework designed for latency-resilient teleoperation.
42
+
43
+ ![TeleopWM Method](TeleopWM-Method-v2.svg)
44
+
45
+ ## Intended Use
46
+
47
+ - Research on latency-resilient vision-based teleoperation
48
+ - Predictive display under communication latency
49
+ - Short-horizon future observation prediction
50
+ - Future action trend prediction
51
+ - CARLA/MILE-style driving rollout analysis
52
+
53
+ ## Out-of-Scope Use
54
+
55
+ - Safety-critical autonomous driving deployment without validation
56
+ - Open-ended video generation
57
+ - Direct real-vehicle deployment without additional testing
58
+ - General-purpose world modeling outside the evaluated driving domain
59
+
60
+ ## Files
61
+
62
+ - `best.pt` — final TeleopWM paper checkpoint
63
+ - `config.json` — training/evaluation configuration associated with the checkpoint
64
+ - `benchmark.json` — runtime benchmark summary
65
+ - `future_action_eval.png` — future action evaluation figure
66
+ - `main_rollout_action_figure_final.png` — qualitative rollout/action alignment figure
67
+
68
+ ## Results Summary
69
+
70
+ | Category | Metric | Value |
71
+ |---|---:|---:|
72
+ | Rollout prediction | Horizon | 8 frames / approximately 533 ms at 15 FPS |
73
+ | Future action prediction | Outputs | longitudinal and steering trends |
74
+ | Runtime | Inference latency | 38.9 ms / rollout |
75
+ | Runtime | Prediction rate | 205.5 FPS |
76
+ | Runtime | Peak VRAM | 1.24 GB |
77
+ | Resolution | Input/output | 320x512 |
78
+
79
+ Runtime values are reference measurements from the final paper configuration and should be re-measured on target hardware.
80
+
81
+ ## Qualitative Rollout Example
82
+
83
+ ![TeleopWM qualitative rollout results](main_rollout_action_figure_final.png)
84
+
85
+ Representative 8-step future RGB rollouts and action alignment across straight, mild-turn, sharp-turn, and intersection scenarios.
86
+
87
+ ## Future Action Prediction
88
+
89
+ ![TeleopWM future action evaluation](future_action_eval.png)
90
+
91
+ Per-step future action error and correlation for longitudinal and steering predictions.
92
+
93
+ ## Usage
94
+
95
+ Download the checkpoint and config:
96
+
97
+ ```bash
98
+ huggingface-cli download bimilab/TeleopWM \
99
+ best.pt config.json \
100
+ --local-dir checkpoints/TeleopWM
101
+ ```
102
+
103
+ Then evaluate using the TeleopWM repository:
104
+
105
+ ```bash
106
+ python scripts/evaluate_teleopwm.py \
107
+ --checkpoint checkpoints/TeleopWM/best.pt \
108
+ --data-root /path/to/mile_action_diverse/test/Town05 \
109
+ --split test \
110
+ --sample-strategy uniform \
111
+ --max-samples 64 \
112
+ --device cuda
113
+ ```
114
+
115
+ Runtime benchmarking:
116
+
117
+ ```bash
118
+ python scripts/benchmark_teleopwm.py \
119
+ --checkpoint checkpoints/TeleopWM/best.pt \
120
+ --device cuda \
121
+ --batch-size 1 \
122
+ --warmup 20 \
123
+ --iters 200
124
+ ```
125
+
126
+ ## Citation
127
+
128
+ If you use TeleopWM, please cite:
129
+
130
+ ```bibtex
131
+ @misc{teleopwm2026,
132
+ title={TeleopWM: A Real-Time Predictive World Model for Latency-Resilient Vision-Based Teleoperation},
133
+ author={Khalil, Aws and Kwon, Jaerock},
134
+ year={2026},
135
+ note={Under review}
136
+ }
137
+ ```
138
+
139
+ ## License
140
+
141
+ This model is released under the MIT License.