中文

ILP-M Conv:面向移动端 GPU 单图卷积神经网络推理的卷积算法优化

分布式、并行与集群计算 2019-10-04 v2 计算机视觉与模式识别 性能

摘要

卷积神经网络广泛应用于移动端应用。然而,现有 GPU 卷积算法多为小批量神经网络训练而设计,针对移动端 GPU 上单图卷积神经网络推理算法的研究尚不充分。在讨论使用差异并考察现有卷积算法后,我们提出了 HNTMP 卷积算法。据我们所知,HNTMP 卷积算法相比最流行的 im2col 卷积算法实现了 14.6 倍的加速,相比最快的现有卷积算法(直接卷积)实现了 2.30 倍的加速。

关键词

引用

@article{arxiv.1909.02765,
  title  = {ILP-M Conv: Optimize Convolution Algorithm for Single-Image Convolution Neural Network Inference on Mobile GPUs},
  author = {Zhuoran Ji},
  journal= {arXiv preprint arXiv:1909.02765},
  year   = {2019}
}