本文对开放世界目标检测任务进行重新梳理,并提出了两种新的衡量指标
针对CEC模块存在理解问题(文中红色问号处)

Revisiting Open World Object Detection



发表于 Computer Vision and Pattern Recognition 2022/01/04


Introduction


(1) 概要

  • 领域:目标检测
  • 行文目的:OWOD的实验设置不合理(benchmark不合理、度量标准不合理、研究方法不合理)
  • 贡献:
    • (1)提出了5条基本原则,用以指导OWOD的基准建设
    • (2)设计了2种公平的评价指标

(2) OWOD

  • closed set object detection(传统意义上的目标检测)

    • 只需检测出已知的类别
    • 推理阶段也只针对已知类别的检测性能进行评价
    • 未知类别的目标可以直接忽略,或者误分类为已知类别
  • open world object detection(考虑未知类别)
    • OWOD任务包括多个增量子任务
    • 每个子任务中,训练阶段:模型仅在已知类别上做训练;测试阶段:模型需正确检测出所有已知类别,针对未知类别需分类为“unkown”
    • 人类标注员针对检测出的未知类别进行手工标注,再进入下一个增量子任务(下一次任务中就增加了新的类别标签)

(3) 存在问题及改进

i. 存在问题:(OWOD概念在《Towards open world object detection》中首次提出)

  • 基准不合理
  • 度量不合理
  • 方法不合理
    ii. 改进:(本文思路)
  • 提出5项基准原则,用以指导OWOD基准建设
  • 提出2项衡量指标 Unknown Detection Recall (UDR)、Unknown Detection Precision (UDP),更有利于衡量未知类别的检测性能
  • 提出1种简单有效的OWOD框架,包含Proposal ADvisor (PAD)模块、Class-specific Expelling Classifier (CEC)模块,解决未知类别与背景、已知类别间的难区分问题:
    • PAD模块:

      • 无参的模块
      • 协助PRN确认未知类别的proposal,引导RPN从背景中识别出未知类的proposal
    • CEC模块:
      • 标定over-confident activation boundary
      • 通过 class-specific expelling function,过滤掉不确定的预测结果,避免将未知类别分类到已知类别

(4) 本文贡献

  • 总结出5条OWOD基准原则
  • 提出2项OWOD评价指标
  • 提出一个简单有效的OWOD框架,包含PAD、CEC
  • 实验论证了本文提出的OWOD框架在原有指标及新提出的指标上的有效性

Rethinking OWOD


(1) 五原则

  • 类别的开放性(Class Openness):

    • 已知类别的集合记为 K\mathcal{K}K
    • 未知类别的集合记为 U\mathcal{U}U
    • K∩U=∅\mathcal{K} \cap \mathcal{U} = \varnothingK∩U=∅
    • 训练阶段:仅有K\mathcal{K}K参与训练
    • 测试阶段:测试集中包含K\mathcal{K}K 和 U\mathcal{U}U
  • 任务的递增性(Task Increment):
    • 前一个个子任务完成后,标注员会介入进行未知类别的人工标注,因此类别数会递增,任务也会呈现递增性
    • 在task t阶段:已知类别为 Kt\mathcal{K}_tKt​,未知类别为 Ut\mathcal{U}_tUt​
    • 在task t+1阶段:原 Ut\mathcal{U}_tUt​ 中的 Ut(k)\mathcal{U}_t^{(k)}Ut(k)​ (Ut(k)∈Ut\mathcal{U}_{t}^{(k)} \in \mathcal{U}_{t}Ut(k)​∈Ut​),对 Ut(k)\mathcal{U}_{t}^{(k)}Ut(k)​ 进行标注后,并入已知类别中,Kt+1=Kt∪Ut(k)\mathcal{K}_{t+1}=\mathcal{K}_{t} \cup \mathcal{U}_{t}^{(k)}Kt+1​=Kt​∪Ut(k)​
    • 重复上述操作,直至未知类别集合为空
  • 标注的特异性(Annotation Specificity):
    • 标签:Y=[L,B]Y = [L, B]Y=[L,B],其中Y为类别,B为边界框
    • 训练、验证阶段:仅有已知类别参与训练
    • 测试阶段:未知类别也参与测试,L赋值为“Unknown”
    • 原OWOD未遵循此原则,使用了包含未知标签的验证集来训练
  • 标签的完整性(Label Integrity):
    • 测试集中的标注信息应当完备,即所有对象(包括未知类别)在测试集标签中均应标注清楚
    • 例如PASCAL VOC的测试集中仅标注了已知的20个类别,对未知类别未标注
  • 数据的特异性(Data Specificity):
    • 训练集、验证集、测试集之间不允许有交集
    • 每个数据集中不能有重复样本
    • 否则,会影响评价结果

