SVM and SVM ensembles in breast cancer prediction

Min Wei Huang, Chih Wen Chen, Wei Chao Lin, Shih Wen Ke, Chih Fong Tsai

Research output: Contribution to journalArticlepeer-review

121 Scopus citations

Abstract

Breast cancer is an all too common disease in women, making how to effectively predict it an active research problem. A number of statistical and machine learning techniques have been employed to develop various breast cancer prediction models. Among them, support vector machines (SVM) have been shown to outperform many related techniques. To construct the SVM classifier, it is first necessary to decide the kernel function, and different kernel functions can result in different prediction performance. However, there have been very few studies focused on examining the prediction performances of SVM based on different kernel functions. Moreover, it is unknown whether SVM classifier ensembles which have been proposed to improve the performance of single classifiers can outperform single SVM classifiers in terms of breast cancer prediction. Therefore, the aim of this paper is to fully assess the prediction performance of SVM and SVM ensembles over small and large scale breast cancer datasets. The classification accuracy, ROC, F-measure, and computational times of training SVM and SVM ensembles are compared. The experimental results show that linear kernel based SVM ensembles based on the bagging method and RBF kernel based SVM ensembles with the boosting method can be the better choices for a small scale dataset, where feature selection should be performed in the data pre-processing stage. For a large scale dataset, RBF kernel based SVM ensembles based on boosting perform better than the other classifiers.

Original languageEnglish
Article number0161501
JournalPLoS ONE
Volume12
Issue number1
DOIs
StatePublished - Jan 2017

Fingerprint

Dive into the research topics of 'SVM and SVM ensembles in breast cancer prediction'. Together they form a unique fingerprint.

Cite this