RNAS-CL: Robust Neural Architecture Search by Cross-Layer Knowledge Distillation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Nath, Utkarsh, Wang, Yancheng, Yang, Yingzhen
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914907424817152
author Nath, Utkarsh
Wang, Yancheng
Yang, Yingzhen
author_facet Nath, Utkarsh
Wang, Yancheng
Yang, Yingzhen
contents Deep Neural Networks are vulnerable to adversarial attacks. Neural Architecture Search (NAS), one of the driving tools of deep neural networks, demonstrates superior performance in prediction accuracy in various machine learning applications. However, it is unclear how it performs against adversarial attacks. Given the presence of a robust teacher, it would be interesting to investigate if NAS would produce robust neural architecture by inheriting robustness from the teacher. In this paper, we propose Robust Neural Architecture Search by Cross-Layer Knowledge Distillation (RNAS-CL), a novel NAS algorithm that improves the robustness of NAS by learning from a robust teacher through cross-layer knowledge distillation. Unlike previous knowledge distillation methods that encourage close student/teacher output only in the last layer, RNAS-CL automatically searches for the best teacher layer to supervise each student layer. Experimental result evidences the effectiveness of RNAS-CL and shows that RNAS-CL produces small and robust neural architecture.
format Preprint
id arxiv_https___arxiv_org_abs_2301_08092
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle RNAS-CL: Robust Neural Architecture Search by Cross-Layer Knowledge Distillation
Nath, Utkarsh
Wang, Yancheng
Yang, Yingzhen
Computer Vision and Pattern Recognition
Machine Learning
Deep Neural Networks are vulnerable to adversarial attacks. Neural Architecture Search (NAS), one of the driving tools of deep neural networks, demonstrates superior performance in prediction accuracy in various machine learning applications. However, it is unclear how it performs against adversarial attacks. Given the presence of a robust teacher, it would be interesting to investigate if NAS would produce robust neural architecture by inheriting robustness from the teacher. In this paper, we propose Robust Neural Architecture Search by Cross-Layer Knowledge Distillation (RNAS-CL), a novel NAS algorithm that improves the robustness of NAS by learning from a robust teacher through cross-layer knowledge distillation. Unlike previous knowledge distillation methods that encourage close student/teacher output only in the last layer, RNAS-CL automatically searches for the best teacher layer to supervise each student layer. Experimental result evidences the effectiveness of RNAS-CL and shows that RNAS-CL produces small and robust neural architecture.
title RNAS-CL: Robust Neural Architecture Search by Cross-Layer Knowledge Distillation
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2301.08092