基于注意力的多尺度选择特征蒸馏助力增量目标检测

    Attention-based multi-scale selective feature distillation assists incremental object detection

    • 摘要: 由于社会与需求的演变,目标检测需覆盖的类别越来越多,如果只用新数据微调模型,可能会出现灾难性遗忘。知识蒸馏是一种很好的解决灾难性遗忘的方法,但是以往的特征蒸馏方法主要关注浅层特征,而较少关注深层特征的高级语义信息。因此,本文提出了基于注意力的多尺度选择特征蒸馏方法来解决此问题,同时对浅层和深层特征进行蒸馏,综合利用不同尺度的特征信息,并且在蒸馏的过程中引入基于注意力的选择模块,动态关注重要特征,选择性地进行蒸馏,使模型在新旧任务中平衡,最终将其与分类蒸馏、定位蒸馏结合在一起。在MS COCO数据集上进行了大量的实验,证明了基于注意力的多尺度选择特征蒸馏方法的有效性。

       

      Abstract: As technology advances and human needs evolve, the demand for object detection across an expanding number of categories has significantly increased. However, fine-tuning models using only new data often results in catastrophic forgetting, where the model loses previously learned knowledge. While knowledge distillation has emerged as a promising approach to alleviate this issue, traditional feature distillation techniques primarily focus on shallow features, paying less attention to the rich semantic information embedded in deeper features. To address this, an attention-based multi-scale selective feature distillation method is proposed to distill both shallow and deep features, making comprehensive use of feature information at various scales. An attention-based selective module is incorporated during the distillation process to dynamically emphasize important features and selectively distill them. This approach enables the model to balance performance across both new and previous tasks and is further combined with classification and localization distillation. Extensive experiments on the MS COCO dataset have demonstrated the effectiveness of the attention-based multi-scale selective feature distillation method.

       

    /

    返回文章
    返回