GPUMLib: A New Library to Combine Machine Learning Algorithms with Graphics Processing Units



The Graphics Processing Unit (GPU) is a highly parallel, multi-threaded, many-core device with enormous computational power, especially well-suited to address Machine Learning (ML) problems that can be expressed as data-parallel computations. As problems become increasingly demanding, parallel implementations of ML algorithms become critical for developing hybrid intelligent real-world applications.
The relative low cost of GPUs combined with the unprecedent computational power they offer, makes them particularly well positioned to fulfill the need to automatically analyze and capture relevant information from large amounts of data.
Although in ML field there are countless powerful learning algorithms suitable for a wide range of applications, the true potential of these methods is underused, because many implementations are not openly shared. In the GPU arena the panorama is even worse, because few algorithms have yet been implemented. In order to mitigate this problem we propose the creation of an open source GPU Machine Learning Library (GPUMLib) that aims to provide the basis and the building blocks for the scientific community to develop GPU ML algorithms.
Experimental results on benchmark datasets demonstrate that the GPUMLib algorithms already implemented achieve significant savings over the counterpart CPU implementations. Future work is foreseen towards extending the GPUMLib and its validation in complex hybrid systems.


GPU Computing, Machine Learning


GPU Computing, Machine Learning


10th International Conference on Hybrid Intelligent Systems (IEEE), August 2010


Cited by

Year 2015 : 2 citations

 Strnad, D., & Nerat, A. (2015). Parallel construction of classification trees on a GPU. Concurrency and Computation: Practice and Experience.

 Ashari, A., Tatikonda, S., Boehm, M., Reinwald, B., Campbell, K., Keenleyside, J., & Sadayappan, P. (2015, January). On optimizing machine learning workloads via kernel fusion. In ACM SIGPLAN Notices (Vol. 50, No. 8, pp. 173-182). ACM.

Year 2014 : 1 citations

 Wu, Zheng Yi, "Portable GPU-Based Artificial Neural Networks For Data-Driven Modeling" (2014). Proceedings of the 11th International Conference on Hydroinformatics. CUNY Academic Works.

Year 2013 : 2 citations

 M. Behandish. Generalized GPU-based artificial neural network surrogate model for extended period hydraulic simulation. In World Environmental and Water Resources Congress 2013, pages 716–731, 2013.

 Mohammadreza Baharani, Hamid Noori, Mohammad Aliasgari, and Zain Navabi. High-level design space exploration of locally linear neuro-fuzzy models for embedded systems. Fuzzy Sets and Systems, 2013.

Year 2012 : 1 citations

 Karl Jansson. Performance study of using the direct compute API for implementing support vector machines on GPUs. Department of Computer and Systems Sciences, Stockholm University, 2012.

Year 2010 : 1 citations

 Art Gresham, Literature Review: Blind Audio Source Separation on the GPU, 2010.