用信息论量化知识蒸馏中的有用信息,提升小模型性能
Quantifying Knowledge Distillation Using Partial Information Decomposition
- 引入部分信息分解法,分离教师模型中与任务相关的信息
- 提出冗余信息蒸馏框架,显著提升在干扰教师下的泛化能力
- 适合关注知识蒸馏理论优化的研究者和部署场景
知识蒸馏通过训练小型学生模型模仿复杂教师模型的内部表征,实现资源受限环境下的模型部署。然而,教师模型的表征可能包含与下游任务无关的干扰信息,若将其蒸馏给容量受限的学生模型,反而会损害性能。本文的核心问题是:知识蒸馏的信息论极限是什么?为此,我们采用部分信息分解(Partial Information Decomposition)来量化并解释传递的知识以及尚未被蒸馏的任务相关信息。理论上证明,任务相关的知识可由教师与学生之间关于任务的冗余信息量精确捕捉。据此,我们提出一种多层级优化方法,将冗余信息作为正则项,构建冗余信息蒸馏(RID)框架。实验表明,该方法在存在干扰教师时仍能实现更稳健、高效的蒸馏,因其精准聚焦于任务相关知识,而非简单对齐表征。
原文摘要 · Abstract (English)
Knowledge distillation deploys complex machine learning models in resource-constrained environments by training a smaller student model to emulate internal representations of a complex teacher model. However, the teacher's representations can also encode nuisance or additional information not relevant to the downstream task. Distilling such irrelevant information can actually impede the performance of a capacity-limited student model. This observation motivates our primary question: What are the information-theoretic limits of knowledge distillation? To this end, we leverage Partial Information Decomposition to quantify and explain the transferred knowledge and knowledge left to distill for a downstream task. We theoretically demonstrate that the task-relevant transferred knowledge is succinctly captured by the measure of redundant information about the task between the teacher and student. We propose a novel multi-level optimization to incorporate redundant information as a regularizer, leading to our framework of Redundant Information Distillation (RID). RID leads to more resilient and effective distillation under nuisance teachers as it succinctly quantifies task-relevant knowledge rather than simply aligning student and teacher representations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。