中文

一种用于分割任务分布外检测的元认知方法

计算机视觉与模式识别 2023-11-15 v1 人工智能 机器学习

摘要

尽管深度神经网络在封闭世界中的语义场景分割表现优异,但其对开放世界中自主智能体所需的新颖实例分割能力较差。为改进分割的分布外(OOD)检测,我们提出了一种轻量级模块形式的元认知方法,该方法利用熵度量、分割预测和空间上下文来刻画分割模型的不确定性,并实时检测像素级 OOD 数据。此外,我们的方法引入了一种在分布内数据语境下生成合成 OOD 数据的新方法,并用于通过最大熵训练微调现有分割模型。这进一步提升了元认知模块的性能,且无需访问 OOD 数据,同时兼容已有的预训练模型。如语义分割 OOD 检测基准上的 SOTA 性能所示,我们的最终方法能够可靠地检测场景中的 OOD 实例。

关键词

引用

@article{arxiv.2311.07578,
  title  = {A Metacognitive Approach to Out-of-Distribution Detection for Segmentation},
  author = {Meghna Gummadi and Cassandra Kent and Karl Schmeckpeper and Eric Eaton},
  journal= {arXiv preprint arXiv:2311.07578},
  year   = {2023}
}