Prediction of Post-Operative Survival Expectancy in Thoracic Lung Cancer Surgery Using Extreme Learning Machine and SMOTE

Authors

  • Ajwa Helisa Lambung Mangkurat University
  • Triando Hamonangan Saragih Lambung Mangkurat University
  • Irwan Budiman Lambung Mangkurat University
  • Fatma Indriani Lambung Mangkurat University
  • Dwi Kartini Lambung Mangkurat University

DOI:

https://doi.org/10.26555/jiteki.v9i2.25973

Keywords:

Classification, Extreme learning machine, Smote, Imbalance data, Lung cancer, Thoracic surgery

Abstract

Lung cancer is the most common cause of cancer death globally. Thoracic surgery is a common treatment for patients with lung cancer. However, there are many risks and postoperative complications leading to death. In this study, we will predict life expectancy for lung cancer patients one year after thoracic surgery The data used is secondary data for lung cancer patients in 2007-2011. There are 470 data consisting of 70 death class data and 400 survival class data for one year after surgery. The algorithm used is Extreme learning machine (ELM) for classification, which tends to be fast in the learning process and has good generalization performance. Synthetic Minority Over-sampling (SMOTE) is used to solve the problem of imbalanced data. The proposed solution combines the benefits of using SMOTE for imbalanced data along with ELM. The results show ELM and SMOTE outperform other algorithms such as Naïve Bayes, Decision stump, J48, and Random Forest. The best results on ELM were obtained at 50 neurons with 89.1% accuracy, F-Measure 0.86, and ROC 0.794. In the combination of ELM and SMOTE, the accuracy is 85.22%, F-measure 0.864, and ROC 0.855 on neuron 45 using a data division proportion of 90:10. The test results show that the proposed method can significantly improve the performance of the ELM algorithm in overcoming class imbalance. The contribution of this study is to build a machine learning model with good performance so that it can be a support system for medical informatics experts and doctors in early detection to predict the life expectancy of lung cancer patients.

Downloads

Published

2023-04-18

How to Cite

[1]
A. Helisa, T. H. Saragih, I. Budiman, F. Indriani, and D. Kartini, “Prediction of Post-Operative Survival Expectancy in Thoracic Lung Cancer Surgery Using Extreme Learning Machine and SMOTE”, J. Ilm. Tek. Elektro Komput. Dan Inform, vol. 9, no. 2, pp. 239–249, Apr. 2023.

Issue

Section

Articles

Similar Articles

1 2 3 > >> 

You may also start an advanced similarity search for this article.