Hyperparameter Estimation in SVM with GPU Acceleration for Prediction of Protein-Protein Interactions

For classification tasks, such as protein-protein interactions (PPI), support vector machines (SVMs) have been continually utilised as a standard machine learning model. However, most practices in PPIs classifications are limited to common circumstances with small datasets and low feature dimensions, due to the big computation burden of kernel functions and quadratic optimization of SVM. Alternatively, these practical experiences might tend to employ a linear model once the dataset becomes larger, which may have exclusively lost the kernel function’s potential. Since there are different defined kernels and various groups of hyperparameter, the time costs in estimating a best set of hyperparameter by traditional grid search are subsequently tremendous for PPI classification. To address this challenge, in this paper, we present a more efficient solution of hyperparameter estimation by gaining acceleration with GPU, which trains SVM efficiently and accurately with kernel functions calculation accelerated on various PPI datasets. The experiments are firstly conducted on PPI classification task, and we have exclusively evaluated the effectiveness on five public classification datasets. Our solution demonstrates a faster and more accurate performance comparing with the state-of-the-art.