提出轻量级抗攻击神经网络,兼顾小体积与强鲁棒性。
Quantitative Analysis of Deeply Quantized Tiny Neural Networks Robust to Adversarial Attacks
- 用量化训练+逐层雅可比正则化协同优化模型
- 在CIFAR-10和语音数据集上抗攻性能优于现有基准
- 适合边缘设备部署,对对抗扰动敏感度低
降低机器学习模型尤其是深度神经网络(DNN)的内存占用,对在资源受限的边缘设备上部署至关重要。然而,DNN易受对抗攻击影响,微小输入扰动即可导致误判。本文提出一种紧凑型DNN模型,对黑盒与白盒攻击均具鲁棒性。通过QKeras量化感知训练框架结合雅可比正则化(Jacobian Regularization, JR)进行逐层协同优化,并采用随机三值量化(Stochastic Ternary Quantization, STQ)实现压缩。在CIFAR-10图像与Google Speech Commands语音数据集上,该模型在面对多种白盒与黑盒攻击时,平均表现优于Quanos和MLCommons/TinyML(MLC/T)基准。模型具备小体积与高鲁棒性的双重优势。
原文摘要 · Abstract (English)
Reducing the memory footprint of Machine Learning (ML) models, especially Deep Neural Networks (DNNs), is imperative to facilitate their deployment on resource-constrained edge devices. However, a notable drawback of DNN models lies in their susceptibility to adversarial attacks, wherein minor input perturbations can deceive them. A primary challenge revolves around the development of accurate, resilient, and compact DNN models suitable for deployment on resource-constrained edge devices. This paper presents the outcomes of a compact DNN model that exhibits resilience against both black-box and white-box adversarial attacks. This work has achieved this resilience through training with the QKeras quantization-aware training framework. The study explores the potential of QKeras and an adversarial robustness technique, Jacobian Regularization (JR), to co-optimize the DNN architecture through per-layer JR methodology. As a result, this paper has devised a DNN model employing this co-optimization strategy based on Stochastic Ternary Quantization (STQ). Its performance was compared against existing DNN models in the face of various white-box and black-box attacks. The experimental findings revealed that, the proposed DNN model had small footprint and on average, it exhibited better performance than Quanos and DS-CNN MLCommons/TinyML (MLC/T) benchmarks when challenged with white-box and black-box attacks, respectively, on the CIFAR-10 image and Google Speech Commands audio datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。