• DocumentCode
    1768930
  • Title

    A fast deep learning system using GPU

  • Author

    Zhilu Chen ; Jing Wang ; Haibo He ; Xinming Huang

  • Author_Institution
    Dept. of Electr. & Comput. Eng., Worcester Polytech. Inst., Worcester, MA, USA
  • fYear
    2014
  • fDate
    1-5 June 2014
  • Firstpage
    1552
  • Lastpage
    1555
  • Abstract
    The invention of deep belief network (DBN) provides a powerful tool for data modeling. The key advantage of DBN is that it is driven by training data only, which can alleviate researchers from the routine of devising explicit models or features for data with complicated distributions. However, as the dimensionality and quantity of data increase, the computing load of training a DBN increases rapidly. Prospectively, the remarkable computing power provided by modern GPU devices can reduce the training time of DBN significantly. As highly efficient computational libraries become available, it provides additional support for GPU based parallel computing. Moreover, GPU server is more affordable and accessible compared with computer cluster or supercomputer. In this paper, we implement a variant of the DBNs, called folded-DBN, on NVIDA´s Tesla K20 GPU. In our simulations, two sets of database are used to train the folded-DBNs on both CPU and GPU platforms. Comparing execution time of the fine-tuning process, the GPU implementation results 7 to 11 times speedup over the CPU platform.
  • Keywords
    belief networks; data handling; data models; graphics processing units; learning (artificial intelligence); parallel processing; DBN; GPU based parallel computing; GPU devices; GPU server; NVIDA Tesla K20 GPU; computer cluster; data dimensionality; data modeling; data quantity; deep belief network; fast deep learning system; fine-tuning process; folded-DBN; supercomputer; training data; Acceleration; Databases; Face; Graphics processing units; Libraries; MATLAB; Training; Deep belief network; Deep learning; GPU; Parallel computing; Restricted Boltzmann machine;
  • fLanguage
    English
  • Publisher
    ieee
  • Conference_Titel
    Circuits and Systems (ISCAS), 2014 IEEE International Symposium on
  • Conference_Location
    Melbourne VIC
  • Print_ISBN
    978-1-4799-3431-7
  • Type

    conf

  • DOI
    10.1109/ISCAS.2014.6865444
  • Filename
    6865444