Out-of-Distribution Detection in LiDAR Semantic Segmentation Using Epistemic Uncertainty from Hierarchical GMMs
Miandashti, Brenner
In addition to accurate scene understanding through precise semantic segmentation of LiDAR point clouds, detecting out-of-distribution (OOD) objects, instances not encountered during training, is essential to prevent the incorrect assignment of unknown objects to known classes. While supervised OOD detection methods depend on auxiliary OOD datasets, unsupervised methods avoid this requirement but typically rely on predictive entropy, the entropy of the predictive distribution obtained by averaging over an ensemble or multiple posterior weight samples. However, these methods often conflate epistemic (model) and aleatoric (data) uncertainties, misclassifying ambiguous in distribution regions as OOD. To address this issue, we present an unsupervised OOD detection approach that employs epistemic uncertainty derived from hierarchical Bayesian modeling of Gaussian Mixture Model (GMM) parameters in the feature space of a deep neural network. Without requiring auxiliary data or additional training stages, our approach outperforms existing uncertainty-based methods on the SemanticKITTI dataset, achieving an 18\% improvement in AUROC, 22\% increase in AUPRC, and 36\% reduction in FPR95 (from 76\% to 40\%), compared to the predictive entropy approach used in prior works.
academic
Out-of-Distribution Detection in LiDAR Semantic Segmentation Using Epistemic Uncertainty from Hierarchical GMMs
This paper proposes an epistemic uncertainty estimation method based on hierarchical Bayesian Gaussian Mixture Model (GMM) parameter modeling for out-of-distribution (OOD) detection in LiDAR semantic segmentation. The method requires neither auxiliary data nor additional training stages, achieving significant improvements over existing prediction entropy-based methods on the SemanticKITTI dataset: 18% improvement in AUROC, 22% improvement in AUPRC, and reduction of FPR95 from 76% to 40%.
In safety-critical applications such as autonomous driving, LiDAR semantic segmentation must not only accurately understand known scenes but also identify out-of-distribution (OOD) objects unseen during training, avoiding misclassification of unknown objects as known categories.
Supervised Methods: Rely on auxiliary OOD datasets, which are difficult to obtain in the LiDAR domain
Unsupervised Methods: Primarily depend on prediction entropy, conflating epistemic and aleatoric uncertainty
Uncertainty Confusion: Prediction entropy mixes model uncertainty with data noise, leading to misclassification of ambiguous in-distribution regions as OOD
Proposed OOD detection method based on epistemic uncertainty: Separates epistemic and aleatoric uncertainty through hierarchical Bayesian modeling of GMM parameters
Unsupervised Framework: Requires neither auxiliary OOD data nor additional training stages, enhancing practicality
Significant Performance Improvements: Substantially surpasses existing methods on the SemanticKITTI dataset
Theoretical Contribution: Demonstrates the advantages of epistemic uncertainty over prediction entropy in OOD detection
Based on the GMMSeg architecture, employing deep neural networks to extract pixel-level feature representations z ∈ R^D, then modeling class-conditional Gaussian mixture distributions for each semantic class c in feature space:
p(z | c) = Σ(k=1 to K) π_k^(c) N(z | μ_k^(c), Σ_k^(c))
where π_k^(c), μ_k^(c), Σ_k^(c) represent the mixture weight, mean, and covariance of the k-th Gaussian component, respectively.
Uncertainty Separation: Directly estimates epistemic uncertainty through parameter sampling, avoiding the conflation of two uncertainty types in prediction entropy
No Retraining Required: Leverages the feature space of existing segmentation networks, achieving uncertainty estimation through GMM modeling
Theoretical Foundation: Based on mutual information theory, epistemic uncertainty is defined as Iy; θ | x = H(p(y|x)) - E_θH(p(y|x,θ))
The paper cites important works in related fields, including:
GMMSeg 18: Foundational work in generative semantic segmentation
Deep Ensembles 16: Classical method for uncertainty estimation
MC Dropout 7: Representative work in Bayesian deep learning
SemanticKITTI 1: Standard dataset for LiDAR semantic segmentation
This paper makes important contributions to the field of OOD detection in LiDAR semantic segmentation. Through clever technical design, it achieves effective estimation of epistemic uncertainty, providing valuable solutions for improving the safety and reliability of autonomous driving systems.