arXiv:2509.04816cs.CVcs.LG2025-09ICCV被引 7

用专家混合模型直接提取语义分割的可靠不确定性估计。

Extracting Uncertainty Estimates from Mixtures of Experts for Semantic Segmentation

  • 不改架构,用门控网络动态加权专家预测来估算不确定性。
  • 在A2D2数据集上,MoE比集成方法在分布外数据下更准确。
  • 专家数量增加能提升不确定性校准,简单门控更优。

准确且校准良好的预测不确定性估计对提升计算机视觉模型的可靠性至关重要,尤其在交通场景感知等安全关键应用中。尽管集成方法通过组合多个模型来量化不确定性,但专家混合(MoE)通过门控网络根据输入动态加权专家预测,提供了一种更高效的方法。基于我们之前在语义分割中使用MoE的成果,本文表明无需修改架构即可从MoE中提取校准良好的不确定性估计。我们研究了三种方法:预测熵、互信息和专家方差。在基于A2D2数据集语义划分训练的双专家MoE上评估,结果表明,对于分布外(OOD)数据,MoE在条件正确性指标上优于集成方法。此外,通过门控熵计算路由不确定性,发现简单门控机制比复杂的类别门控带来更好的校准效果。最后,在Cityscapes数据集上的实验表明,增加专家数量可进一步提升不确定性校准。代码已开源:https://github.com/KASTEL-MobilityLab/mixtures-of-experts/。

原文摘要 · Abstract (English)

Estimating accurate and well-calibrated predictive uncertainty is important for enhancing the reliability of computer vision models, especially in safety-critical applications like traffic scene perception. While ensemble methods are commonly used to quantify uncertainty by combining multiple models, a mixture of experts (MoE) offers an efficient alternative by leveraging a gating network to dynamically weight expert predictions based on the input. Building on the promising use of MoEs for semantic segmentation in our previous works, we show that well-calibrated predictive uncertainty estimates can be extracted from MoEs without architectural modifications. We investigate three methods to extract predictive uncertainty estimates: predictive entropy, mutual information, and expert variance. We evaluate these methods for an MoE with two experts trained on a semantical split of the A2D2 dataset. Our results show that MoEs yield more reliable uncertainty estimates than ensembles in terms of conditional correctness metrics under out-of-distribution (OOD) data. Additionally, we evaluate routing uncertainty computed via gate entropy and find that simple gating mechanisms lead to better calibration of routing uncertainty estimates than more complex classwise gates. Finally, our experiments on the Cityscapes dataset suggest that increasing the number of experts can further enhance uncertainty calibration. Our code is available at https://github.com/KASTEL-MobilityLab/mixtures-of-experts/.

不确定性估计专家混合语义分割校准

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。