ProOOD: Prototype-Guided Out-of-Distribution 3D Occupancy Prediction
About
3D semantic occupancy prediction is central to autonomous driving, yet current methods are vulnerable to long-tailed class bias and out-of-distribution (OOD) inputs, often overconfidently assigning anomalies to rare classes. We present ProOOD, a lightweight, plug-and-play method that couples prototype-guided refinement with training-free OOD scoring. ProOOD comprises (i) prototype-guided semantic imputation that fills occluded regions with class-consistent features, (ii) prototype-guided tail mining that strengthens rare-class representations to curb OOD absorption, and (iii) EchoOOD, which fuses local logit coherence with local and global prototype matching to produce reliable voxel-level OOD scores. Extensive experiments on five datasets demonstrate that ProOOD achieves state-of-the-art performance on both in-distribution 3D occupancy prediction and OOD detection. On SemanticKITTI, it surpasses baselines by +3.57% mIoU overall and +24.80% tail-class mIoU; on VAA-KITTI, it improves AuPRCr by +19.34 points, with consistent gains across benchmarks. These improvements yield more calibrated occupancy estimates and more reliable OOD detection in safety-critical urban driving. The source code is publicly available at https://github.com/7uHeng/ProOOD.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic Occupancy Prediction | SemanticKITTI (test) | mIoU46.75 | 47 | |
| 3D Semantic Occupancy Prediction | SemanticKITTI (val) | mIoU18.86 | 24 | |
| 3D Semantic Occupancy Prediction | SSCBench-KITTI-360 (test) | Overall IoU48.23 | 24 | |
| Out-of-Distribution Detection | VAA-KITTI | AuPRCr (0.8m)27.86 | 5 | |
| Out-of-Distribution Detection | KITTI-360 VAA | AuPRCr (0.8m)14.56 | 5 | |
| Out-of-Distribution Detection | VAA-STU | AuPRCr (0.8m)7.98 | 5 | |
| 3D Occupancy Prediction | VAA-KITTI-360 | Overall IoU42.54 | 4 | |
| 3D Semantic Occupancy Prediction | VAA-KITTI | IoU40.88 | 4 |