计算机研究与发展2024,Vol.61Issue(6) :1370-1387.DOI:10.7544/issn1000-1239.202331005

容错深度学习加速器跨层优化

Cross-Layer Optimization for Fault-Tolerant Deep Learning Accelerators

张青 刘成 刘波 黄海同 王颖 李华伟 李晓维
计算机研究与发展2024,Vol.61Issue(6) :1370-1387.DOI:10.7544/issn1000-1239.202331005

容错深度学习加速器跨层优化

Cross-Layer Optimization for Fault-Tolerant Deep Learning Accelerators

张青 1刘成 1刘波 2黄海同 1王颖 1李华伟 1李晓维1
扫码查看

作者信息

  • 1. 处理器芯片全国重点实验室(中国科学院计算技术研究所) 北京 100190;中国科学院大学 北京 100049
  • 2. 北京控制工程研究所 北京 100094
  • 折叠

摘要

容错深度学习加速器是保障高可靠深度学习的基石,也是深度学习应用于安全关键领域如宇航、机器人等面临的一个关键环节.然而,深度学习计算和访存都非常密集,传统基于冗余计算的容错方法直接应用于深度学习加速器的容错设计会导致严重的功耗、芯片面积等硬件资源开销.为此,从神经元计算任务和神经元的数据位宽 2个维度挖掘深度学习模型对于故障的敏感度差异,并利用这些差异从架构和电路层分别对于敏感的部分提供更多的保护以降低容错代价.同时,利用深度学习自身的容错特性,通过限制量化缩小电路层需要保护的电路逻辑规模.最后,利用贝叶斯优化协同优化算法、架构和电路的跨层设计参数,在保障深度学习可靠性、精度以及性能的前提下,最小化硬件资源开销.

Abstract

Fault-tolerant deep learning accelerator is the basis for highly reliable deep learning processing,and is also critical to deploy deep learning in safety-critical applications such as avionics and robotics.Since deep learning is known to be both computing-intensive and memory-intensive,traditional fault-tolerant approaches based on redundant computing will incur substantial overhead including power consumption and chip area.To this end,we propose to characterize deep learning vulnerability difference across both neurons and bits of each neuron,and leverage the vulnerability difference to enable selective protection of the deep learning processing components from the perspective of architecture layer and circuit layer respectively for the sake of lower fault-tolerant design overhead.At the same time,we observe the correlation between model quantization and bit protection overhead of the underlying processing elements of deep learning accelerators,and propose to reduce the bit protection overhead by adding additional quantization constrain without compromising the model accuracy.Finally,we employ Bayesian optimization strategy to co-optimize the correlated cross-layer design parameters at algorithm layer,architecture layer,and circuit layer to minimize the hardware resource consumption while fulfilling multiple user constraints including reliability,accuracy,and performance of the deep learning processing at the same time.

关键词

跨层优化/容错深度学习加速器/脆弱因子/异构架构/选择性冗余

Key words

cross-layer optimization/fault-tolerant deep learning accelerator/vulnerability factor/hybrid architecture/selective redundancy

引用本文复制引用

基金项目

国家重点研发计划(2022YFB4500405)

国家自然科学基金(62174162)

空间可信计算与电子信息技术实验室开放基金(OBC)

空间可信计算与电子信息技术实验室开放基金(ETL-2022-07)

出版年

2024
计算机研究与发展
中国科学院计算技术研究所 中国计算机学会

计算机研究与发展

CSTPCDCSCD北大核心
影响因子:2.649
ISSN:1000-1239
参考文献量48
段落导航相关论文