YOLO-IOD: Towards Real Time Incremental Object Detection
By: Shizhou Zhang , Xueqiang Lv , Yinghui Xing and more
Potential Business Impact:
Teaches robots to learn new things without forgetting.
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.
Similar Papers
Teach YOLO to Remember: A Self-Distillation Approach for Continual Object Detection
CV and Pattern Recognition
Teaches AI to learn new things without forgetting old ones.
YOLO-Master: MOE-Accelerated with Specialized Transformers for Enhanced Real-time Detection
CV and Pattern Recognition
Lets computers see objects better by thinking smarter.
Incremental Object Detection with Prompt-based Methods
CV and Pattern Recognition
Teaches computers to find new things in pictures.