arXiv:2409.07694cs.CV2024-09被引 2

解决长尾数据中教师模型偏倚问题,让小类别也能学好知识

Learn from Balance: Rectifying Knowledge Transfer for Long-Tailed Scenarios

  • 用均衡类别先验修正教师模型的偏见
  • 在多个长尾数据集上显著提升小类别的识别准确率
  • 适合部署在资源受限设备上的轻量级模型训练

知识蒸馏(KD)将大型预训练教师网络的知识迁移到紧凑高效的学生网络中,适用于资源受限的终端设备部署。然而,传统KD方法需要平衡数据以保证训练鲁棒性,这在实际应用中往往难以满足。在长尾场景下,少数头部类别占据大量样本,导致教师网络偏向头部类别,进而使教师和学生网络在尾部类别上性能严重下降。本文提出一种新框架——知识修正蒸馏(KRDistill),通过引入均衡类别先验来纠正教师网络中继承的不平衡知识,并重点修正教师对尾部类别的偏差预测。由此,教师网络能提供更均衡、准确的知识,用于训练可靠的轻量学生网络。在多个长尾数据集上的大量实验表明,该方法能在真实数据不平衡场景下有效训练出可靠的学生网络。

原文摘要 · Abstract (English)

Knowledge Distillation (KD) transfers knowledge from a large pre-trained teacher network to a compact and efficient student network, making it suitable for deployment on resource-limited media terminals. However, traditional KD methods require balanced data to ensure robust training, which is often unavailable in practical applications. In such scenarios, a few head categories occupy a substantial proportion of examples. This imbalance biases the trained teacher network towards the head categories, resulting in severe performance degradation on the less represented tail categories for both the teacher and student networks. In this paper, we propose a novel framework called Knowledge Rectification Distillation (KRDistill) to address the imbalanced knowledge inherited in the teacher network through the incorporation of the balanced category priors. Furthermore, we rectify the biased predictions produced by the teacher network, particularly focusing on the tail categories. Consequently, the teacher network can provide balanced and accurate knowledge to train a reliable student network. Intensive experiments conducted on various long-tailed datasets demonstrate that our KRDistill can effectively train reliable student networks in realistic scenarios of data imbalance.

知识蒸馏长尾学习模型压缩

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。