Advertisement

Speedup Critical Stage of Machine Learning with Batch Scheduling in GPU

  • Yuan Gao
  • Rui Wang
  • Ning An
  • Yanjiang Wei
  • Depei Qian
Part of the Lecture Notes in Computer Science book series (LNCS, volume 8707)

Abstract

As a superior data analysis method, Machine Learning suffers the bottleneck from limited computing capability for many years. With the advent of numerous parallel computing hardwares, modern GPU is becoming a promising carrier for the tasks of Machine Learning. In this paper, we propose an efficient GPU execution framework to speedup the forward propagation process of convolution neural network. By extending the convolution unrolling method to fit this batch mode, we get a significant increase of throughput but very little overhead.

Keywords

convolution neural network framework GPU batch process 

Copyright information

© IFIP International Federation for Information Processing 2014

Authors and Affiliations

  • Yuan Gao
    • 1
  • Rui Wang
    • 1
  • Ning An
    • 1
  • Yanjiang Wei
    • 1
  • Depei Qian
    • 1
  1. 1.BeiHang UniversityBeijingChina

Personalised recommendations