中文

NeuroTrainer:一种用于深度学习训练的智能内存模块

硬件体系结构 2017-10-13 v1

摘要

本文提出NeuroTrainer,一种带有内存内加速器的智能内存模块,它构成了可扩展架构的基本构件,用于深度神经网络的高效节能训练。所提架构基于在3D内存模块的逻辑层中集成由多个处理引擎组成的同质计算基底。NeuroTrainer利用基于可编程数据流的执行模型,在训练操作的不同阶段优化内存映射和数据重用。编程模型及支持架构利用灵活的数据流高效加速各类DNN的训练。在15nm FinFET工艺下的周期级仿真与综合设计显示出500 GFLOPS/W的能效,以及对包括卷积、循环、多层感知机和混合(CNN+RNN)网络在内的广泛DNN几乎相似的吞吐量。

关键词

引用

@article{arxiv.1710.04347,
  title  = {NeuroTrainer: An Intelligent Memory Module for Deep Learning Training},
  author = {Duckhwan Kim and Taesik Na and Sudhakar Yalamanchili and Saibal Mukhopadhyay},
  journal= {arXiv preprint arXiv:1710.04347},
  year   = {2017}
}