指标检测是计算机视觉中一个非常重要的基础任务,与常见的的图象分类 / 鉴别任务不同,指标检测需要模型在给出指标的类型之上,进一步给出指标的位置和大小信息,在 CV 三大任务(鉴别、检测、分割)中处于承上启下的关键地位。
当前大火的多模态 GPT-4 在视觉本领上只具备指标鉴别的本领,还无法完成更高难度的指标检测任务。而鉴别出图象或视频中物体的类型、位置和大小信息,是现实生产中众多人工智能应用的关键,例如自动驾驶中的行人车辆鉴别、安防监控应用中的人脸锁定、医学图象分析中的肿瘤定位等等。
已有的指标检测方法如 YOLO 系列、R-CNN 系列等指标检测算法在科研人员的不断努力下已经具备很高的指标检测精度与效率,但由于现有方法需要在模型训练前就定义好待检测指标的集合(闭集),导致它们无法检测训练集合之外的指标,比如一个被训练用于检测人脸的模型就不能用于检测车辆;另外,现有方法高度依赖人工标注的数据,当需要增加或者修改待检测的指标类型时,一方面需要对训练数据举行重新标注,另一方面需要对模型举行重新训练,既费时又费力。
一个可能的解决方案是,收集海量的图象,并人工标注 Box 信息与语义信息,但这将需要极高的标注成本,而且使用海量数据对检测模型举行训练也对科研工作者提出了严峻的挑战,如数据的长尾分布问题与人工标注的质量不稳定等因素都将影响检测模型的性能表现。
发表于 CVPR 2021 的文章 OVR-CNN [1] 提出了一种全新的指标检测范式:开放词集指标检测(Open-Vocabulary Detection,OVD,亦称为开放世界指标检测),来应对上文提到的问题,即面向开放世界未知物体的检测场景。
OVD 由于能够在无需人工扩充标注数据量的情形下鉴别并定位任意数量和类型指标的本领,自提出后吸引了学术界与工业界的持续关注,也为经典的指标检测任务带来了新的活力与新的挑战,有望成为指标检测的未来新范式。
具体地,OVD 手艺不需要人工标注海量的图片来增强检测模型对未知类型的检测本领,而是通过将具有良好泛化性的无类型(class-agnostic)地区检测器与经过海量无标注数据训练的跨模态模型相结合,通过图象地区特点与待检测指标的描述性文字举行跨模态对齐来扩展指标检测模型对开放世界指标的理解本领。
跨模态和多模态大模型工作近期的发展非常迅速,如 CLIP [2]、ALIGN [3] 与 R2D2 [4] 等,而它们的发展也促进了 OVD 的诞生与 OVD 领域相关工作的快速迭代与进化。
OVD 手艺涉及两大关键问题的解决:1)如何提升地区 (Region) 信息与跨模态大模型之间的适配;2)如何提升泛类型指标检测器对新类型的泛化本领。从这两个角度出发,下面将详细介绍一些 OVD 领域的相关工作。
OVD 基本流程示意 [1]
OVD 的基础概念:OVD 的使用主要涉及到 few-shot 和 zero-shot 两大类场景,few-shot 是指有少量人工标注训练样本的指标类型,zero-shot 则是指不存在任何人工标注训练样本的指标类型。在常用的学术评测数据集 COCO、LVIS 上,数据集会被划分为 Base 类和 Novel 类,其中 Base 类对应 few-shot 场景,Novel 类对应 zero-shot 场景。如 COCO 数据集包含 65 种类型,常用的评测设定是 Base 集包含 48 种类型,few-shot 训练中只使用这 48 个类型。Novel 集包含 17 种类型,在训练时完全不可见。测试指标主要参考 Novel 类的 AP50 数值举行比较。
论文1:Open-Vocabulary Object Detection Using Captions
论文地址:https://arxiv.org/pdf/2011.10678.pdf
代码地址:https://github.com/alirezazareian/ovr-cnn
OVR-CNN 是 CVPR 2021 的 Oral-Paper,也是 OVD 领域的开山之作。它的二阶段训练范式,影响了后续很多的 OVD 工作。如下图所示,第一阶段主要使用 image-caption pairs 对视觉编码器举行预训练,其中借助 BERT (参数固定) 来生成词掩码,并与加载 ImageNet 预训练权重的 ResNet50 举行弱监督的 Grounding 立室,作者认为弱监督会让立室陷入局部最优,于是加入多模态 Transformer 举行词掩码预测来增加鲁棒性。
第二阶段的训练流程与 Faster-RCNN 类似,区别点在于,特点提取的 Backbone 来自于第一阶段预训练得到的 ResNet50 的 1-3 层,RPN 后依然使用 ResNet50 的第四层举行特点加工,随后将特点分别用于 Box 回归与分类预测。分类预测是 OVD 任务区别于常规检测的关键标志,OVR-CNN 中将特点输入一阶段训练得到的 V2L 模块 (参数固定的图向量转词向量模块) 得到一个图文向量,随后与标签词向量组举行立室,对类型举行预测。在二阶段训练中,主要使用 Base 类对检测器模型举行框回归训练与类型立室训练。由于 V2L 模块始终固定,配合指标检测模型定位本领向新类型迁移,使得检测模型能够鉴别并定位到全新类型的指标。
如下图所示,OVR-CNN 在 COCO 数据集上的表现远超之前的 Zero-shot 指标检测算法。
论文2:RegionCLIP: Region-based Language-Image Pretraining
论文地址:https://arxiv.org/abs/2112.09106
代码地址:https://github.com/microsoft/RegionCLIP
OVR-CNN 中使用 BERT 与多模态 Transfomer 举行 iamge-text pairs 预训练,但随着跨模态大模型研究的兴起,科研工作者开始利用 CLIP,ALIGN 等更强大的跨模态大模型对 OVD 任务举行训练。检测器模型本身主要针对 Proposals,即地区信息举行分类鉴别,发表于 CVPR 2022 的 RegionCLIP [5] 发现当前已有的大模型,如 CLIP,对裁剪地区的分类本领远低于对原图本身的分类本领,为了改进这一点,RegionCLIP 提出了一个全新的两阶段 OVD 方案。
第一阶段,数据集主要使用 CC3M,COCO-caption 等图文立室数据集举行地区级别的蒸馏预训练。具体地:
1. 将原先存在于长文本中的词汇举行提取,组成 Concept Pool,进一步形成一组关于 Region 的简单描述,用于训练。
2. 利用基于 LVIS 预训练的 RPN 提取 Proposal Regions,并利用原始 CLIP 对提取到的不同 Region 与准备好的描述举行立室分类,并进一步组装成伪造的语义标签。
3. 将准备好的 Proposal Regions 与语义标签在新的 CLIP 模型上举行 Region-text 对比学习,进而得到一个专精于 Region 信息的 CLIP 模型。
4. 在预训练中,新的 CLIP 模型还会通过蒸馏策略学习原始 CLIP 的分类本领,以及举行全图级别的 image-text 对比学习,来维持新的 CLIP 模型对完整图象的表达本领。
第二阶段,将得到的预训练模型在检测模型上举行迁移学习。
RegionCLIP 进一步拓展了已有跨模态大模型在常规检测模型上的表征本领,进而取得了更加出色的性能,如下图所示,RegionCLIP 相比 OVR-CNN 在 Novel 类型上取得了较大提升。RegionCLIP 通过一阶段的预训练有效地的提升了地区 (Region) 信息与多模态大模型之间的适应本领,但 CORA 认为其使用更大参数规模的跨模态大模型举行一阶段训练时,训练成本将会非常高昂。
论文3:CORA: Adapting CLIP for Open-Vocabulary Detection with Region Prompting and Anchor Pre-Matching
论文地址:https://arxiv.org/abs/2303.13076
代码地址:https://github.com/tgxs002/CORA
CORA [6] 已被收录于 CVPR 2023,为了克服其所提出当前 OVD 任务所面临的两个阻碍,设计了一个类 DETR 的 OVD 模型。如其文章标题所示,模型主要包含了 Region Prompting 与 Anchor Pre-Matching 两个策略。前者通过 Prompt 手艺来优化基于 CLIP 的地区分类器所提取的地区特点,进而缓解整体与地区的分布差距,后者通过 DETR 检测方法中的锚点预立室策略来提升 OVD 模型对新类型物体定位本领的泛化性。
CLIP 原始视觉编码器的整体图象特点与地区特点之间存在分布差距,进而导致检测器的分类精度较低(这一点与 RegionCLIP 的出发点类似)。因此,CORA 提出 Region Prompting 来适应 CLIP 图象编码器,提高对地区信息的分类性能。具体地,首先通过 CLIP 编码器的前 3 层将整幅图象编码成一个特点映射,然后由 RoI Align 生成锚点框或预测框,并将其合并成地区特点。随后由 CLIP 图象编码器的第四层举行编码。为了缓解 CLIP 图象编码器的全图特点图与地区特点之间存在分布差距,设置了可学习的 Region Prompts 并与第四层输出的特点举行组合,进而生成最终的地区特点用来与文本特点举行立室,立室损失使用了朴素的交叉熵损失,且训练过程中与 CLIP 相关的参数模型全都冻结。
CORA 是一个类 DETR 的检测器模型,类似于 DETR,其也使用了锚点预立室策略来提前生成候选框用于框回归训练。具体来说,锚点预立室是将每个标签框与最接近的一组锚点框举行立室,以确定哪些锚点框应该被视为正样本,哪些应该被视为负样本。这个立室过程通常是基于 IoU(交并比)举行的,如果锚点框与标签框的 IoU 超过一个预定义的阈值,则将其视为正样本,否则将其视为负样本。CORA 表明该策略能够有效提高对新类型定位本领的泛化性。
但是使用锚点预立室机制也会带来一些问题,比如只有在至少有一个锚点框与标签框形成立室时,才可正常举行训练。否则,该标签框将被忽略,同时阻碍模型的收敛。进一步,即使标签框获得了较为准确的锚点框,由于 Region Classifier 的鉴别精度有限,进而导致该标签框仍可能被忽略,即标签框对应的类型信息没有与基于 CLIP 训练的 Region Classifier 形成对齐。因此,CORA 用 CLIP-Aligned 手艺利用 CLIP 的语义鉴别本领,与预训练 ROI 的定位本领,在较少人力情形下对训练数据集的图象举行重新标注,使用这种手艺,可以让模型在训练中立室更多的标签框。
相比于 RegionCLIP,CORA 在 COCO 数据集上进一步提升了 2.4 的 AP50 数值。
总结与展望
OVD 手艺不仅与当前流行的跨 / 多模态大模型的发展紧密联系,同时也承接了过去科研工作者对指标检测领域的手艺累积,是传统 AI 手艺与面向通用 AI 本领研究的一次成功衔接。OVD 更是一项面向未来的全新指标检测手艺,可以预料到的是,OVD 可以检测并定位任意指标的本领,也将反过来推进多模态大模型的进一步发展,有希望成为多模态 AGI 发展中的重要基石。当下,多模态大模型的训练数据来源是网络上的海量粗糙信息对,即文本图象对或文本语音对。若利用 OVD 手艺对原本粗糙的图象信息举行精准定位,并辅助预测图象的语义信息来筛选语料,将会进一步提升大模型预训练数据的质量,进而优化大模型的表征本领与理解本领。
一个很好的例子便是 SAM (Segment Anything)[7],SAM 不仅让科研工作者们看到了通用视觉大模型未来方向,也引发了很多思考。值得注意的是,OVD 手艺可以很好的接入 SAM,来增强 SAM 的语义理解本领,自动的生成 SAM 需要的 box 信息,从而进一步解放人力。同样的对于 AIGC (人工智能生成内容),OVD 手艺同样可以增强与用户之间的交互本领,如当用户需要指定一张图片的某一个指标举行变化,或对该指标生成一句描述的时候,可以利用 OVD 的语言理解本领与 OVD 对未知指标检测的本领实现对用户描述对象的精准定位,进而实现更高质量的内容生成。当下 OVD 领域的相关研究蓬勃发展,OVD 手艺对未来通用 AI 大模型能够带来的改变值得期待。
最新计算机视觉大赛报名信息
为了推动 OVD 研究在国内的普及和发展,360 人工智能研究院联合中国图象图形学学会举办了 2023 开放世界指标检测竞赛,目前竞赛正在火热报名中。竞赛可以帮助大家找到 OVD 方向的研究同好,与他们切磋交流,并能接触实际业务场景数据,体验 OVD 手艺在实际生产中的优势与魅力,欢迎报名和转发。
大赛地址:https://360cvgroup.github.io/OVD_Contest/
360 人工智能研究院简介
360 人工智能研究院隶属于 360 手艺中台。自 2015 年成立以来积累了大量人工智能与机器学习前沿本领,范围包括但不限于自然语言理解、机器视觉与运动、语音语义交互等方面,手艺水平行业领先,核心成员和团队多次荣获 AI 相关比赛冠军 / 提名奖项,发表顶会、顶刊论文数十篇。业务落地方面,研究院提供智能安全大数据、互联网信息分发、企业数字化、AIoT、智能汽车等 360 集团全量业务场景支持,支持千万级硬件设备,亿级用户,产生千亿规模数据量。2023 年着重攻坚大语言模型、CV 大模型和多模态大模型,为 360 集团和行业 AIGC 手艺发展应用提供底层手艺支撑。
参考文献
[1] Zareian A, Rosa K D, Hu D H, et al. Open-vocabulary object detection using captions [C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2021: 14393-14402.
[2] Radford A, Kim J W, Hallacy C, et al. Learning transferable visual models from natural language supervision [C]//International conference on machine learning. PMLR, 2021: 8748-8763.
[3] Li J, Selvaraju R, Gotmare A, et al. Align before fuse: Vision and language representation learning with momentum distillation [J]. Advances in neural information processing systems, 2021, 34: 9694-9705.
[4] Xie C, Cai H, Song J, et al. Zero and R2D2: A Large-scale Chinese Cross-modal Benchmark and A Vision-Language Framework [J]. arXiv preprint arXiv:2205.03860, 2022.
[5] Zhong Y, Yang J, Zhang P, et al. Regionclip: Region-based language-image pretraining [C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2022: 16793-16803.
[6] Wu X, Zhu F, Zhao R, et al. CORA: Adapting CLIP for Open-Vocabulary Detection with Region Prompting and Anchor Pre-Matching [J]. arXiv preprint arXiv:2303.13076, 2023.
[7] Kirillov A, Mintun E, Ravi N, et al. Segment anything [J]. arXiv preprint arXiv:2304.02643, 2023.