arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2105.08239cs.DCcs.AR
收藏

TRIM:深度神经网络推理和训练加速器的设计空间探索模型

TRIM: A Design Space Exploration Model for Deep Neural Networks Inference and Training Accelerators

Yangjie Qi, Shuo Zhang, Tarek M. Taha

摘要:

在边缘/物联网环境和高性能计算系统中,对用于训练深度神经网络的专用硬件的需求越来越大。由于处理架构、深度神经网络配置和低功耗选项的广泛范围,这种硬件的设计空间非常大。这使得开发深度神经网络处理器非常复杂,特别是对于训练。我们提出TRIM,这是一种基础设施,可以帮助硬件架构师探索深度神经网络加速器的设计空间,以便在早期设计阶段进行推理和训练。该模型在整个网络级别进行评估,同时考虑层间和层内活动。给定应用程序、基本硬件规格和设计目标,TRIM可以快速探索不同的硬件设计选项,选择最佳硬件流程并指导新的硬件架构设计。我们通过基于FPGA的深度神经网络加速器和基于ASIC的架构的实现验证了TRIM。我们还展示了如何使用TRIM探索设计空间,通过几个案例研究。TRIM是一个强大的工具,可以帮助架构师评估不同的硬件选择,以开发有效的推理和培训架构设计。

英文摘要:

There is increasing demand for specialized hardware for training deep neural networks, both in edge/IoT environments and in high-performance computing systems. The design space of such hardware is very large due to the wide range of processing architectures, deep neural network configurations, and dataflow options. This makes developing deep neural network processors quite complex, especially for training. We present TRIM, an infrastructure to help hardware architects explore the design space of deep neural network accelerators for both inference and training in the early design stages. The model evaluates at the whole network level, considering both inter-layer and intra-layer activities. Given applications, essential hardware specifications, and a design goal, TRIM can quickly explore different hardware design options, select the optimal dataflow and guide new hardware architecture design. We validated TRIM with FPGA-based implementation of deep neural network accelerators and ASIC-based architectures. We also show how to use TRIM to explore the design space through several case studies. TRIM is a powerful tool to help architects evaluate different hardware choices to develop efficient inference and training architecture design.

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

↑