(2) 原始的ORE原则和本文的5原则对比


使用5原则对COCO数据集进行修改:

(3) 新的评价指标

OWOD任务存需解决的关键难点:

  • i)需从背景中识别出未知对象
  • ii)需将未知对象与已知对象作区分,确保不能将未知对象错误识别为已知的某个类别
    原评价指标不能完全评价OWOD的性能:
  • standard mAP: 衡量分类性能
  • WI、Absolute Open-Set Error(A-OSE)衡量检测未知类别的性能
  • 其中,WI不能直接应用于OWOD任务中。
    本文提出2个新的评价指标:
  • UDR(Unknow Detection Recall):
    • 衡量未知对象的定位精度
    • UDR=TPu+FNu∗TPu+FNu\mathrm{UDR}=\frac{\mathrm{TP}_{u}+\mathrm{FN}_{u}^{*}}{\mathrm{TP}_{u}+\mathrm{FN}_{u}}UDR=TPu​+FNu​TPu​+FNu∗​​
    • TPuTP_uTPu​: true positive proposals of unknow classes
    • FNuFN_uFNu​: false negative proposals of unknow classes
    • FNu∗FN^*_uFNu∗​: the number of ground-truth boxes recalled by misclassified predicted bounding boxes
  • UDP(Unknow Detection Precision):
    • 衡量未知对象的分类精度
    • UDP=TPuTPu+FNu∗\mathrm{UDP}=\frac{\mathrm{TP}_{u}}{\mathrm{TP}_{u}+\mathrm{FN}_{u}^{*}}UDP=TPu​+FNu∗​TPu​​

Method


(1) PAD(Proposal ADvisior)模块

未知样本标签的缺失,导致RPN很难生成未知样本的proposal,因此本文添加了PAD,辅助RPN针对此类proposal生成。

i) 对RPN的未知proposal结果进行再确认

  • RPN获得高分的negative proposal,有可能是被错误分类为背景的未知样本的proposal,记为P(u)+P^{(u)+}P(u)+
  • P(u)+P^{(u)+}P(u)+可能包含未知样本的proposal,同时可能包含背景
  • advisor的实现:选择性搜索(Selective Search)算法
    • 步骤一:使用《Efficient Graph-Based Image Segmentation》论文里的方法产生初始的分割区域
    • 步骤二:相似度计算后将小的区域进行相互合并
    • 重复步骤一、二
  • 确认阶段:
    • RPN 输出:P(u)+P^{(u)+}P(u)+; advisor 输出: P~+\widetilde{\mathbf{P}}+P+
    • S‾i=Si×I{max⁡1≤j≤∣P~+∣(IOU⁡(Pi(u)+,P~j+))>θ}\overline{\mathbf{S}}_{i}=\mathbf{S}_{i} \times \mathcal{I}\left\{\max _{1 \leq j \leq\left|\widetilde{\mathbf{P}}^{+}\right|}\left(\operatorname{IOU}\left(\mathbf{P}_{i}^{(u)+}, \widetilde{\mathbf{P}}_{j}^{+}\right)\right)>\theta\right\}Si​=Si​×I{max1≤j≤∣P+∣​(IOU(Pi(u)+​,Pj+​))>θ}
      • IOU 大于阈值,表示advisor和RPN均认为该区域是positive proposal

ii) 指导未知类别定位任务的无监督训练过程

  • 获取到精确的proposal后,将对应类别确认改为“前景”
  • 将对应Anchors从negative anchor set中移动到positive anchor set中
  • 将新的anchor set 输入到RPN的分类器中,计算损失:∑a∈A+∪A(u)+BCE⁡(f(a),1)+∑a∈A−\A(u)+BCE⁡(f(a),0)\sum_{\mathbf{a} \in \mathcal{A}+\cup \mathcal{A}^{(u)+}} \operatorname{BCE}(f(\mathbf{a}), \mathbf{1})+\sum_{\mathbf{a} \in \mathcal{A}^{-} \backslash \mathcal{A}^{(u)+}} \operatorname{BCE}(f(\mathbf{a}), \mathbf{0})∑a∈A+∪A(u)+​BCE(f(a),1)+∑a∈A−\A(u)+​BCE(f(a),0)

