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 language | English |
---|---|
Title of host publication | Proceedings of the 2024 IEEE 22nd International Conference on Industrial Informatics (INDIN) |
Number of pages | 6 |
Publisher | IEEE |
Publication date | 2024 |
ISBN (Print) | 979-8-3315-2748-8 |
ISBN (Electronic) | 979-8-3315-2747-1 |
DOIs | |
Publication status | Published - 2024 |
Event | IEEE 22nd International Conference on Industrial Informatics - Beijing, China Duration: 17 Aug 2024 → 20 Aug 2024 |
Conference
Conference | IEEE 22nd International Conference on Industrial Informatics |
---|---|
Country/Territory | China |
City | Beijing |
Period | 17/08/2024 → 20/08/2024 |
Keywords
- Object detection
- Incremental learning
- Knowledge distillation