Incremental Learning Strategy with Multi-dimensional Knowledge Distillation for One-stage Object Detection

Yimeng Li, Xuejiao Liu*, Xueshuang Xiang*, Yu-an Tan, Weizhi Meng

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingArticle in proceedingsResearchpeer-review

Abstract

In practical applications, object detectors often encounter unknown classes samples. As object detection datasets undergo continuous development, the need arises for object detectors to adeptly recognize an escalating array of new classes while retaining the original detection capability. This paper introduces a new class incremental object detection learning framework with multi-dimensional knowledge distillation, using Yolov5 as the foundational detection model. Specifically, we utilize local feature distillation, attention distillation and global feature distillation to preserve information in the original model's feature maps. Additionally, we employ response distillation to ensure the model remains responsive to previous classes. In comparison to the state-of-the-art methods, our approach achieves higher overall accuracy on PASCAL VOC 2007. Especially in the “19+1” incremental task, our approach gets the highest [email protected] of 0.697 with a minimum mean AP decline rate of 4.45%. For the “4+3” and “6+1” tasks on KITTI, our approach preserves the model's responsiveness to the original task more effectively while achieving the highest overall accuracy.
Original languageEnglish
Title of host publicationProceedings of the 2024 IEEE 22nd International Conference on Industrial Informatics (INDIN)
Number of pages6
PublisherIEEE
Publication date2024
ISBN (Print)979-8-3315-2748-8
ISBN (Electronic)979-8-3315-2747-1
DOIs
Publication statusPublished - 2024
EventIEEE 22nd International Conference on Industrial Informatics - Beijing, China
Duration: 17 Aug 202420 Aug 2024

Conference

ConferenceIEEE 22nd International Conference on Industrial Informatics
Country/TerritoryChina
CityBeijing
Period17/08/202420/08/2024

Keywords

  • Object detection
  • Incremental learning
  • Knowledge distillation

Fingerprint

Dive into the research topics of 'Incremental Learning Strategy with Multi-dimensional Knowledge Distillation for One-stage Object Detection'. Together they form a unique fingerprint.

Cite this