一次训练生成多种鲁棒轻量模型,降低算力与碳排放。
ProARD: progressive adversarial robustness distillation: provide wide range of robust students
- 动态网络结构支持宽范围架构,通过权重共享联合训练
- 采样机制下随机选学生效果差,需优化采样策略提升性能
- 适合边缘设备部署,解决多场景轻量化模型训练难题
对抗鲁棒性蒸馏(ARD)已成为提升轻量级深度神经网络对抗攻击鲁棒性的有效方法。现有方法依赖一个大型鲁棒教师网络来训练单一鲁棒轻量学生网络。然而,由于边缘设备多样性及资源限制,当前方法需为每种约束重新从头训练学生网络,带来巨大计算开销和二氧化碳排放。本文提出渐进式对抗鲁棒性蒸馏(ProARD),实现一次训练即可生成多种精度与鲁棒性兼备的学生网络,无需重复训练。首先,基于动态层设计支持宽度、深度和扩展率变化的动态神经网络,以覆盖广泛架构;其次,将最大尺寸的学生网络作为动态教师网络,利用权重共享机制联合优化教师与内部所有学生网络。但因计算所有学生精确梯度成本过高,需采用采样机制选取子集。我们发现,每次迭代中随机采样学生无法产生准确且鲁棒的学生网络。
原文摘要 · Abstract (English)
Adversarial Robustness Distillation (ARD) has emerged as an effective method to enhance the robustness of lightweight deep neural networks against adversarial attacks. Current ARD approaches have leveraged a large robust teacher network to train one robust lightweight student. However, due to the diverse range of edge devices and resource constraints, current approaches require training a new student network from scratch to meet specific constraints, leading to substantial computational costs and increased CO2 emissions. This paper proposes Progressive Adversarial Robustness Distillation (ProARD), enabling the efficient one-time training of a dynamic network that supports a diverse range of accurate and robust student networks without requiring retraining. We first make a dynamic deep neural network based on dynamic layers by encompassing variations in width, depth, and expansion in each design stage to support a wide range of architectures. Then, we consider the student network with the largest size as the dynamic teacher network. ProARD trains this dynamic network using a weight-sharing mechanism to jointly optimize the dynamic teacher network and its internal student networks. However, due to the high computational cost of calculating exact gradients for all the students within the dynamic network, a sampling mechanism is required to select a subset of students. We show that random student sampling in each iteration fails to produce accurate and robust students.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。