(2) CEC(Class-specific Expelling Classifier)模块

  • 问题:未知类别的对象有可能被预测为已知的某类,且赋予较高的置信度
  • 解决:Class-specific Expelling Classifier模块
    • CEC模块会淘汰掉错误预测为已知类别的对象
    • Φ(L‾ic)=L‾ic−α×1M∑j∣B~∣∑k∣Bc∣[I(IOU⁡(B~j,Bkc)>φ)×L~ic]\Phi\left(\overline{\mathbf{L}}_{i}^{c}\right)=\overline{\mathbf{L}}_{i}^{c}-\alpha \times \frac{1}{M} \sum_{j}^{|\widetilde{\mathbf{B}}|} \sum_{k}^{\left|\mathbf{B}^{c}\right|}\left[\mathcal{I}\left(\operatorname{IOU}\left(\widetilde{\mathbf{B}}_{j}, \mathbf{B}_{k}^{c}\right)>\varphi\right) \times \widetilde{\mathbf{L}}_{i}^{c}\right]Φ(Lic​)=Lic​−α×M1​∑j∣B∣​∑k∣Bc∣​[I(IOU(Bj​,Bkc​)>φ)×Lic​] ???
    • 其中,Y=[L,B]\mathbf{Y} = [\mathbf{L}, \mathbf{B}]Y=[L,B] 为ground-truth
    • Y‾=[L‾,B‾]\overline{\mathbf{Y}}=[\overline{\mathbf{L}}, \overline{\mathbf{B}}]Y=[L,B] 为在测试集的预测结果
    • L~=[L~,B~]\widetilde{\mathbf{L}}=[\widetilde{\mathbf{L}}, \widetilde{\mathbf{B}}]L=[L,B] 为训练集上的预测结果(原文为training images)?

Experiments


(1) 实验设置

  • 算力:8 * Tesla V100
  • 模型:标准Faster-RCNN,backbone网络为ResNet-50
  • 4个递增任务
  • batch_size: 8, optimizer: SGD(lr=0.01, momentum=0.9, weight_decay=0.0001)
  • task1:迭代次数 90k
  • task2-task4:迭代次数 [52k, 40k, 41k],fine-tuning迭代次数 [4k, 2.5k, 3k]
  • RPN的unknow proposals 选前top-50
  • advisor的proposals 选前top-50
  • advisor的超参θ=0.7\theta= 0.7θ=0.7
  • CEC的超参 ϕ=0.9\phi=0.9ϕ=0.9 、α=0.5\alpha=0.5α=0.5

(2) 实验结果

  • Faster-RCNN
  • Faster-RCNN + finetuning
  • ORE
  • ORE*, 不包含 annotation-leakage EBUI的ORE
  • Oracle detector 可认为是OWOD的性能上限

    结果分析:
  • WI指标,体现了本文模型对已知的类别检测性能影响最小
  • mAP指标,体现了本文模型在上阶段与当前阶段的任务中,对已知类别检测性能都很好,证明了增量学习的能力
  • Faster-RCNN虽然在当前任务上性能好,但对先前的预测结果较差,不符合增量学习要求
  • UR、UDP、UDR指标,体现了本文模型能从背景中区分出未知类别的对象,并避免错分为已知类别

(3) 消融实验

论证了各模块的有效性

i) 针对PAD的实验

ii) 针对CEC的实验

不同超参下的性能对比

  • α\alphaα 的增加,会让预测精度提升,但会损伤召回率

