Accelerating Collaborative Filtering for Implicit Feedback Datasets using GPUs

TitleAccelerating Collaborative Filtering for Implicit Feedback Datasets using GPUs
Publication TypeConference Paper
Year of Publication2015
AuthorsGates, M., H. Anzt, J. Kurzak, and J. Dongarra
Conference Name2015 IEEE International Conference on Big Data (IEEE BigData 2015)
Date Published11-2015
Conference LocationSanta Clara, CA

In this paper we accelerate the Alternating Least Squares (ALS) algorithm used for generating product recommendations on the basis of implicit feedback datasets. We approach the algorithm with concepts proven to be successful in High Performance Computing. This includes the formulation of the algorithm as a mix of cache-optimized algorithm-specific kernels and standard BLAS routines, acceleration via graphics processing units (GPUs), use of parallel batched kernels, and autotuning to identify performance winners. For benchmark datasets, the multi-threaded CPU implementation we propose achieves more than a 10 times speedup over the implementations available in the GraphLab and Spark MLlib software packages. For the GPU implementation, the parameters of an algorithm-specific kernel were optimized using a comprehensive autotuning sweep. This results in an additional 2 times speedup over our CPU implementation.

Project Tags: