https://github.com/sisl/interpretableselfawareprediction
https://github.com/sisl/interpretableselfawareprediction
Last synced: 3 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/sisl/interpretableselfawareprediction
- Owner: sisl
- License: apache-2.0
- Created: 2022-05-23T04:31:12.000Z (about 4 years ago)
- Default Branch: main
- Last Pushed: 2022-12-16T00:15:03.000Z (over 3 years ago)
- Last Synced: 2024-03-24T17:10:24.040Z (about 2 years ago)
- Language: Python
- Size: 39.1 KB
- Stars: 15
- Watchers: 8
- Forks: 1
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Interpretable Self-Aware Prediction (ISAP)
We propose the use of evidential deep learning to perform one-shot epistemic uncertainty estimation over a low-dimensional, interpretable latent space in a trajectory prediction setting. This code runs the qualitative and quantitative experiments to validate the proposed Interpretable Self-Aware Prediction (ISAP) framework.
See our [paper](https://arxiv.org/abs/2211.08701) for more details:
M. Itkina and M. J. Kochenderfer. "Interpretable Self-Aware Neural Networks for Robust Trajectory Prediction". In Conference on Robot Learning (CoRL), 2022.
Qualitative results for our ISAP framework on in-distribution (ID) and out-of-distribution (OOD) examples for the input trajectory experiment. We see that the ID example (left) has a slower moving agent of interest (red history boxes closer together) than the OOD example. Thus, ISAP learns the epistemic uncertainty in the agent behavior latent variable to be higher (α0,agent is lower) for the OOD case than the ID case.
## Instructions
The required dependencies are listed in `dependencies.txt`.
The NuScenes trajectory prediction dataset has to be downloaded from: https://www.nuscenes.org/nuscenes#download and placed into the `data/nuscenes/` folder, including a `covernet_traj_set` containing the trajectory sets, `maps` directory, and `v1.0-trainval` data. The NuScenes github repository: https://github.com/nutonomy/nuscenes-devkit should be cloned and the `nuscenes-devkit` folder to be placed at the top-level.
The PostNet code should be cloned from: https://github.com/sharpenb/Posterior-Network and placed at the top-level.
This code was developed and tested with `Python 3.6.12`.
To replicate the input trajectory speed experiments, please run the following files:
``run_isap_agent_speed.sh``
``run_postcovernet_agent_speed.sh``
``run_ensembles_agent_speed.sh``