arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.10375eess.SPcs.LGstat.ML
收藏

FTT-NAS:发现容错卷积神经结构

FTT-NAS: Discovering Fault-Tolerant Convolutional Neural Architecture

Tsinghua University(清华大学);Tencent(腾讯);Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所);Tianjin University(天津大学)

Xuefei Ning, Guangjun Ge, Wenshuo Li, Zhenhua Zhu, Yin Zheng, Xiaoming Chen, Zhen Gao, Yu Wang, Huazhong Yang

摘要:

随着嵌入式深度学习计算系统的快速发展,由深度学习驱动的应用程序正在从云端移动到边缘。当在复杂环境下将神经网络(NN)部署到设备上时,存在各种类型的可能故障:由宇宙辐射和放射性杂质引起的软错误,电压不稳定,老化,温度变化和恶意攻击者。因此,部署NN的安全风险现在引起了人们的广泛关注。在本文中,在分析了各种类型的神经网络加速器中可能出现的故障后,我们从算法的角度形式化并实现了各种故障模型。我们提出了容错神经架构搜索(FT-NAS),以自动发现卷积神经网络(CNN)架构,是可靠的各种故障,在当今的设备。然后,我们将容错训练(FTT)在搜索过程中,以达到更好的结果,这是被称为FTT-NAS。在CIFAR-10上的实验表明,所发现的架构明显优于其他手动设计的基准架构,具有可比或更少的浮点运算(FLOP)和参数。具体而言,在相同的故障设置下,在特征故障模型下发现的F-FTT-Net达到了86.2%的准确率(与MobileNet-V2实现的68.1%相比),在权重故障模型下发现的W-FTT-Net达到了69.6%的准确率(与ResNet-20实现的60.8%相比)。通过对所发现的结构的考察,我们发现操作原语、权值量化范围、模型容量和连接模式对神经网络模型的故障恢复能力都有影响。

英文摘要:

With the fast evolvement of embedded deep-learning computing systems, applications powered by deep learning are moving from the cloud to the edge. When deploying neural networks (NNs) onto the devices under complex environments, there are various types of possible faults: soft errors caused by cosmic radiation and radioactive impurities, voltage instability, aging, temperature variations, and malicious attackers. Thus the safety risk of deploying NNs is now drawing much attention. In this paper, after the analysis of the possible faults in various types of NN accelerators, we formalize and implement various fault models from the algorithmic perspective. We propose Fault-Tolerant Neural Architecture Search (FT-NAS) to automatically discover convolutional neural network (CNN) architectures that are reliable to various faults in nowadays devices. Then we incorporate fault-tolerant training (FTT) in the search process to achieve better results, which is referred to as FTT-NAS. Experiments on CIFAR-10 show that the discovered architectures outperform other manually designed baseline architectures significantly, with comparable or fewer floating-point operations (FLOPs) and parameters. Specifically, with the same fault settings, F-FTT-Net discovered under the feature fault model achieves an accuracy of 86.2% (VS. 68.1% achieved by MobileNet-V2), and W-FTT-Net discovered under the weight fault model achieves an accuracy of 69.6% (VS. 60.8% achieved by ResNet-20). By inspecting the discovered architectures, we find that the operation primitives, the weight quantization range, the capacity of the model, and the connection pattern have influences on the fault resilience capability of NN models.

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

↑