UFO: Unifying Feed-Forward and Optimization-based Methods for Large Driving Scene Modeling
Organizations: Xiaomi EV · UIUC
Abstract
Dynamic driving scene reconstruction is critical for autonomous driving simulation and closed-loop learning. While recent feed-forward methods have shown promise for 3D reconstruction, they struggle with long-range driving sequences due to quadratic complexity in sequence length and challenges in modeling dynamic objects over extended durations. We propose UFO, a novel recurrent paradigm that combines the benefits of optimization-based and feed-forward methods for efficient long-range 4D reconstruction. Our approach maintains a 4D scene representation that is iteratively refined as new observations arrive, using a visibility-based filtering mechanism to select informative scene tokens and enable efficient processing of long sequences. For dynamic objects, we introduce an object pose-guided modeling approach that supports accurate long-range motion capture. Experiments on the Waymo Open Dataset demonstrate that our method significantly outperforms both per-scene optimization and existing feed-forward methods across various sequence lengths. Notably, our approach can reconstruct 16-second driving logs within 0.5 second while maintaining superior visual quality and geometric accuracy.
Figures & tables
| Methods | 2s Sequences | 8s Sequences | 16s Sequences ( zero-shot ) | ||||||
|---|---|---|---|---|---|---|---|---|---|
| PSNR | SSIM | D-RMSE | PSNR | SSIM | D-RMSE | PSNR | SSIM | D-RMSE | |
| 3DGS [ 19 ] | 21.07 | 0.578 | 13.52 | 19.57 | 0.517 | 14.42 | 17.18 | 0.454 | 17.01 |
| PVG [ 5 ] | 23.81 | 0.649 | 13.82 | 22.90 | 0.619 | 18.24 | 21.79 | 0.599 | 17.21 |
| Street Gaussians [ 46 ] | 22.96 | 0.652 | 12.15 | 21.69 | 0.609 | 13.17 | 22.67 | 0.675 | 14.88 |
| DeformableGS [ 51 ] | 23.40 | 0.669 | 11.55 | 21.47 | 0.611 | 13.12 | 19.79 | 0.600 | 15.82 |
| GS-LRM [ 57 ] | 25.18 | 0.753 | 7.94 | 21.81 | 0.584 | 7.37 | 16.98 | 0.500 | 9.81 |
| Method | PSNR | SSIM | D-RMSE |
|---|---|---|---|
| (1) w/o iterative training | 20.67 | 0.625 | 11.81 |
| (2) w/o scene tokens | 26.31 | 0.787 | 5.31 |
| (3) w/o refinement | 27.08 | 0.824 | 5.13 |
| (4) w/o lifespan | 26.47 | 0.790 | 5.21 |
| (5) w/o bounding box | 26.88 | 0.818 | 5.12 |
| Ours (full) | 27.39 | 0.830 | 5.10 |