Open Access

ARTICLE

Prognosis Analysis of Lung Cancer Patients

Yicheng Xie^1,*, Jinyue Xia²

1 School of Computer Science, Nanjing University of Information Science and Technology, Nanjing, 210044, China
2 International Business Machines Corporation (IBM), NY, 10504, USA

* Corresponding Author: Yicheng Xie. Email:

Journal of Intelligent Medicine and Healthcare 2022, 1(1), 43-54. https://doi.org/10.32604/jimh.2022.032405

Received 25 April 2022; Accepted 27 May 2022; Issue published 14 June 2022

Abstract

Lung cancer is now the most common type of cancer worldwide, with high levels of morbidity and mortality. The cost of treatment and emotional stress put a high burden on families and society. This paper aims to collect relevant information and provide predictive analysis for the prognosis of patients with lung cancer. Using the public data of SEER database and the method of machine learning, a model is constructed to predict the five-year survival of patients with lung cancer. The re-coding method is used for data processing, the eigenvalues are re-coded to adapt to the construction of the model, and the data are balanced by a variety of sampling methods to improve the applicability of the model. The construction method of the model is based on logistic regression, fully connected neural network, random forest and XGBOOST, evaluate the performance and select the optimal model. The results show that among the four constructed models, XGBOOST is selected as the optimal model with faster training speed, higher accuracy and the highest AUC value, it has advantages in memory occupation and time-consuming. The tumor stage and whether surgery or not and treatment difficulty have certain decisive factors for the prognosis of patients.

---

Keywords

---