Title :
MASCOT: Fast and Highly Scalable SVM Cross-Validation Using GPUs and SSDs
Author :
Zeyi Wen ; Rui Zhang ; Ramamohanarao, Kotagiri ; Jianzhong Qi ; Taylor, Kerry
Author_Institution :
Univ. of Melbourne, Melbourne, VIC, Australia
Abstract :
Cross-validation is a commonly used method for evaluating the effectiveness of Support Vector Machines (SVMs). However, existing SVM cross-validation algorithms are not scalable to large datasets because they have to (i) hold the whole dataset in memory and/or (ii) perform a very large number of kernel value computation. In this paper, we propose a scheme to dramatically improve the scalability and efficiency of SVM cross-validation through the following key ideas. (i) To avoid holding the whole dataset in the memory and avoid performing repeated kernel value computation, we precompute the kernel values and reuse them. (ii) We store the precomputed kernel values to a high-speed storage framework, consisting of CPU memory extended by solid state drives (SSDs) and GPU memory as a cache, so that reusing (i.e., Reading) kernel values takes much lesser time than computing them on-the-fly. (iii) To further improve the efficiency of the SVM training, we apply a number of techniques for the extreme example search algorithm, design a parallel kernel value read algorithm, propose a caching strategy well-suited to the characteristics of the storage framework, and parallelize the tasks on the GPU and the CPU. For datasets of sizes that existing algorithms can handle, our scheme achieves several orders of magnitude of speedup. More importantly, our scheme enables SVM cross-validation on datasets of very large scale that existing algorithms are unable to handle.
Keywords :
cache storage; disc drives; graphics processing units; storage management; support vector machines; CPU memory; GPU memory; MASCOT; SSD; caching strategy; high-speed storage framework; parallel kernel value read algorithm; scalable SVM cross-validation; solid state drive; support vector machine; Algorithm design and analysis; Equations; Graphics processing units; Instruction sets; Kernel; Support vector machines; Training; Cross-validation; GPUs; SSDs; SVM; Training;
Conference_Titel :
Data Mining (ICDM), 2014 IEEE International Conference on
Conference_Location :
Shenzhen
Print_ISBN :
978-1-4799-4303-6
DOI :
10.1109/ICDM.2014.35