arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.08035cs.LGcs.AIcs.CV

ISimDL:基于重要性采样的深度学习鲁棒性评估故障注入仿真加速方法

ISimDL: Importance Sampling-Driven Acceleration of Fault Injection Simulations for Evaluating the Robustness of Deep Learning

Technische Universität Wien(维也纳工业大学);New York University Abu Dhabi(纽约大学阿布扎比分校)

Alessio Colucci, Andreas Steininger, Muhammad Shafique

AI总结:

本文提出ISimDL方法,利用神经元敏感性进行重要性采样来加速故障注入仿真,无需模型先验知识即可高效评估深度学习系统鲁棒性,并在故障感知训练中将故障搜索开销降低12倍以上。

摘要:

深度学习(DL)系统已在众多应用中广泛普及,需要专用硬件加速器和芯片。在纳米时代,器件越来越容易受到永久性和瞬态故障的影响。因此,我们需要一种高效的方法来分析先进DL系统对此类故障的韧性,并理解神经加速器芯片中的故障如何在DL应用层面表现为错误,其中故障可能导致不可检测和不可恢复的错误。利用故障注入,我们可以通过在软件层面修改神经元权重和输出,模拟硬件受到瞬态故障影响的情形,从而对DL系统进行韧性研究。现有故障模型缩小了搜索空间,能够加快分析速度,但需要对模型具有先验知识,并且不允许对过滤掉的搜索空间进行进一步分析。因此,我们提出ISimDL,这是一种新颖的方法,利用神经元敏感性生成基于重要性采样的故障场景。在没有任何关于被测模型先验知识的情况下,ISimDL提供了与现有工作等效的搜索空间缩减,同时允许进行长时间仿真以覆盖所有可能的故障,改进了现有模型的要求。我们的实验表明,与随机均匀采样相比,重要性采样在选择关键故障方面的精度最高可提高15倍,并且在少于100个故障内即可达到该精度。此外,我们展示了重要性采样在可靠DNN设计中的另一个实际用例,即故障感知训练(FAT)。通过使用ISimDL选择导致错误的故障,我们可以在DNN训练过程中注入这些故障,以增强DNN对此类故障的鲁棒性。在FAT中使用重要性采样可将寻找导致预定精度下降的故障所需的开销降低12倍以上。

英文摘要:

Deep Learning (DL) systems have proliferated in many applications, requiring specialized hardware accelerators and chips. In the nano-era, devices have become increasingly more susceptible to permanent and transient faults. Therefore, we need an efficient methodology for analyzing the resilience of advanced DL systems against such faults, and understand how the faults in neural accelerator chips manifest as errors at the DL application level, where faults can lead to undetectable and unrecoverable errors. Using fault injection, we can perform resilience investigations of the DL system by modifying neuron weights and outputs at the software-level, as if the hardware had been affected by a transient fault. Existing fault models reduce the search space, allowing faster analysis, but requiring a-priori knowledge on the model, and not allowing further analysis of the filtered-out search space. Therefore, we propose ISimDL, a novel methodology that employs neuron sensitivity to generate importance sampling-based fault-scenarios. Without any a-priori knowledge of the model-under-test, ISimDL provides an equivalent reduction of the search space as existing works, while allowing long simulations to cover all the possible faults, improving on existing model requirements. Our experiments show that the importance sampling provides up to 15x higher precision in selecting critical faults than the random uniform sampling, reaching such precision in less than 100 faults. Additionally, we showcase another practical use-case for importance sampling for reliable DNN design, namely Fault Aware Training (FAT). By using ISimDL to select the faults leading to errors, we can insert the faults during the DNN training process to harden the DNN against such faults. Using importance sampling in FAT reduces the overhead required for finding faults that lead to a predetermined drop in accuracy by more than 12x.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