arXiv:2410.04140cs.CV2024-10ICLR被引 2

通过动态教师模型保持师生差距,提升知识蒸馏效果

Gap Preserving Distillation by Building Bidirectional Mappings with A Dynamic Teacher

  • 引入动态教师模型,与学生同步训练以维持合理性能差距
  • 在ResNet18上实现最高1.58%准确率提升,无预训练教师时提升1.80%
  • 支持从零训练和微调场景,部署灵活且无需后处理

知识蒸馏旨在将大模型知识迁移到小模型,但两者常存在显著性能差距,影响训练。本文提出一种间隙保持蒸馏(GPD)方法,从头训练一个动态教师模型,与学生模型协同训练,确保整个过程中师生性能差距合理。为增强蒸馏效果,采用参数共享的硬策略,并构建基于逆重参数化(IR)与通道分支重参数化(CBR)的软双向映射。IR可任意扩展动态教师规模,同时保持与学生相同的初始精度;CBR结合参数共享,能直接从训练好的教师中提取有效学生模型,无需额外训练。实验表明,GPD在CNN与Transformer架构上均显著优于现有方法,最高提升1.58%。尤其在无预训练教师场景下,从零训练和微调分别提升1.80%和0.89%。

原文摘要 · Abstract (English)

Knowledge distillation aims to transfer knowledge from a large teacher model to a compact student counterpart, often coming with a significant performance gap between them. We find that a too-large performance gap can hamper the training process, which is also verified in recent studies. To address this, we propose a Gap Preserving Distillation (GPD) method that trains an additional dynamic teacher model from scratch along with training the student to bridge this gap. In this way, it becomes possible to maintain a reasonable performance gap between teacher and student during the whole distillation process. To further strengthen distillation from the dynamic teacher to the student, we develop a hard strategy by enforcing them to share parameters and encouraging parameter inheritance. Besides hard strategy, we also build the soft bidirectional mappings between them which are built on an Inverse Reparameterization (IR) method and a Channel-Branch Reparameterization (CBR) strategy. We highlight that our IR is able to initialize a larger dynamic teacher with an arbitrary expansion ratio, while preserving exactly the same accuracy as the given student model. In this way, it guarantees that the dynamic teacher and student start from the same point and avoid a too large gap in early stage of training. As for our CBR, with parameter-sharing, it directly extracts an effective student model from the well-learned dynamic teacher without any post-training, making our method highly flexible for model deployment. In the experiments, GPD significantly outperforms existing distillation methods on top of both CNNs and transformers architectures, achieving up to 1.58% accuracy improvement. Interestingly, GPD also generalizes well to the scenarios without a pre-trained teacher, including training from scratch and fine-tuning, yielding a large improvement of 1.80% and 0.89% on ResNet18, respectively.

知识蒸馏动态教师模型压缩参数共享

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