Application of Machine Learning Models for Survival Prognosis in Breast Cancer Studies

The application of machine learning models for prediction and prognosis of disease development has become an irrevocable part of cancer studies aimed at improving the subsequent therapy and management of patients. The application of machine learning models for accurate prediction of survival time in breast cancer on the basis of clinical data is the main objective of the presented study. The paper discusses an approach to the problem in which the main factor used to predict survival time is the originally developed tumor-integrated clinical feature, which combines tumor stage, tumor size, and age at diagnosis. Two datasets from corresponding breast cancer studies are united by applying a data integration approach based on horizontal and vertical integration by using proper document-oriented and graph databases which show good performance and no data losses. Aside from data normalization and classification, the applied machine learning methods provide promising results in terms of accuracy of survival time prediction. The analysis of our experiments shows an advantage of the linear Support Vector Regression, Lasso regression, Kernel Ridge regression, K-neighborhood regression, and Decision Tree regression—these models achieve most accurate survival prognosis results. The cross-validation for accuracy demonstrates best performance of the same models on the studied breast cancer data. As a support for the proposed approach, a Python-based workflow has been developed and the plans for its further improvement are finally discussed in the paper.

[1]  Sunil R. Gupta,et al.  Machine-learning prediction of cancer survival: a retrospective study using electronic administrative records and a cancer registry , 2014, BMJ Open.

[2]  David S. Wishart,et al.  Applications of Machine Learning in Cancer Prediction and Prognosis , 2006, Cancer informatics.

[3]  Huseyin Seker,et al.  Assessment of nodal involvement and survival analysis in breast cancer patients using image cytometric data: statistical, neural network and fuzzy approaches. , 2002, Anticancer research.

[4]  Richard Hull,et al.  Managing semantic heterogeneity in databases: a theoretical prospective , 1997, PODS.

[5]  Geoffrey J. McLachlan,et al.  Analyzing Microarray Gene Expression Data , 2004 .

[6]  Mehmet Fatih Akay,et al.  Support vector machines combined with feature selection for breast cancer diagnosis , 2009, Expert Syst. Appl..

[7]  P. Butow,et al.  Communicating prognosis in cancer care: a systematic review of the literature. , 2005, Annals of oncology : official journal of the European Society for Medical Oncology.

[8]  Lana X. Garmire,et al.  Deep Learning based multi-omics integration robustly predicts survival in liver cancer , 2017, bioRxiv.

[9]  Yi Guo,et al.  Data integration through ontology-based data access to support integrative data analysis: A case study of cancer survival , 2017, 2017 IEEE International Conference on Bioinformatics and Biomedicine (BIBM).

[10]  Paulo J. G. Lisboa,et al.  A Bayesian neural network approach for modelling censored data with an application to prognosis after surgery for breast cancer , 2003, Artif. Intell. Medicine.

[11]  Miriam Seoane Santos,et al.  Predicting Breast Cancer Recurrence Using Machine Learning Techniques , 2016, ACM Comput. Surv..

[12]  Dursun Delen,et al.  Predicting breast cancer survivability: a comparison of three data mining methods , 2005, Artif. Intell. Medicine.

[13]  Leroy Hood,et al.  Systems biology, proteomics, and the future of health care: toward predictive, preventative, and personalized medicine. , 2004, Journal of proteome research.

[14]  Sieu Phan,et al.  An ensemble machine learning approach to predict survival in breast cancer , 2008, Int. J. Comput. Biol. Drug Des..

[15]  Dimitar Vassilev,et al.  Machine Learning Techniques for Survival Time Prediction in Breast Cancer , 2018, AIMSA.

[16]  Nikola Kasabov,et al.  Prediction of clinical behaviour and treatment for cancers. , 2003, Applied bioinformatics.

[17]  Gaël Varoquaux,et al.  Scikit-learn: Machine Learning in Python , 2011, J. Mach. Learn. Res..

[18]  Alon Y. Halevy,et al.  Answering queries using views: A survey , 2001, The VLDB Journal.

[19]  Ting Chen,et al.  Integrative Data Analysis of Multi-Platform Cancer Data with a Multimodal Deep Learning Approach , 2015, IEEE/ACM Transactions on Computational Biology and Bioinformatics.

[20]  Niklas Lindqvist,et al.  Evaluation of Feature Selection Methods for Machine Learning Classification of Breast Cancer , 2018 .

[21]  Jeffrey D. Ullman,et al.  Information integration using logical views , 1997, Theor. Comput. Sci..

[22]  Lars Juhl Jensen,et al.  Are graph databases ready for bioinformatics? , 2013, Bioinform..

[23]  Adel Aloraini Different Machine Learning Algorithms for Breast Cancer Diagnosis , 2012 .

[24]  Dimitrios I. Fotiadis,et al.  Machine learning applications in cancer prognosis and prediction , 2014, Computational and structural biotechnology journal.

[25]  Ameya Nayak Type of NOSQL Databases and its Comparison with Relational Databases , 2013 .

[26]  Cheng Wang,et al.  Decision Tree Based Predictive Models for Breast Cancer Survivability on Imbalanced Data , 2009, 2009 3rd International Conference on Bioinformatics and Biomedical Engineering.

[27]  A. Jemal,et al.  Breast cancer statistics, 2017, racial disparity in mortality by state , 2017, CA: a cancer journal for clinicians.