A Primary Study for Cancer Prognosis based on Classification and Regression Using Support Vector Machine
In medical domain, prognosis prediction treated as a regression problem is generally applied to predict the event duration time, such as the duration time of the recurrence of a certain disease. Recently, machine learning techniques are gaining popularity in this field because of its effectiveness and reliability. In this paper, a method based on support vector machine (SVM) to predict the exact recurrence time has been proposed. The method is compared with other four prognostic methods using Wisconsin Breast Cancer Dataset. Experimental results demonstrate that the method is more simplified to be implemented than the other four prognostic methods, and it performs much better than the medium level.
KeywordsPrognosis prediction Event time Support vector machine
The authors would like to thank professor Li’s at University of South Australia for his constructive comments. This project is supported by national natural science funding of China (project number: 11265007).
- 1.Xu R, Cai X, Wunsch DCII (2006) Gene expression data for DLBCL cancer survival prediction with a combination of machine learning technologies. In: IEEE-EMBS 2005. 27th annual international conference of the engineering in medicine and biology society Jan 2005, IEEE pp 894–897Google Scholar
- 2.Jerez-Aragonés JM, Gómez-Ruiz JA, Ramos-Jiménez G, Muñoz-Pérez J, Alba-Conejo E (2003). A combined neural network and decision trees model for prognosis of breast cancer relapse. Artif Intell Med 27(1):45–63Google Scholar
- 4.Shatkay TH, Chan HWY (2006, May). Breast cancer prognosis via Gaussian mixture regression. In: CCECE’06, Canadian conference on electrical and computer engineering, May 2006, IEEE pp 987–990Google Scholar
- 5.Bagotskaya N, Lossev I, Losseva N, Parakhin M (2005) Prediction of time to event for censored data: ridge regression with linear constraints in kernel space. In: IJCNN’05. Proceedings EEE International Joint Conference on Neural Networks 2005, IEEE, vol 2 pp 1033–1038Google Scholar
- 6.Shivaswamy PK, Chu W, Jansche M (2007, October). A support vector approach to censored targets. In: ICDM Seventh IEEE International Conference on Data Mining, Oct 2007, IEEE, pp 655–660Google Scholar
- 7.Sun BY, Zhu ZH, Li J, Linghu B (2011) Combined Feature Selection and Cancer Prognosis Using Support Vector Machine Regression. IEEE/ACM Trans Comput Biol Bioinf (TCBB), 8(6):1671–1677Google Scholar
- 8.Vapnik V (1999) The nature of statistical learning theory. SpringerGoogle Scholar
- 10.Hsu CW, Chang CC, Lin CJ (2003). A practical guide to support vector classificationGoogle Scholar
- 11.The Center for Machine Learning and Intelligent Systems at the University of California, Irvine. URL:http://archive.ics.uci.edu/ml/machine-learning-databases/breast-cancer-wisconsin/
- 12.Street WN (1994) Cancer diagnosis and prognosis via linear-programming-based machine learning. Oper Res 43(4):570–577Google Scholar
- 13.Ling Z (2002) The relationship between kernel functions based SVM and three-layer feedforward neural networks. Chin J Comput 25(7):1–5Google Scholar