arXiv:2410.22184cs.CVcs.AI2024-10中稿 · WACV 2025被引 3

用多数据集训练的教师模型,通过分层特征提炼提升学生模型性能。

Multi-Level Feature Distillation of Joint Teachers Trained on Distinct Image Datasets

  • 构建联合教师架构,在多层级融合不同数据集的特征。
  • 在七个图像分类和三个动作识别基准上超越单一或联合训练模型。
  • 支持异构教师结构,适合需要泛化能力的迁移学习场景。

我们提出一种新型师生框架,从在不同数据集上独立训练的多个教师中蒸馏知识。每个教师均从头训练于其专属数据集,随后整合为联合架构,在多个表征层级融合所有教师的特征,并在所有数据集样本上微调,以聚合通用信息。最后,采用多层级特征蒸馏过程,将知识传递给学生模型,适用于各数据集。我们在七个图像分类与三个动作识别基准上进行实验,学生模型与各教师架构相同,以凸显蒸馏效果。同时验证了异构教师组合的灵活性。结果表明,所提多层级特征蒸馏(MLFD)显著优于仅在单个数据集或一次性联合训练的等效架构。全面消融实验证实每一步训练设计均有充分依据。代码已公开于 https://github.com/AdrianIordache/MLFD。

原文摘要 · Abstract (English)

We propose a novel teacher-student framework to distill knowledge from multiple teachers trained on distinct datasets. Each teacher is first trained from scratch on its own dataset. Then, the teachers are combined into a joint architecture, which fuses the features of all teachers at multiple representation levels. The joint teacher architecture is fine-tuned on samples from all datasets, thus gathering useful generic information from all data samples. Finally, we employ a multi-level feature distillation procedure to transfer the knowledge to a student model for each of the considered datasets. We conduct image classification experiments on seven benchmarks, and action recognition experiments on three benchmarks. To illustrate the power of our feature distillation procedure, the student architectures are chosen to be identical to those of the individual teachers. To demonstrate the flexibility of our approach, we combine teachers with distinct architectures. We show that our novel Multi-Level Feature Distillation (MLFD) can significantly surpass equivalent architectures that are either trained on individual datasets, or jointly trained on all datasets at once. Furthermore, we confirm that each step of the proposed training procedure is well motivated by a comprehensive ablation study. We publicly release our code at https://github.com/AdrianIordache/MLFD.

知识蒸馏多源学习特征融合

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