Efficient Support Vector Machine Training Algorithm on GPUs

Authors

  • Jiashuai Shi South China University of Technology
  • Zeyi Wen National University of Singapore
  • Bingsheng He National University of Singapore
  • Jian Chen South China University of Technology

DOI:

https://doi.org/10.1609/aaai.v32i1.12163

Keywords:

SVM, GPU, Optimization

Abstract

Support Vector Machines (SVMs) are popular for many machine learning tasks. With rapid growth of dataset size, the high cost of training limits the wide use of SVMs. Several SVM implementations on GPUs have been proposed to accelerate SVMs. However, they support only classification (SVC) or regression (SVR). In this work, we propose a simple and effective SVM training algorithm on GPUs which can be used for SVC, SVR and one-class SVM. Initial experiments show that our implementation outperforms existing ones. We are in the process of encapsulating our algorithm into an easy-to-use library which has Python, R and MATLAB interfaces.

Downloads

Published

2018-04-29

How to Cite

Shi, J., Wen, Z., He, B., & Chen, J. (2018). Efficient Support Vector Machine Training Algorithm on GPUs. Proceedings of the AAAI Conference on Artificial Intelligence, 32(1). https://doi.org/10.1609/aaai.v32i1.12163