DFMSD: Dual Feature Masking Stage-wise Knowledge Distillation for Object Detection

TL;DR

This paper introduces DFMSD, a novel dual feature masking stage-wise knowledge distillation framework that effectively bridges the gap between heterogeneous teacher and student networks for improved object detection performance.

Contribution

The paper proposes a stage-wise adaptation module and masking enhancement strategy to improve heterogeneous knowledge distillation in object detection.

Findings

DFMSD outperforms state-of-the-art distillation methods.

Stage-wise adaptation effectively bridges network discrepancies.

Object-aware masking enhances feature reconstruction quality.

Abstract

In recent years, current mainstream feature masking distillation methods mainly function by reconstructing selectively masked regions of a student network from the feature maps of a teacher network. In these methods, attention mechanisms can help to identify spatially important regions and crucial object-aware channel clues, such that the reconstructed features are encoded with sufficient discriminative and representational power similar to teacher features. However, previous feature-masking distillation methods mainly address homogeneous knowledge distillation without fully taking into account the heterogeneous knowledge distillation scenario. In particular, the huge discrepancy between the teacher and the student frameworks within the heterogeneous distillation paradigm is detrimental to feature masking, leading to deteriorating reconstructed student features. In this study, a novel dual feature-masking heterogeneous distillation framework termed DFMSD is proposed for object detection. More specifically, a stage-wise adaptation learning module is incorporated into the dual feature-masking framework, and thus the student model can be progressively adapted to the teacher models for bridging the gap between heterogeneous networks. Furthermore, a masking enhancement strategy is combined with stage-wise learning such that object-aware masking regions are adaptively strengthened to improve feature-masking reconstruction. In addition, semantic alignment is performed at each Feature Pyramid Network (FPN) layer between the teacher and the student networks for generating consistent feature distributions. Our experiments for the object detection task demonstrate the promise of our approach, suggesting that DFMSD outperforms both the state-of-the-art heterogeneous and homogeneous distillation methods.