Learning A Unified Risk Map for Autonomous Driving in Partially Observable Environments

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Jia, Jie, Su, Yaofeng, Bao, Zeyu, Hong, Yun, Gao, Bingzhao, Gan, Zhongxue, Ding, Wenchao
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916035605561344
author Jia, Jie
Su, Yaofeng
Bao, Zeyu
Hong, Yun
Gao, Bingzhao
Gan, Zhongxue
Ding, Wenchao
author_facet Jia, Jie
Su, Yaofeng
Bao, Zeyu
Hong, Yun
Gao, Bingzhao
Gan, Zhongxue
Ding, Wenchao
contents Occlusion-aware prediction remains a critical challenge in autonomous driving due to the inherent uncertainty of unobserved regions. Existing approaches either overestimate risk based on reachable states or struggle to predict accurate trajectories under high occlusion uncertainty. To address these limitations, we propose a unified risk map modeling and learning framework for partially observable environments. Our method integrates traffic flow risk and collision risk through spatiotemporal modeling, enabling fine-grained assessment of occlusion-induced hazards. To address the scarcity of scenarios involving occluded interactions, we introduce a diffusion-based scenario generation framework that produces realistic yet adversarial scenarios. We integrate the modeling and learning of a unified risk map into a framework that supports risk-aware planning under partial observability. Experiments on the Waymo Open Motion Dataset show that our method significantly outperforms the state-of-the-art occlusion-aware baseline, improving minimum time-to-collision by 0.78 times and average time-to-collision by 1.67 times. The proposed framework offers a comprehensive and practical solution for risk-aware planning in partially observable environments.
format Preprint
id arxiv_https___arxiv_org_abs_2605_22189
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Learning A Unified Risk Map for Autonomous Driving in Partially Observable Environments
Jia, Jie
Su, Yaofeng
Bao, Zeyu
Hong, Yun
Gao, Bingzhao
Gan, Zhongxue
Ding, Wenchao
Robotics
Occlusion-aware prediction remains a critical challenge in autonomous driving due to the inherent uncertainty of unobserved regions. Existing approaches either overestimate risk based on reachable states or struggle to predict accurate trajectories under high occlusion uncertainty. To address these limitations, we propose a unified risk map modeling and learning framework for partially observable environments. Our method integrates traffic flow risk and collision risk through spatiotemporal modeling, enabling fine-grained assessment of occlusion-induced hazards. To address the scarcity of scenarios involving occluded interactions, we introduce a diffusion-based scenario generation framework that produces realistic yet adversarial scenarios. We integrate the modeling and learning of a unified risk map into a framework that supports risk-aware planning under partial observability. Experiments on the Waymo Open Motion Dataset show that our method significantly outperforms the state-of-the-art occlusion-aware baseline, improving minimum time-to-collision by 0.78 times and average time-to-collision by 1.67 times. The proposed framework offers a comprehensive and practical solution for risk-aware planning in partially observable environments.
title Learning A Unified Risk Map for Autonomous Driving in Partially Observable Environments
topic Robotics
url https://arxiv.org/abs/2605.22189