Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Overcoming Catastrophic Forgetting in Incremental Object Detection via Elastic Response Distillation

About

Traditional object detectors are ill-equipped for incremental learning. However, fine-tuning directly on a well-trained detection model with only new data will lead to catastrophic forgetting. Knowledge distillation is a flexible way to mitigate catastrophic forgetting. In Incremental Object Detection (IOD), previous work mainly focuses on distilling for the combination of features and responses. However, they under-explore the information that contains in responses. In this paper, we propose a response-based incremental distillation method, dubbed Elastic Response Distillation (ERD), which focuses on elastically learning responses from the classification head and the regression head. Firstly, our method transfers category knowledge while equipping student detector with the ability to retain localization information during incremental learning. In addition, we further evaluate the quality of all locations and provide valuable responses by the Elastic Response Selection (ERS) strategy. Finally, we elucidate that the knowledge from different responses should be assigned with different importance during incremental distillation. Extensive experiments conducted on MS COCO demonstrate our method achieves state-of-the-art result, which substantially narrows the performance gap towards full training.

Tao Feng, Mang Wang, Hangjie Yuan• 2022

Related benchmarks

TaskDatasetResultRank
Object DetectionMS COCO 2017 (minival)
AP36.9
50
Class Incremental Object DetectionCOCO 2017 (val)
AP45.7
30
Object DetectionCOCO 40+40 scenario 2017 (val)
AP49.9
25
Object DetectionCOCO 70+10 scenario 2017 (val)
AP45.6
25
Incremental Object DetectionCOCO 2017 (70-10)
AP36.9
19
Incremental Object DetectionCOCO 2017 (40-40)
AP36.9
19
Object DetectionPascal VOC series (test)
Session 1 Score86.4
16
Object DetectionBDD100K series (test)
Session 1 Score52.1
16
Incremental Object DetectionCOCO 40-10 multi-step
AP47.3
13
Incremental Object DetectionCOCO 40-20 multi-step
AP50
13
Showing 10 of 33 rows

Other info

Code

Follow for update