[发明专利]一种解耦的增量目标检测方法

说明书

本发明公开了一种解耦的增量目标检测方法，包括以下步骤：S1：目标检测数据集预处理：目标检测数据集由安防检测系统中基本检测目标的图像组成；基本检测目标为安防检测系统中已实现检测的旧类别；S2：对目标检测模型进行初始学习；S3：对目标检测模型进行增量学习；S4：对增量学习后的目标检测模型进行平衡微调；S5：将调整后的目标检测模型部署在安防检测系统中，调整后的目标检测模型根据输入的目标图像进行增量目标检测，并将检测结果进行保存。本发明在保留旧类别检测性能的前提下能够动态增加检测类别、不会产生检测偏好，能够更好的满足安防检测系统需要经常动态增加检测类别并快速部署的要求。

技术领域

本发明属于增量目标检测技术领域，特别是涉及一种解耦的增量目标检测方法。

背景技术

近年来，物体检测方法获得了很大的进步，基于卷积神经网络的物体检测方法在PASCAL VOC和COCO等基准上取得了最先进的成果。然而，现有方法的有效性是基于这样一个强烈的假设：要检测的类别是事先定义好的，并且在训练阶段所有类别的数据都是可用的。在现实中，新的兴趣类别会随着时间的推移而出现，这就要求物体检测方法能够动态地学习新的类别以满足新的应用要求。例如，在需要检测场景中是否存在电子产品的智能安防领域，检测的目标往往需要动态增加，会根据需求增加需要检测的电子产品类别。

因此，增量学习引起了人们极大的研究兴趣，它利用新类别的样本持续更新训练模型。作为研究分支之一，增量目标检测是一项比增量分类和目标检测都更具挑战性的任务。实现增量目标检测的一个自然方法是在新数据上对预训练的模型进行微调。然而，这种方法面临一个棘手的挑战--灾难性的遗忘。更具体地说，在新数据上对模型进行微调通常会导致在以前的数据上出现明显的性能下降。为了解决这个问题，大多数增量目标检测方法利用知识蒸馏来确保分类逻辑和回归目标不出现明显的变化。然而，知识蒸馏法在训练过程中所添加的额外正则化项会使其难以学习新的类别。这个问题也被称为稳定性-可塑性困境，一方面，模型需要学习更多的新知识来实现可塑性，另一方面，模型需要保留更多的旧知识来实现稳定性。

目前，Faster R-CNN被广泛用作增量物体检测方法的基本检测器。然而，它的原始结构是为传统检测设计的，缺乏对增量学习场景的专门考虑，这限制了该方法的上限。例如，Faster R-CNN包含了RPN和RCNN这两个重要组件，然而由于这两种组件需要完成的任务不同，在通过共享骨干网络在之间进行端到端的联合优化时会遇到难以优化的问题。此外，由于大多数增量目标检测方法会通过知识蒸馏提炼共享骨干网络特征，导致优化困难问题将会进一步加剧。同时，还发现Faster R-CNN在增量训练过程中会出现类别不平衡的问题。具体来说，训练方法只得到了没有或很少的旧类别的样本，但有足够的新类别的样本。在这种情况下，训练过程的重点会明显偏向于新的类别，导致分类器的权重和偏差都更有利于新的类别，从而加剧模型对于旧类的遗忘。

所以，目前在目标检测问题中，仍然存在以下三个问题：

(1)传统的目标检测方法是静态的，当需要模型识别新的类别时，模型需要使用所有的数据进行重新训练，这会导致当存储资源有限，不足以保存全部数据的时候，模型的识别精度无法保证；同时重新训练模型需要消耗大量的算力，会耗费大量的时间，同时也会付出大量的经济成本(如电费、服务器租用费等)。

(2)Faster R-CNN原始结构(如图1)是为传统检测设计的，缺乏对增量学习场景的专门考虑，这限制了现有方法的上限。具体的说，就是基于Faster R-CNN的增量目标检测方法存在骨干网络难以优化问题，很难应用于增量学习中。

(3)为了区分新类别和旧类别，模型在计算损失函数时会使用所有类别的logit。在增量学习过程中，由于数据的不平衡问题，Faster R-CNN输出层将会向新类偏移，导致与旧类有关的权重会被错误的更新。