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 条
  • [21] Baneshi M.R., Talei A.R., Impact of Imputation of Missing Data on Estimation of Survival Rates: an Example in Breast Cancer, (2010)
  • [22] Baruah B., Dutta M.P., Bhattacharyya D.K., An effective ensemble method for missing data imputation, Int. J. Inf. Comput. Secur., 1, 1, (2022)
  • [23] Batra S., Khurana R., Khan M.Z., Boulila W., Koubaa A., Srivastava P., A pragmatic ensemble strategy for missing values imputation in health records, Entropy, 24, 4, (2022)
  • [24] Aleryani A., Wang W., de la Iglesia B., Multiple imputation ensembles (MIE) for dealing with missing data, Social Netw. Comput. Sci., 1, 3, pp. 1-20, (2020)
  • [25] Grzymala-Busse J.W., Handling missing attribute values, Data Mining and Knowledge Discovery Handbook, pp. 33-51, (2010)
  • [26] Batista G.E.A.P.A., Monard M.C., An analysis of four missing data treatment methods for supervised learning, Appl. Artif. Intell., 17, 5-6, pp. 519-533, (2003)
  • [27] Van Buuren S., Groothuis-Oudshoorn K., mice: Multivariate imputation by chained equations in R, J. Stat. Software, 45, pp. 1-67, (2011)
  • [28] Cox D.R., Regression models and life-tables, J. Royal Stat. Soc., Ser. B Methodol., 34, 2, pp. 187-202, (1972)
  • [29] Ishwaran H., Kogalur U.B., Chen X., Minn A.J., Random survival forests for high-dimensional data, ASA Data Sci. J., 4, 1, pp. 115-132, (2011)
  • [30] Ke G., Lightgbm: A highly efficient gradient boosting decision tree, Proc. Adv. Neural Inf. Process. Syst., pp. 1-11, (2017)