Scene-Adaptive Readout and Decoding in Diffusion Driving VLAs
Organizations: Korea Advanced Institute of Science and Technology (KAIST)
Abstract
Diffusion driving vision-language-action models repeatedly execute the backbone at every denoising stage, incurring substantial inference cost, while it remains unclear whether all scenes require the same decoding computation. To distinguish when a trajectory becomes readable from how much further refinement it requires, we analyze trajectory readout across network depth and decoding progress. We identify a depth-decoding dissociation in which late-layer representations already support useful trajectory readout after the first forward sweep, whereas early-layer representations benefit more from continued decoding. We further find that both the preferred readout layer and decoding depth vary across scenes. Based on these observations, we propose a scene-adaptive readout and decoding framework. A scene-conditioned readout MoE adaptively weights and combines trajectory predictions from layer-specific late-layer representations after a single backbone sweep, while an adaptive decoding policy applies additional native refinement only when required by the scene. We evaluate our framework on two diffusion driving VLAs across the Waymo E2E and nuScenes benchmarks. Our method improves validation RFS by up to 0.56 while also improving test performance, and achieves an end-to-end speedup of up to on Waymo E2E, while consistently reducing trajectory error on nuScenes. These results demonstrate that scene-level readout and decoding allocation improves planning quality while avoiding unnecessary decoding. Our code is available at https://github.com/sjyu001/SARD
Figures & tables
| Model | Backbone | Validation | Test | Latency | ||||
|---|---|---|---|---|---|---|---|---|
| RFS | ADE@3s | ADE@5s | RFS | ADE@3s | ADE@5s | ms/scene | ||
| Autoregressive VLA baselines | ||||||||
| OpenEMMA ( Xing et al., 2025 ) | AR | 6.698 | 2.725 | 4.957 | 5.158 | 6.684 | 12.476 | 22,291 |
| LightEMMA ( Qiao et al., 2025 ) | AR | 6.670 | 2.595 | 4.951 | 6.517 | 1.705 | 3.740 | 16,414 |
| AutoVLA ( Zhou et al., 2025 ) | AR | – | – | – | 7.557 | 1.351 | 2.958 | 7,855 |
| Poutine ( Rowe et al., 2025 ) | AR | 8.120 | – | – | 7.909 | 1.272 | 2.941 | – |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Configuration | Fast-dDrive | dVLM-AD |
|---|---|---|
| Backbone and features | ||
| Language model / hidden size | Qwen2.5-VL-3B / 2,048 | LLaDA-V-8B / 4,096 |
| Native decoding | Block diffusion; stages | Semi-autoregressive; block size 512 |
| Trajectory slots per scene | 100 | |
| Readout heads | ||
| Training epochs / batch size | 60 / 256 | 60 / 64 |
| Readout layer | L21 | L22 | L23 | L24 | L25 | L26 | L27 | L28 | L29 | |
| Fast-dDrive | val | 0.2 | 0.0 | 2.1 | 0.0 | 0.0 | 0.4 | 94.4 | 0.4 | 2.5 |
| Fast-dDrive | test | 0.4 | 0.0 | 2.5 | 0.0 | 0.0 | 0.6 | 93.8 | 0.7 | 2.1 |
| dVLM-AD | val | 0.6 | 0.4 | 3.2 | 0.4 | 0.5 | 1.3 | 89.7 | 0.9 | 3.1 |
| dVLM-AD | test | 0.9 | 0.4 | 3.8 | 0.5 | 0.6 | 1.3 | 88.0 | 1.4 | 3.0 |
| Decoding stage | Readout | 1 | 2 | 3 | 4 | 5 | end | – | ||
| Fast-dDrive | val | 91.6 | 5.2 | 0.8 | 1.3 | 0.4 | 0.0 | 0.6 | – | |
Vision-Language-Action Models
Vision-language-action models turn visual observations and natural-language instructions into actions for robots and other embodied agents.
See all 1,202 papers