3 个月前

基于扩散模型的交通场景及更广泛场景下的分布外检测

基于扩散模型的交通场景及更广泛场景下的分布外检测

摘要

近年来,面向语义分割的分布外(Out-of-Distribution, OoD)检测研究主要聚焦于道路场景——这一领域具有有限的语义多样性。在本工作中,我们突破这一限制,将该任务的适用范围拓展至一般自然图像。为此,我们提出:1)基于ADE20k数据集构建的ADE-OoD基准,该基准涵盖来自多样化领域的图像,具有较高的语义多样性;2)一种新颖的基于扩散模型得分匹配的OoD检测方法(Diffusion-based OoD Detection, DOoD),该方法对语义多样性的提升具有良好的鲁棒性。ADE-OoD基准包含室内与室外图像,定义了150个语义类别作为分布内(in-distribution)类别,并包含多种分布外(OoD)物体。对于DOoD方法,我们在语义分布内嵌入表示上训练一个基于MLP架构的扩散模型,并基于得分匹配的理论框架,在推理阶段计算像素级的OoD得分。在常见的道路场景OoD检测基准上,DOoD的性能达到或优于现有最先进方法,且无需使用异常样本进行训练,也不依赖对数据域的先验假设。在ADE-OoD基准上,DOoD显著优于此前的方法,但仍存在较大的提升空间,为未来研究提供了广阔方向。

代码仓库

lmb-freiburg/diffusion-for-ood
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
anomaly-detection-on-road-anomalyDOoD
AP: 89.1
FPR95: 8.8
out-of-distribution-detection-on-ade-oodDOoD
AP: 63.03
FPR@95: 36.50

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于扩散模型的交通场景及更广泛场景下的分布外检测 | 论文 | HyperAI超神经