CNN卷积计算在移动GPU上的加速研究

• 作者: 文梅 时洋 王湘新
• 来源期刊: 计算机工程与科学
• 学科: 工学
• 关键词: CNN 手机 移动GPU 快速算法 OpenCL
• 年，卷（期）: 2018,（1）
• 所属期刊栏目: 高性能计算
• 页码范围: 34-39
• 页数: 6页
• 分类号: TP391.4
• 字数: 4095字
• 语种: 中文
• DOI: 10.3969/j.issn.1007-130X.2018.01.005

摘要

卷积神经网络(CNN)凭借其优秀的表现正在诸如图像分类、语音识别等领域里扮演着越来越重要的角色,已经有一些研究人员想要将这个深度学习过程复制到手机上.但是,由于CNN巨大的计算量,移植程序的性能一直难以令人满意.为了探讨如何解决这一问题,借助MXNet这样一个深度学习的框架在手机上实现了CNN的前向过程,并且将注意力放在了使用手机上另一个强大的计算设备——GPU上.最终选择使用OpenCL通用编程框架将前向过程中最耗时的卷积操作利用矩阵乘来完成,并转移到GPU上进行.在此基础之上还针对手机GPU做了一些优化.最终,实验结果显示我们成功地将前向过程的时间降低到了原来时间的一半.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	文梅	国防科技大学计算机学院	16	49	4.0	6.0
2	王湘新		1	7	1.0	1.0
3	时洋	国防科技大学计算机学院	1	7	1.0	1.0