An Alternating Least Square Based Algorithm for Predicting Patient Survivability
Breast cancer is the most common cancer to females worldwide. Using machine learning technology to predict breast-cancer patients’ survivability has drawn a lot of research interest. However, it still faces many issues, such as missing-value imputation. As such, the main objective of this paper is to develop a novel imputation algorithm, inspired by the recommendation system. More precisely, features with missing values are regarded as items to be evaluated for recommendation.
Consequently, a matrix factorisation algorithm (Alternating Least Square, ALS) is employed to replace missing values; accordingly, four different prediction strategies based on the ALS result are further discussed. The proposed ALS-based imputation algorithm is evaluated by using a large patient dataset from the Surveillance, Epidemiology, and End Results (SEER) program. Experimental results demonstrates a significant improvement on the survivability prediction, compared to existing methods.
KeywordsSEER dataset Survivability prediction Missing-value imputation Alternating Least Square
- 2.Liu, Y.Q., Wang, C., Zhang, L.: Neural network based models for predicting breast cancer survivability. Chin. J. Biomed. Eng. 28, 221–225 (2009)Google Scholar
- 3.Solti, D., Zhai, H.: Predicting breast cancer patient survival using machine learning. In: ACM Conference on Bioinformatics, Computational Biology and Biomedical Informatics, BCB 2013, pp. 704–705. ACM (2013)Google Scholar
- 5.Surveillance, Epidemiology, and End Results. http://www.seer.cancer.gov
- 6.McGale, P., et al.: Effect of radiotherapy after mastectomy and axillary surgery on 10-year recurrence and 20-year breast cancer mortality: meta-analysis of individual patient data for 8135 women in 22 randomised trials. Lancet (London) 383, 2127–2135 (2014). https://doi.org/10.1016/S0140-6736(14)60488-8CrossRefGoogle Scholar
- 7.Jia, Y., Sun, C., Liu, Z., Wang, W., Zhou, X.: Primary breast diffuse large B-cell lymphoma: a population-based study from 1975 to 2014. Oncotarget 9, 3956–3967 (2018)Google Scholar
- 16.Bian, Y., Li, H.: Recommendation system based on trusted relation transmission. In: 12th International Conference Intelligent Systems and Knowledge Engineering (ISKE), pp. 1–8. IEEE, November 2017. https://doi.org/10.1109/ISKE.2017.8258843