Knowledge Distillation (KD) aims to transfer knowledge from a cumbersome teacher network to a more compact student network, enabling efficient deployment on devices with constrained resources. Traditional KD methods rely on balanced datasets to ensure the effectiveness of model training. Nevertheless, in real-life situations, data often show a distribution with a long tail. Due to the limited number of tail samples, current KD methods frequently struggle to obtain sufficient information from these samples, impacting the overall performance. To tackle the challenge of insufficient tail class information, this work introduces a Dynamic Inter-class Correlation Knowledge Distillation(DIC-KD) method. Specifically, we first define dynamic inter-class correlations and extract rich inter-class knowledge from the teacher model. Then, we transfer this knowledge to the student model to enhance its learning ability for various categories, especially for tail class samples, thus improving the overall performance. Intensive experiments demonstrate that DIC-KD on long-tailed datasets can proficiently train stable student networks.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Dynamic Inter-Class Correlation in Knowledge Distillation for Long-Tailed Scenarios

  • Ping Xiang,
  • Ning Jiang,
  • Jialiang Tang,
  • Haodong Liu,
  • Xinlei Huang,
  • Xin Cheng,
  • Yin Long,
  • Jinjia Zhou

摘要

Knowledge Distillation (KD) aims to transfer knowledge from a cumbersome teacher network to a more compact student network, enabling efficient deployment on devices with constrained resources. Traditional KD methods rely on balanced datasets to ensure the effectiveness of model training. Nevertheless, in real-life situations, data often show a distribution with a long tail. Due to the limited number of tail samples, current KD methods frequently struggle to obtain sufficient information from these samples, impacting the overall performance. To tackle the challenge of insufficient tail class information, this work introduces a Dynamic Inter-class Correlation Knowledge Distillation(DIC-KD) method. Specifically, we first define dynamic inter-class correlations and extract rich inter-class knowledge from the teacher model. Then, we transfer this knowledge to the student model to enhance its learning ability for various categories, especially for tail class samples, thus improving the overall performance. Intensive experiments demonstrate that DIC-KD on long-tailed datasets can proficiently train stable student networks.