用单模型特征空间同时做障碍物检测与不确定性估计
Uncertainty Estimation and Out-of-Distribution Detection for LiDAR Scene Semantic Segmentation
- 基于单模型特征空间构建高斯混合模型,无需额外训练数据
- 通过马氏距离和卡方分布检测分布外样本,准确率显著提升
- 适合自动驾驶等对安全性要求高的实时场景应用
安全导航依赖于对环境的准确理解,包括激光雷达场景分割、分布外(OOD)障碍物检测和不确定性计算。本文提出一种方法,利用单一确定性模型的特征空间区分分布内(ID)与分布外(OOD)样本,并量化认知不确定性(epistemic)和随机不确定性(aleatoric)。训练完语义分割网络后,在其特征空间上拟合高斯混合模型(GMM)。通过检查样本到各高斯分量的平方马氏距离是否符合卡方分布来检测OOD样本,无需额外的OOD训练集。由于多维高斯分布的均值和协方差矩阵分别服从高斯和逆威沙特分布,通过从这些分布中采样生成多个GMM,以分类结果的变异性评估认知不确定性。随机不确定性则由各高斯成分的责任值熵得出。实验表明,该方法在真实场景下对认知和随机不确定性的量化以及OOD检测性能优于深度集成和对数概率采样,尤其能有效识别高不确定性样本。
原文摘要 · Abstract (English)
Safe navigation in new environments requires autonomous vehicles and robots to accurately interpret their surroundings, relying on LiDAR scene segmentation, out-of-distribution (OOD) obstacle detection, and uncertainty computation. We propose a method to distinguish in-distribution (ID) from OOD samples and quantify both epistemic and aleatoric uncertainties using the feature space of a single deterministic model. After training a semantic segmentation network, a Gaussian Mixture Model (GMM) is fitted to its feature space. OOD samples are detected by checking if their squared Mahalanobis distances to each Gaussian component conform to a chi-squared distribution, eliminating the need for an additional OOD training set. Given that the estimated mean and covariance matrix of a multivariate Gaussian distribution follow Gaussian and Inverse-Wishart distributions, multiple GMMs are generated by sampling from these distributions to assess epistemic uncertainty through classification variability. Aleatoric uncertainty is derived from the entropy of responsibility values within Gaussian components. Comparing our method with deep ensembles and logit-sampling for uncertainty computation demonstrates its superior performance in real-world applications for quantifying epistemic and aleatoric uncertainty, as well as detecting OOD samples. While deep ensembles miss some highly uncertain samples, our method successfully detects them and assigns high epistemic uncertainty.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。