Saved in:
Bibliographic Details
Main Authors: Deng, Jieren, Palmer, Aaron, Mahmood, Rigel, Rathbun, Ethan, Bi, Jinbo, Mahmood, Kaleel, Aguiar, Derek
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2402.15586
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929256047575040
author Deng, Jieren
Palmer, Aaron
Mahmood, Rigel
Rathbun, Ethan
Bi, Jinbo
Mahmood, Kaleel
Aguiar, Derek
author_facet Deng, Jieren
Palmer, Aaron
Mahmood, Rigel
Rathbun, Ethan
Bi, Jinbo
Mahmood, Kaleel
Aguiar, Derek
contents Achieving resiliency against adversarial attacks is necessary prior to deploying neural network classifiers in domains where misclassification incurs substantial costs, e.g., self-driving cars or medical imaging. Recent work has demonstrated that robustness can be transferred from an adversarially trained teacher to a student model using knowledge distillation. However, current methods perform distillation using a single adversarial and vanilla teacher and consider homogeneous architectures (i.e., residual networks) that are susceptible to misclassify examples from similar adversarial subspaces. In this work, we develop a defense framework against adversarial attacks by distilling adversarial robustness using heterogeneous teachers (DARHT). In DARHT, the student model explicitly represents teacher logits in a student-teacher feature map and leverages multiple teachers that exhibit low adversarial example transferability (i.e., exhibit high performance on dissimilar adversarial examples). Experiments on classification tasks in both white-box and black-box scenarios demonstrate that DARHT achieves state-of-the-art clean and robust accuracies when compared to competing adversarial training and distillation methods in the CIFAR-10, CIFAR-100, and Tiny ImageNet datasets. Comparisons with homogeneous and heterogeneous teacher sets suggest that leveraging teachers with low adversarial example transferability increases student model robustness.
format Preprint
id arxiv_https___arxiv_org_abs_2402_15586
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Distilling Adversarial Robustness Using Heterogeneous Teachers
Deng, Jieren
Palmer, Aaron
Mahmood, Rigel
Rathbun, Ethan
Bi, Jinbo
Mahmood, Kaleel
Aguiar, Derek
Computer Vision and Pattern Recognition
Cryptography and Security
Achieving resiliency against adversarial attacks is necessary prior to deploying neural network classifiers in domains where misclassification incurs substantial costs, e.g., self-driving cars or medical imaging. Recent work has demonstrated that robustness can be transferred from an adversarially trained teacher to a student model using knowledge distillation. However, current methods perform distillation using a single adversarial and vanilla teacher and consider homogeneous architectures (i.e., residual networks) that are susceptible to misclassify examples from similar adversarial subspaces. In this work, we develop a defense framework against adversarial attacks by distilling adversarial robustness using heterogeneous teachers (DARHT). In DARHT, the student model explicitly represents teacher logits in a student-teacher feature map and leverages multiple teachers that exhibit low adversarial example transferability (i.e., exhibit high performance on dissimilar adversarial examples). Experiments on classification tasks in both white-box and black-box scenarios demonstrate that DARHT achieves state-of-the-art clean and robust accuracies when compared to competing adversarial training and distillation methods in the CIFAR-10, CIFAR-100, and Tiny ImageNet datasets. Comparisons with homogeneous and heterogeneous teacher sets suggest that leveraging teachers with low adversarial example transferability increases student model robustness.
title Distilling Adversarial Robustness Using Heterogeneous Teachers
topic Computer Vision and Pattern Recognition
Cryptography and Security
url https://arxiv.org/abs/2402.15586