A Novel Aggregated Multiple Imputation Approach for Enhanced Survival Prediction and Classification on Breast Cancer and Lung Cancer Data

被引:0
作者
Deepa, P. [1 ]
Gunavathi, C. [2 ]
机构
[1] Vellore Inst Technol, Sch Comp Sci Engn & Informat Syst, Vellore 632014, India
[2] Vellore Inst Technol, Sch Comp Sci & Engn, Vellore 632014, India
来源
IEEE ACCESS | 2024年 / 12卷
关键词
Imputation; Cancer; Tumors; Data models; Lung cancer; Breast cancer; Predictive models; Accuracy; Prognostics and health management; Lymph nodes; Classification; missing data; multiple imputation; survival analysis;
D O I
10.1109/ACCESS.2024.3516837
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Survival analysis is the method of finding the time of occurrence of an event. Survival analysis is used as a prognostic tool in healthcare especially in diagnosing cancer. Any healthcare data suffers with the missing data problem, survival data is not an exception. Data imputation is the way of handling missing data. In this paper we propose an Aggregated Multiple Imputation technique (AMI) which imputes the missing data with three base imputation techniques like mean imputation, K-nearest neighbour (kNN) imputation and iterative imputation. These techniques were combined by weighted average approach. AMI makes use of the advantages of each method to create imputed values that are more accurate and dependable by using a weighted average approach. The proposed method generates multiple datasets while applying the base imputation techniques. The imputed datasets are then combined using a weighted average, resulting in generation of reliable data by reducing the bias and improving the precision of the imputed value. Breast cancer and lung cancer data from the Surveillance, Epidemiology, and End Results (SEER) program is used for validation of the proposed technique. The imputed data improves the performance of various classifiers and survival prediction models in predicting the overall survival of the cancer patients. The results show that the data imputed using AMI approach improves the performance of the various classifiers and the survival prediction models, compared to the data imputed using the single imputation method. The highest accuracy achieved using the dataset is 91% and the least accuracy is 76% for breast cancer data. The highest accuracy achieved using the dataset is 87% and the least accuracy is 72% for lung cancer data.
引用
收藏
页码:189102 / 189121
页数:20
相关论文
共 38 条
  • [31] Rincy T.N., Gupta R., Ensemble learning techniques and its efficiency in machine learning: A survey, Proc. 2nd Int. Conf. Data, Eng. Appl. (IDEA), pp. 1-6, (2020)
  • [32] Lopez-Bernal D., Balderas D., Ponce P., Molina A., Education 4.0: Teaching the basics of KNN, LDA and simple perceptron algorithms for binary classification problems, Future Internet, 13, 8, (2021)
  • [33] Xanthopoulos P., Linear discriminant analysis, Robust Data Mining, 3, pp. 27-33, (2013)
  • [34] Steinbach M., Tan P.N., kNN: K-nearest neighbors, The Top Ten Algorithms in Data Mining, pp. 151-162, (2009)
  • [35] Harrell F.E., Evaluating the yield of medical tests, JAMA, J. Amer. Med. Assoc., 247, 18, (1982)
  • [36] Li M., Gao Q., Yu T., Kappa statistic considerations in evaluating inter-rater reliability between two raters: Which, when and context matters, BMC Cancer, 23, 1, (2023)
  • [37] Chicco D., Totsch N., Jurman G., The Matthews correlation coefficient (MCC) is more reliable than balanced accuracy, bookmaker informedness, and markedness in two-class confusion matrix evaluation, BioData Mining, 14, 1, pp. 1-22, (2021)
  • [38] Chicco D., Jurman G., The Matthews correlation coefficient (MCC) should replace the ROC AUC as the standard metric for assessing binary classification, BioData Mining, 16, 1, pp. 1-23, (2023)