中文

多样化知识蒸馏(DKD):一种提升集成模型对抗对抗攻击鲁棒性的解决方案

图像与视频处理 2021-01-08 v3 机器学习

摘要

本文提出一种抵抗对抗攻击的集成学习模型。为构建鲁棒性,我们引入了一种训练过程,其中每个成员学习截然不同的潜空间。成员模型被逐个添加到集成中。同时,损失函数受反向知识蒸馏调节,迫使新成员学习不同的特征并映射到与现有成员安全远离的潜空间。我们使用CIFAR10和MNIST数据集在图像分类任务上评估了所提方案的安全性和性能,并显示出相较于最先进防御方法的安全性和性能提升。

关键词

引用

@article{arxiv.2006.15127,
  title  = {Diverse Knowledge Distillation (DKD): A Solution for Improving The Robustness of Ensemble Models Against Adversarial Attacks},
  author = {Ali Mirzaeian and Jana Kosecka and Houman Homayoun and Tinoosh Mohsenin and Avesta Sasan},
  journal= {arXiv preprint arXiv:2006.15127},
  year   = {2021}
}