首页|期刊导航|计算机科学与探索|反向最大化学习:反蒸馏场景下的暗知识清除

反向最大化学习:反蒸馏场景下的暗知识清除OA

Reversed Maximization Learning:Dark Knowledge Elimination for Anti-distillation

中文摘要英文摘要

知识蒸馏是一种通过传递教师模型的学习表征能力来增强学生模型性能的技术.然而,随着无数据知识蒸馏技术的发展,这种技术也可能被用于不正当的模型复制或增强,导致模型知识产权受到侵害,模型安全面临挑战.为应对这一挑战,反蒸馏技术应运而生,旨在去除教师模型中的暗知识,防止潜在的盗窃者通过知识蒸馏学习到教师模型的特征提取能力.现有研究主要通过生成多峰的输出来混淆学生模型,然而这种方法可能会在反蒸馏模型的训练过程中引入噪声,从而影响模型性能.为了解决这一问题,提出了一种名为反向最大化学习的方法,旨在训练一个能够保持原模型性能的反蒸馏模型,降低知识蒸馏过程中的潜在风险.该方法通过二元概率机制将原模型输出中的正负类解耦,同时通过反向排序模块学习原模型反向的负类输出,消除反蒸馏模型输出中负类之间的置信度差异以破坏知识蒸馏,同时保持正类的置信度优势以维持原始性能.在Cifar-100和ImageNet200等数据集以及多种模型上的实验表明提出的方法在有数据和无数据两种知识蒸馏场景下均具有显著的有效性,优于对比方法.

Knowledge distillation is a technique that enhances the performance of a student model by transferring the rep-resentational capabilities learnt by a teacher model.However,with the development of data-free knowledge distillation,this technology may also be exploited for unauthorized model replication or enhancement,posing risks of intellectual property infringement and challenges to model security.To address this issue,anti-distillation techniques have emerged,aiming to remove the dark knowledge from the teacher model,thereby preventing potential adversaries from extracting the feature representation capabilities of the teacher model through knowledge distillation.Current research mainly focuses on generating multi-peak outputs to confuse the student model during knowledge distillation,thereby mitigating infringe-ment in knowledge distillation.However,this approach may introduce noise during the training of anti-distillation models,potentially degrading their performance.To tackle this problem,this paper proposes a novel method termed reversed maxi-mization learning(RML),which aims to train an anti-distillation model that preserves the original model's performance while reducing the potential risks associated with knowledge distillation.The proposed method decouples the positive and negative classes in the output of original model through binary probability mechanism.Meanwhile,it employs reverse-ranking module to learn the inverted negative-class outputs of the original model,thereby eliminating confidence differences among negative classes in the anti-distillation model to disrupt knowledge distillation,while preserving the confi-dence advantage of the positive class to maintain the original performance.Extensive experiments on datasets such as Cifar-100 and ImageNet200,as well as across various model architectures,demonstrate that the proposed method achieves significant effectiveness in both data-driven and data-free knowledge distillation scenarios,outperforming comparison approaches.

张瑶;李阳;潘志松

中国人民解放军陆军工程大学指挥控制工程学院,南京 210000中国人民解放军陆军工程大学指挥控制工程学院,南京 210000中国人民解放军陆军工程大学指挥控制工程学院,南京 210000

信息技术与安全科学

知识蒸馏反蒸馏反向最大化学习二元概率分布模型保护

knowledge distillationanti-knowledge distillationreversed maximization learningbinary probability distri-butionmodel protection

《计算机科学与探索》 2026 (1)

143-153,11

国家自然科学基金(62076251).This work was supported by the National Natural Science Foundation of China(62076251).

10.3778/j.issn.1673-9418.2505049

评论