Prognosis cancer prediction model using deep belief network approach

Cancer is one of main non-communicable disease. Analysis of cancer prognosis is necessary to determine the proper treatment for each patient. However, cancer data analysis is challenging because multiple risk factors may influence the prognosis of cancer, including genes and clinical condition of patients. This study aims to develop prediction model for cancer prognosis using clinical and gene expression (microarray) data. In this research, Principal Component Analysis (PCA) is applied to microarray data to reduce its dimension, then two Deep Belief Network (DBN) models for both clinical and microarray data are trained separately. Probabilities obtained from Clinical DBN model and Microarray DBN model are integrated using softmax nodes on Bayesian Network structure. Based on various experiments, the best DBN-BN integration model obtains prediction accuracy 73.3535% for overall survival prediction and 71.3434% for disease-free survival prediction.

[1]  Ito Wasito,et al.  Nearest neighbours in least-squares data imputation algorithms with different missing patterns , 2006, Comput. Stat. Data Anal..

[2]  Aidong Zhang,et al.  Identifying informative risk factors and predicting bone disease progression via deep belief networks. , 2014, Methods.

[3]  Christian Gieger,et al.  Harmonising and linking biomedical and clinical data across disparate data archives to enable integrative cross-biobank research , 2015, European Journal of Human Genetics.

[4]  Bart De Moor,et al.  Predicting the prognosis of breast cancer by integrating clinical and microarray data with Bayesian networks , 2006, ISMB.

[5]  Mark Roantree,et al.  A Configurable Deep Network for high-dimensional clinical trial data , 2015, 2015 International Joint Conference on Neural Networks (IJCNN).

[6]  Nedialko S. Nedialkov,et al.  Probabilistic Graphical Models and Deep Belief Networks for Prognosis of Breast Cancer , 2015, 2015 IEEE 14th International Conference on Machine Learning and Applications (ICMLA).

[7]  Mohammad Mehdi Homayounpour,et al.  A brief survey on deep belief networks and introducing a new object oriented MATLAB toolbox (DeeBNet) , 2014, ArXiv.

[8]  Pooja Gupta,et al.  Using deep learning to enhance head and neck cancer diagnosis and classification , 2018, 2018 IEEE International Conference on System, Computation, Automation and Networking (ICSCA).

[9]  Yee Whye Teh,et al.  A Fast Learning Algorithm for Deep Belief Nets , 2006, Neural Computation.

[10]  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.

[11]  Kevin Murphy,et al.  Bayes net toolbox for Matlab , 1999 .

[12]  Ayman M. Eldeib,et al.  Breast cancer classification using deep belief networks , 2016, Expert Syst. Appl..