arXiv:2507.00012cs.LGcs.AI2025-07被引 5

通过最小化条件互信息,训练出无法被知识蒸馏的模型。

Towards Undistillable Models by Minimizing Conditional Mutual Information

  • 用条件互信息衡量输出分布簇的集中度,指导模型训练。
  • 在多种蒸馏方法下,蒸馏出的学生均不如独立训练的平滑标签学生。
  • 既能防蒸馏,又能提升自身准确率,适合模型保护场景。

深度神经网络(DNN)若作为黑盒教师时无法通过知识蒸馏(KD)被有效压缩,则称为不可蒸馏模型。此时,由该教师蒸馏出的‘仿冒学生’(knockoff student)在预测准确率上低于独立训练的标签平滑学生(LS student)。为保护DNN知识产权,构建不可蒸馏模型至关重要。研究发现,不可蒸馏模型的输出概率分布对同一标签样本的响应应高度集中,理想情况下每类分布应坍缩为单一分布。基于此,提出一种新的训练方法——条件互信息最小化(CMIM),通过联合最小化交叉熵损失和温度缩放后所有簇的条件互信息(CMI),在整个温度范围内优化模型。大量实验表明,经CMIM训练的模型无法被文献中所有测试的蒸馏方法成功蒸馏,其生成的敲诈学生均劣于对应标签平滑学生。此外,CMIM模型自身预测精度也优于仅使用交叉熵损失训练的模型。

原文摘要 · Abstract (English)

A deep neural network (DNN) is said to be undistillable if, when used as a black-box input-output teacher, it cannot be distilled through knowledge distillation (KD). In this case, the distilled student (referred to as the knockoff student) does not outperform a student trained independently with label smoothing (LS student) in terms of prediction accuracy. To protect intellectual property of DNNs, it is desirable to build undistillable DNNs. To this end, it is first observed that an undistillable DNN may have the trait that each cluster of its output probability distributions in response to all sample instances with the same label should be highly concentrated to the extent that each cluster corresponding to each label should ideally collapse into one probability distribution. Based on this observation and by measuring the concentration of each cluster in terms of conditional mutual information (CMI), a new training method called CMI minimized (CMIM) method is proposed, which trains a DNN by jointly minimizing the conventional cross entropy (CE) loss and the CMI values of all temperature scaled clusters across the entire temperature spectrum. The resulting CMIM model is shown, by extensive experiments, to be undistillable by all tested KD methods existing in the literature. That is, the knockoff students distilled by these KD methods from the CMIM model underperform the respective LS students. In addition, the CMIM model is also shown to performs better than the model trained with the CE loss alone in terms of their own prediction accuracy.

知识蒸馏模型保护互信息

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