中文

多视角对比Logit蒸馏

计算机视觉与模式识别 2025-03-11 v2

摘要

在以往的知识蒸馏研究中,logit蒸馏的重要性常被忽视。为重振logit蒸馏,我们通过基于logit的语义属性重新审视其计算,并探索如何更高效地利用它,提出了新视角。Logit往往包含大量高层语义信息;然而,利用logit计算Kullback-Leibler(KL)散度的传统方法未考虑其语义属性。此外,直接计算KL散度无法充分发挥logit的潜力。为应对这些挑战,我们提出一种新颖高效的logit蒸馏方法——多视角对比Logit蒸馏(MCLD),大幅提升了logit蒸馏的性能与效效。与现有logit蒸馏方法及复杂特征蒸馏方法相比,MCLD在图像分类和迁移学习任务上跨多个数据集(包括CIFAR-100、ImageNet、Tiny-ImageNet和STL-10)达到最先进性能。此外,MCLD展现出卓越的训练效率,在Vision Transformers上的蒸馏表现优异,进一步凸显其显著优势。本研究揭示了logit在知识蒸馏中的巨大潜力,并为未来研究提供有价值见解。

关键词

引用

@article{arxiv.2411.10693,
  title  = {Multi-perspective Contrastive Logit Distillation},
  author = {Qi Wang and Jinjia Zhou},
  journal= {arXiv preprint arXiv:2411.10693},
  year   = {2025}
}

备注

10 pages, 6 figures, 9 tabels, 12 formulas