Abstract
The support vector machine (SVM) is a supervised learning algorithm used for recognizing patterns in data. It is a very popular technique in machine learning and has been successfully used in applications such as image classification, protein classification, and handwriting recognition. However, the computational complexity of the kernelized version of the algorithm grows quadratically with the number of training examples. To tackle this high computational complexity, we have developed a directive-based approach that converts a gradient-ascent based training algorithm for the CPU to an efficient graphics processing unit (GPU) implementation. We compare our GPU-based SVM training algorithm to the standard LibSVM CPU implementation, a highly optimized GPU-LibSVM implementation, as well as to a directive-based OpenACC implementation. The results on different handwritten digit classification datasets demonstrate an important speed-up for the current approach when compared to the CPU and OpenACC versions.Furthermore, our solution is almost as fast and sometimes even faster than the highly optimized CUBLAS-based GPU-LibSVM implementation, without sacrificing the algorithm’s accuracy. Copyright © 2014 JohnWiley & Sons, Ltd.
Original language | English |
---|---|
Pages (from-to) | 2274-2294 |
Number of pages | 21 |
Journal | Concurrency and Computation: Practice and Experience |
Volume | 26 |
Issue number | 7 |
Early online date | 9 Oct 2014 |
DOIs | |
Publication status | Published - 2016 |
Keywords
- support vector machine (SVM)
- machine learning
- computational complexity