arXiv:2412.16493cs.CV2024-12中稿 · ACM Multimedia 202…被引 14

通过视图一致性正则化提升知识蒸馏效果,无需增加参数。

Cross-View Consistency Regularisation for Knowledge Distillation

  • 引入视内与跨视图正则化,缓解教师过自信和确认偏差问题。
  • 在CIFAR-100、Tiny-ImageNet和ImageNet上达到新SOTA性能。
  • 方法简洁有效,适用于多种师生模型架构,无额外参数开销。

知识蒸馏(KD)是将复杂模型的知识迁移至轻量高效模型的成熟范式。近年来,基于logit的蒸馏方法在性能上快速逼近基于特征的方法。然而,先前研究指出,这类方法在训练过程中仍受两大核心问题制约:教师过自信和确认偏差。受半监督学习中跨视图学习成功的启发,本文在标准logit蒸馏框架中引入视内与跨视图正则化,以解决上述关键问题。同时,采用基于置信度的软标签挖掘策略,提升教师蒸馏信号质量,进一步缓解确认偏差。尽管方法看似简单,所提出的基于一致性正则化的logit蒸馏(CRLD)显著提升了学生模型的学习效果,在CIFAR-100、Tiny-ImageNet和ImageNet等多个数据集上,针对多种师生模型架构均取得新SOTA结果,且不引入额外网络参数。该方法与现有logit蒸馏研究正交,具备优异泛化能力,无需复杂技巧即可大幅提升多种已有方法的性能。

原文摘要 · Abstract (English)

Knowledge distillation (KD) is an established paradigm for transferring privileged knowledge from a cumbersome model to a lightweight and efficient one. In recent years, logit-based KD methods are quickly catching up in performance with their feature-based counterparts. However, previous research has pointed out that logit-based methods are still fundamentally limited by two major issues in their training process, namely overconfident teacher and confirmation bias. Inspired by the success of cross-view learning in fields such as semi-supervised learning, in this work we introduce within-view and cross-view regularisations to standard logit-based distillation frameworks to combat the above cruxes. We also perform confidence-based soft label mining to improve the quality of distilling signals from the teacher, which further mitigates the confirmation bias problem. Despite its apparent simplicity, the proposed Consistency-Regularisation-based Logit Distillation (CRLD) significantly boosts student learning, setting new state-of-the-art results on the standard CIFAR-100, Tiny-ImageNet, and ImageNet datasets across a diversity of teacher and student architectures, whilst introducing no extra network parameters. Orthogonal to on-going logit-based distillation research, our method enjoys excellent generalisation properties and, without bells and whistles, boosts the performance of various existing approaches by considerable margins.

知识蒸馏一致性正则图像分类

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