Upload 4 files
Browse files- README.md +93 -3
- cluster_points_8192_.npy +3 -0
- frame_mapping_navhard.yaml +0 -0
- global_pose_navhard.npy +3 -0
README.md
CHANGED
|
@@ -1,3 +1,93 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: apache-2.0
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
tags:
|
| 6 |
+
- autonomous-driving
|
| 7 |
+
- end-to-end-driving
|
| 8 |
+
- diffusion-model
|
| 9 |
+
- trajectory-planning
|
| 10 |
+
- uncertainty-estimation
|
| 11 |
+
- navsim
|
| 12 |
+
pretty_name: Mimir Data
|
| 13 |
+
size_categories:
|
| 14 |
+
- 100M<n<1G
|
| 15 |
+
---
|
| 16 |
+
|
| 17 |
+
# Mimir: Hierarchical Goal-Driven Diffusion with Uncertainty Propagation for End-to-End Autonomous Driving
|
| 18 |
+
|
| 19 |
+
> [**Mimir: Hierarchical Goal-Driven Diffusion with Uncertainty Propagation for End-to-End Autonomous Driving**](https://arxiv.org/pdf/2512.07130)
|
| 20 |
+
>
|
| 21 |
+
> Zebin Xing, Yupeng Zheng, Qichao Zhang, Zhixing Ding, Pengxuan Yang, Songen Gu, Zhongpu Xia, Dongbin Zhao
|
| 22 |
+
>
|
| 23 |
+
> Institute of Automation, Chinese Academy of Sciences; China University of Geosciences; Fudan University
|
| 24 |
+
>
|
| 25 |
+
> IEEE Robotics and Automation Letters (RAL), 2025
|
| 26 |
+
|
| 27 |
+
**Mimir** is a hierarchical goal-driven diffusion model for end-to-end autonomous driving. It improves upon **GoalFlow** by explicitly modeling goal uncertainty and accelerating goal inference, enabling more robust and efficient planning. With a ResNet-34 backbone, Mimir achieves 89.3 PDMS and 34.6 EPDMS, demonstrating strong performance in both accuracy and efficiency.
|
| 28 |
+
|
| 29 |
+
This repository contains the **model weights and intermediate data files** required for Mimir's evaluation and training. For the full codebase, please visit the [main Mimir repository](https://github.com/ZebinX/Mimir-Uncertainty-Driving).
|
| 30 |
+
|
| 31 |
+
---
|
| 32 |
+
|
| 33 |
+
## 📦 Repository Structure
|
| 34 |
+
|
| 35 |
+
| Path | Description |
|
| 36 |
+
|:---|:---|
|
| 37 |
+
| `ckpts/mimir_epoch94.ckpt` | Mimir planning model (~746MB) |
|
| 38 |
+
| `ckpts/mimir_unc_epoch99.ckpt` | Uncertainty modeling model (~724MB) |
|
| 39 |
+
| `goalflow_goal/` | GoalFlow goal points for `mimir_unc` training initialization (navhard/navtest/navtrain) |
|
| 40 |
+
| `goalflow_dac/` | Predicted DAC using GoalFlow (navhard/navtest/navtrain) |
|
| 41 |
+
| `navhard_naviunc/` | Navigation & uncertainty data for Navhard |
|
| 42 |
+
| `navstest_naviunc/` | Navigation & uncertainty data for Navtest |
|
| 43 |
+
| `navtrain_naviunc/` | Navigation & uncertainty data for Navtrain |
|
| 44 |
+
| `cluster_points_8192_.npy` | 8192 cluster centers for goal point discretization |
|
| 45 |
+
| `frame_mapping_navhard.yaml` | Frame index mapping for Navhard |
|
| 46 |
+
| `global_pose_navhard.npy` | Global pose data for Navhard |
|
| 47 |
+
|
| 48 |
+
> Each `*_naviunc/` directory contains `navi.npy` (navigation points) and `unc.npy` (uncertainty estimates).
|
| 49 |
+
|
| 50 |
+
|
| 51 |
+
|
| 52 |
+
## 📊 Results
|
| 53 |
+
|
| 54 |
+
### Navtest (NAVSIM v1)
|
| 55 |
+
|
| 56 |
+
| Method | NC↑ | DAC↑ | TTC↑ | EP↑ | PDMS↑ |
|
| 57 |
+
|:---|:---:|:---:|:---:|:---:|:---:|
|
| 58 |
+
| GoalFlow‡ | 98.2 | 96.4 | 93.8 | 82.6 | 87.9 |
|
| 59 |
+
| DiffusionDrive | 98.2 | 96.2 | 94.7 | 82.2 | 88.1 |
|
| 60 |
+
| **Mimir** | **98.2** | **97.5** | 94.6 | **83.6** | **89.3** |
|
| 61 |
+
|
| 62 |
+
### Navhard (NAVSIM v2)
|
| 63 |
+
|
| 64 |
+
| Method | EPDMS↑ |
|
| 65 |
+
|:---|:---:|
|
| 66 |
+
| GoalFlow† | 28.9 |
|
| 67 |
+
| DiffusionDrive† | 29.1 |
|
| 68 |
+
| **Mimir** | **34.6** |
|
| 69 |
+
|
| 70 |
+
Please refer to the [paper](https://arxiv.org/pdf/2512.07130) for full comparison tables.
|
| 71 |
+
|
| 72 |
+
---
|
| 73 |
+
|
| 74 |
+
## 📄 Citation
|
| 75 |
+
|
| 76 |
+
If you find Mimir useful, please consider citing our paper:
|
| 77 |
+
|
| 78 |
+
```bibtex
|
| 79 |
+
@ARTICLE{11282450,
|
| 80 |
+
author={Xing, Zebin and Zheng, Yupeng and Zhang, Qichao and Ding, Zhixing and Yang, Pengxuan and Gu, Songen and Xia, Zhongpu and Zhao, Dongbin},
|
| 81 |
+
journal={IEEE Robotics and Automation Letters},
|
| 82 |
+
title={Mimir: Hierarchical Goal-Driven Diffusion With Uncertainty Propagation for End-to-End Autonomous Driving},
|
| 83 |
+
year={2026},
|
| 84 |
+
volume={11},
|
| 85 |
+
number={2},
|
| 86 |
+
pages={2178-2185},
|
| 87 |
+
keywords={Uncertainty;Trajectory;Predictive models;Autonomous vehicles;Laser radar;Vocabulary;Planning;Feature extraction;Estimation;Artificial intelligence;Learning from demonstration;imitation learning;autonomous vehicle navigation},
|
| 88 |
+
doi={10.1109/LRA.2025.3641129}}
|
| 89 |
+
```
|
| 90 |
+
|
| 91 |
+
## Contact
|
| 92 |
+
|
| 93 |
+
If you have any questions or suggestions, please open an issue or contact: xzebin@bupt.edu.cn
|
cluster_points_8192_.npy
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a08a1c7a897db771c88c98d42a0bf448074936c858f0b50536918ee50cfb66a0
|
| 3 |
+
size 196736
|
frame_mapping_navhard.yaml
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
global_pose_navhard.npy
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:97bd2877033075842d638b5ec3f48d7f9fda325ed22b00f231aa60101006f0fc
|
| 3 |
+
size 619941
|