arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1912.04481cs.LGcs.DC
收藏

SMAUG:面向深度学习工作负载的端到端全栈仿真基础设施

SMAUG: End-to-End Full-Stack Simulation Infrastructure for Deep Learning Workloads

Harvard University(哈佛大学)

Sam Likun Xi, Yuan Yao, Kshitij Bhardwaj, Paul Whatmough, Gu-Yeon Wei, David Brooks

摘要:

近年来,深度神经网络的硬件加速取得了巨大的进步。然而,大多数研究都集中在优化加速器微架构,以提高每层的性能和能效。我们发现,对于整个单批推理延迟,加速器可能只占25- 40%,其余部分用于数据移动和深度学习软件框架。到目前为止,在早期设计阶段(在RTL可用之前)研究端到端DNN性能非常困难,因为没有现有的DNN框架支持端到端仿真和轻松的自定义硬件加速器集成。为了解决研究基础设施中的这一差距,我们提出了SMAUG,这是第一个专门为模拟端到端深度学习应用而构建的DNN框架。SMAUG为研究人员提供了广泛的功能来评估DNN工作负载,从不同的网络拓扑到轻松的加速器建模和SoC集成。为了展示SMAUG的强大功能和价值,我们展示了案例研究,展示了我们如何在不改变加速器微架构的任何部分的情况下,优化整体性能和能效,使加速比达到基线系统的1.8- 5倍,并展示了SMAUG如何为相机供电的深度学习管道调整SoC。

英文摘要:

In recent years, there has been tremendous advances in hardware acceleration of deep neural networks. However, most of the research has focused on optimizing accelerator microarchitecture for higher performance and energy efficiency on a per-layer basis. We find that for overall single-batch inference latency, the accelerator may only make up 25-40%, with the rest spent on data movement and in the deep learning software framework. Thus far, it has been very difficult to study end-to-end DNN performance during early stage design (before RTL is available) because there are no existing DNN frameworks that support end-to-end simulation with easy custom hardware accelerator integration. To address this gap in research infrastructure, we present SMAUG, the first DNN framework that is purpose-built for simulation of end-to-end deep learning applications. SMAUG offers researchers a wide range of capabilities for evaluating DNN workloads, from diverse network topologies to easy accelerator modeling and SoC integration. To demonstrate the power and value of SMAUG, we present case studies that show how we can optimize overall performance and energy efficiency for up to 1.8-5x speedup over a baseline system, without changing any part of the accelerator microarchitecture, as well as show how SMAUG can tune an SoC for a camera-powered deep learning pipeline.

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

↑