【论文研读】【目标检测】Revisiting Open World Object Detection相关推荐

  1. 目标检测20年(Object Detection in 20 Years)

    Object Detection in 20 Years: A Survey Zhengxia Zou, Zhenwei Shi, Member, IEEE, Yuhong Guo, and Jiep ...

  2. 【Whalepaper】CV论文研读 - OneNet:Towards End-to-End One-Stage Object Detection

    Whalepaper是由周郴莲负责的一个每周分享论文的活动. NLP 论文分享:每周日 晚上 九点 CV 论文分享: 每周日 晚上 九点 Res 论文分享:每周六 晚上 九点半 欢迎对paper感兴趣 ...

  3. 【论文阅读】【3d目标检测】Group-Free 3D Object Detection via Transformers

    论文标题:Group-Free 3D Object Detection via Transformers iccv2021 本文主要是针对votenet等网络中采用手工group的问题提出的改进 我们 ...

  4. 【3D目标检测】Monocular 3D Object Detection with Pseudo-LiDAR Point Cloud

    目录 概述 整体流程 伪点云vs点云 2D-3D边界框的一致性约束 概述 本文的输入数据仅仅是单目图像,在方法上是融合了伪点云(Pseudo-LiDAR)的深度信息表示方法与Frustum Point ...

  5. CVPR 2021 Oral | OWOD:面向开放世界的目标检测 Towards Open World Object Detection

    Towards Open World Object Detection 论文:https://arxiv.org/abs/2103.02603 代码:https://github.com/Joseph ...

  6. AI_综述:3D目标检测于RGB-D(Object detection in RGB-D images)

    转载自:https://zhuanlan.zhihu.com/p/34887948 基于深度学习的图像目标检测(上) 基于深度学习的图像目标检测(下) 大牛讲堂 | 基于DenesBox的目标检测在自 ...

  7. 【DETR用于3D目标检测】DETR3D: 3D Object Detection from Multi-view Images via 3D-to-2D Queries

    DETR3D: 3D Object Detection from Multi-view Images via 3D-to-2D Queries 论文简介: 论文贡献: 具体实现: 总体框架 特征学习 ...

  8. 实时目标检测--Pelee: A Real-Time Object Detection System on Mobile Devices

    Pelee: A Real-Time Object Detection System on Mobile Devices ICLR 2018 Code: https://github.com/Robe ...

  9. 目标检测开源代码汇总 object detection algorithm codes

    无人机目标检测 SlimYOLOv3: Narrower, Faster and Better for Real-Time UAV Applications https://arxiv.org/abs ...

  10. 【3D目标检测】Accurate Monocular Object Detection via Color-Embedded 3D Reconstruction for AD

    目录 概述 细节 整体流程 第一阶段:3D数据生成 第二阶段:3D边界框估计 前景分割 颜色嵌入 3D 边界框估计 损失函数 概述 本文是基于图像的3D目标检测方法. 和伪点云原论文在流程上是相同的, ...

最新文章

  1. 直线轨道上声音延迟信号分析
  2. 【原创】Aspose.Words组件介绍及使用—基本介绍与DOM概述
  3. java.lang.InstantiationException
  4. zabbix学习笔记(4)
  5. https证书相关概念
  6. python二进制文件读写文件_Python二进制文件读写
  7. 计算机弹琴游戏,电脑键盘钢琴软件(弹钢琴小游戏)
  8. 【自撰】Redis客户端Jedis
  9. 换掉7z-zip默认的ico图标,自定义压缩文件图标更美观。
  10. 中英文标点符号切换的组合键_切换中英文标点快捷键
  11. 信用卡诈骗检测(经过测试)
  12. android studio 制作app欢迎界面-两种方法(功能)(备忘)
  13. Tensorboard远程连接
  14. [散分] 眼见为实?_眼见为实
  15. 测试类型-- 按测试对象分 ※
  16. 大数据开发hive数据库常用命令汇总
  17. 金堂五月花计算机学校招聘,2019年成都市金堂五月花学校招生简介
  18. Oracle 定时执行计划任务
  19. android webview应用,Android WebView简单应用
  20. 四十、Flume自定义MySQLSource

热门文章

  1. 计算机组策略没有权限,运行组策略或程序时提示没有权限
  2. 关于游戏中美术文件夹的划分
  3. 计算机密码怎么重置,电脑忘记密码了怎么重置密码
  4. 机器视觉学习笔记(7)——基于OpenCV的双目摄像机标定
  5. 如何从0到1进行电商平台订单系统的搭建?
  6. 线性表、栈和队列的相同点和不同点
  7. 虚拟机您的计算机无法启动,一键解决win10虚拟机无法启动的问题
  8. 日子计算(结构+函数)
  9. 真核有参转录组测序标准分析
  10. 关于23届大数据岗实习总结