用随机教师表示提升模型性能,无需多模型训练。
Learning from Stochastic Teacher Representations Using Student-Guided Knowledge Distillation
- 训练单个模型,通过训练时丢弃生成多样教师表征。
- 学生引导蒸馏过滤噪声,仅保留任务相关知识。
- 适合资源受限设备,性能超越现有方法且开销极小。
自蒸馏进展表明,当使用相同深度学习架构从教师到学生蒸馏知识时,若网络过参数化且教师采用早停训练,学生性能可超越教师。集成学习虽能提升性能,但随着模型数量增加,训练、存储和部署成本急剧上升。即使将集成蒸馏为单个学生或使用权重平均,仍需训练多个教师模型,且未能充分利用深度学习模型内在的随机性来生成和蒸馏多样性。这一限制在可穿戴设备等资源受限或延迟敏感场景中尤为突出。本文提出仅训练一个模型,通过蒸馏时的丢弃机制生成多个多样化的教师表征。然而,随机生成的表征存在噪声且与任务不一致。为此,引入一种新型随机自蒸馏(SSD)策略,利用学生引导的知识蒸馏(SGKD)对教师表征进行过滤和加权,仅蒸馏任务相关的表示。每个蒸馏步骤中的学生表征作为指导权威。在真实情感计算、来自UCR档案的可穿戴/生物信号数据集、HAR数据集及图像分类数据集上的实验结果表明,所提SSD方法在不增加模型大小的情况下,优于当前最优方法,且在训练和测试阶段计算复杂度几乎不变,远低于现有集成学习和权重平均方法。
原文摘要 · Abstract (English)
Advances in self-distillation have shown that when knowledge is distilled from a teacher to a student using the same deep learning (DL) architecture, the student performance can surpass the teacher particularly when the network is overparameterized and the teacher is trained with early stopping. Alternatively, ensemble learning also improves performance, although training, storing, and deploying multiple models becomes impractical as the number of models grows. Even distilling an ensemble to a single student model or weight averaging methods first requires training of multiple teacher models and does not fully leverage the inherent stochasticity for generating and distilling diversity in DL models. These constraints are particularly prohibitive in resource-constrained or latency-sensitive applications such as wearable devices. This paper proposes to train only one model and generate multiple diverse teacher representations using distillation-time dropout. However, generating these representations stochastically leads to noisy representations that are misaligned with the learned task. To overcome this problem, a novel stochastic self-distillation (SSD) training strategy is introduced for filtering and weighting teacher representation to distill from task-relevant representations only, using student-guided knowledge distillation (SGKD). The student representation at each distillation step is used as authority to guide the distillation process. Experimental results on real-world affective computing, wearable/biosignal datasets from the UCR Archive, the HAR dataset, and image classification datasets show that the proposed SSD method can outperform state-of-the-art methods without increasing the model size at both training and testing time, and incurs negligible computational complexity compared to state-of-the-art ensemble learning and weight averaging methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。