用数据影响力加权提升模型蒸馏效果,让小模型更高效、可解释。
Knowledge Distillation with Adapted Weight
- 基于影响函数为训练数据动态分配权重,增强蒸馏鲁棒性
- 在多个基准上优于现有方法,半监督学习表现突出
- 兼顾公平性与可解释性,适合需要透明部署的场景
尽管大模型在自然语言和计算机视觉等领域展现出强大的能力,但其庞大的参数量难以在实时系统中部署,受限于计算与能耗。为此,教师-学生架构的知识蒸馏提供了一种可持续的压缩路径,可在不显著损失性能的前提下将大模型知识迁移至小型模型。然而,如何理解单个训练数据对模型性能的影响仍是一个未被充分探索的问题。本文提出知识蒸馏自适应影响权重(KD-AIF)框架,利用稳健统计中的影响函数为训练数据分配权重,遵循可持续性、准确性、公平性和可解释性四项核心原则。该方法不仅优化了蒸馏过程,还通过揭示数据重要性提升了透明度。对不同更新机制的探索表明,该框架能显著提升学生模型的学习效率与泛化能力。KD-AIF在知识蒸馏任务中表现优异,并在多个基准(CIFAR-100、CIFAR-10-4k、SVHN-1k、GLUE)上的半监督学习任务中超越现有基线方法。
原文摘要 · Abstract (English)
Although large models have shown a strong capacity to solve large-scale problems in many areas including natural language and computer vision, their voluminous parameters are hard to deploy in a real-time system due to computational and energy constraints. Addressing this, knowledge distillation through Teacher-Student architecture offers a sustainable pathway to compress the knowledge of large models into more manageable sizes without significantly compromising performance. To enhance the robustness and interpretability of this framework, it is critical to understand how individual training data impact model performance, which is an area that remains underexplored. We propose the \textbf{Knowledge Distillation with Adaptive Influence Weight (KD-AIF)} framework which leverages influence functions from robust statistics to assign weights to training data, grounded in the four key SAFE principles: Sustainability, Accuracy, Fairness, and Explainability. This novel approach not only optimizes distillation but also increases transparency by revealing the significance of different data. The exploration of various update mechanisms within the KD-AIF framework further elucidates its potential to significantly improve learning efficiency and generalization in student models, marking a step toward more explainable and deployable Large Models. KD-AIF is effective in knowledge distillation while also showing exceptional performance in semi-supervised learning with outperforms existing baselines and methods in multiple benchmarks (CIFAR-100, CIFAR-10-4k, SVHN-1k, and GLUE).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。