Current methods for incremental object detection (IOD) primarily rely on Faster R-CNN or DETR series detectors; however, these approaches do not accommodate the real-time YOLO detection frameworks. In this paper, we first identify three primary types of knowledge conflicts that contribute to catastrophic forgetting in YOLO-based incremental detectors: foreground-background confusion, parameter interference, and misaligned knowledge distillation. Subsequently, we introduce YOLO-IOD, a real-time Incremental Object Detection (IOD) framework that is constructed upon the pretrained YOLO-World model, facilitating incremental learning via a stage-wise parameter-efficient fine-tuning process. Specifically, YOLO-IOD encompasses three principal components: 1) Conflict-Aware Pseudo-Label Refinement (CPR), which mitigates the foreground-background confusion by leveraging the confidence levels of pseudo labels and identifying potential objects relevant to future tasks. 2) Importancebased Kernel Selection (IKS), which identifies and updates the pivotal convolution kernels pertinent to the current task during the current learning stage. 3) Cross-Stage Asymmetric Knowledge Distillation (CAKD), which addresses the misaligned knowledge distillation conflict by transmitting the features of the student target detector through the detection heads of both the previous and current teacher detectors, thereby facilitating asymmetric distillation between existing and newly introduced categories. We further introduce LoCo COCO, a more realistic benchmark that eliminates data leakage across stages. Experiments on both conventional and LoCo COCO benchmarks show that YOLO-IOD achieves superior performance with minimal forgetting.
翻译:当前增量目标检测方法主要基于Faster R-CNN或DETR系列检测器,但这些方法无法适配实时YOLO检测框架。本文首先识别出导致YOLO增量检测器灾难性遗忘的三种主要知识冲突类型:前景-背景混淆、参数干扰与知识蒸馏错位。随后,我们提出YOLO-IOD——一种基于预训练YOLO-World模型构建的实时增量目标检测框架,通过分阶段参数高效微调实现增量学习。具体而言,YOLO-IOD包含三个核心组件:1)冲突感知伪标签优化模块,通过利用伪标签置信度识别与未来任务相关的潜在目标,缓解前景-背景混淆问题;2)基于重要性的卷积核选择机制,在当前学习阶段识别并更新与当前任务相关的关键卷积核;3)跨阶段非对称知识蒸馏方法,通过将学生目标检测器的特征同时输入到前阶段与当前阶段教师检测器的检测头中,实现新旧类别间的非对称蒸馏,从而解决知识蒸馏错位冲突。我们进一步提出LoCo COCO基准数据集,该数据集消除了阶段间的数据泄露问题,更具现实意义。在传统基准与LoCo COCO上的实验表明,YOLO-IOD能以最小遗忘实现卓越性能。