Relative difficulty distillation for semantic segmentation

doi:10.1007/s11432-023-4061-2

Relative difficulty distillation for semantic segmentation

2024-08-20

0

PRE

AI

黄圣君 (Sheng-Jun Huang)

DOI：10.1007/s11432-023-4061-2

Original

Original request for help

Save

Abstract

En 中文

Current knowledge distillation (KD) methods primarily focus on transferring various structured knowledge and designing corresponding optimization goals to encourage the student network to imitate the output of the teacher network. However, introducing too many additional optimization objectives may lead to unstable training, such as gradient conflicts. Moreover, these methods ignored the guidelines of relative learning difficulty between the teacher and student networks. Inspired by human cognitive science, in this paper, we redefine knowledge from a new perspective - the student and teacher networks' relative difficulty of samples, and propose a pixel-level KD paradigm for semantic segmentation named relative difficulty distillation (RDD). We propose a two-stage RDD framework: teacher-full evaluated RDD (TFE-RDD) and teacher-student evaluated RDD (TSE-RDD). RDD allows the teacher network to provide effective guidance on learning focus without additional optimization goals, thus avoiding adjusting learning weights for multiple losses. Extensive experimental evaluations using a general distillation loss function on popular datasets such as Cityscapes, CamVid, Pascal VOC, and ADE20k demonstrate the effectiveness of RDD against state-of-the-art KD methods. Additionally, our research showcases that RDD can integrate with existing KD methods to improve their upper performance bound. Codes are available at https://github.com/sunyueue/RDD.git.

Keywords:

knowledge distillation

semantic segmentation

relative difficulty

sample weighting

prediction discrepancy

Relative difficulty distillation for semantic segmentation

Abstract

Journal

Organization