Abstract:Outcomes for cancer patients have been previously estimated by applying various machine learning techniques to large datasets such as the Surveillance, Epidemiology, and End Results (SEER) program database. In particular for lung cancer, it is not well understood which types of techniques would yield more predictive information, and which data attributes should be used in order to determine this information. In this study, a number of supervised learning techniques is applied to the SEER database to classify lung cancer patients in terms of survival, including linear regression, Decision Trees, Gradient Boosting Machines (GBM), Support Vector Machines (SVM), and a custom ensemble. Key data attributes in applying these methods include tumor grade, tumor size, gender, age, stage, and number of primaries, with the goal to enable comparison of predictive power between the various methods The prediction is treated like a continuous target, rather than a classification into categories, as a first step towards improving survival prediction. The results show that the predicted values agree with actual values for low to moderate survival times, which constitute the majority of the data. The best performing technique was the custom ensemble with a Root Mean Square Error (RMSE) value of 15.05. The most influential model within the custom ensemble was GBM, while Decision Trees may be inapplicable as it had too few discrete outputs. The results further show that among the five individual models generated, the most accurate was GBM with an RMSE value of 15.32. Although SVM underperformed with an RMSE value of 15.82, statistical analysis singles the SVM as the only model that generated a distinctive output. The results of the models are consistent with a classical Cox proportional hazards model used as a reference technique. We conclude that application of these supervised learning techniques to lung cancer data in the SEER database may be of use to estimate patient survival time with the ultimate goal to inform patient care decisions, and that the performance of these techniques with this particular dataset may be on par with that of classical methods.

Intelligent Analysis for Improving The Clinical Pathway of Lung Cancer

Evaluation of Machine Learning Algorithms for the Prognosis of Breast Cancer from the Surveillance, Epidemiology, and End Results Database

Survival analysis for lung cancer patients: A comparison of Cox regression and machine learning models

Medical decision support system for cancer treatment in precision medicine in developing countries

A comparative analysis of classical machine learning and deep learning techniques for predicting lung cancer survivability

Development of a machine learning-based multimode diagnosis system for lung cancer

Optimizing the Prognostic Model of Cervical Cancer Based on Artificial Intelligence Algorithm and Data Mining Technology

A Classifier for Improving Early Lung Cancer Diagnosis Incorporating Artificial Intelligence and Liquid Biopsy

Prediction of lung cancer patient survival via supervised machine learning classification techniques

A Convolutional Neural Network-Based Intelligent Medical System with Sensors for Assistive Diagnosis and Decision-Making in Non-Small Cell Lung Cancer

Deep learning to estimate durable clinical benefit and prognosis from patients with non-small cell lung cancer treated with PD-1/PD-L1 blockade

Improving Lung Cancer Diagnosis and Survival Prediction with Deep Learning and CT Imaging

Implementing Machine Learning to Predict Survival Outcomes in Patients with Resected Pulmonary Large Cell Neuroendocrine Carcinoma

Multimodal Learning for Non-small Cell Lung Cancer Prognosis

Predicting lung cancer survival prognosis based on the conditional survival bayesian network

Artificial intelligence predictive system of individual survival rate for lung adenocarcinoma

Efficient lung cancer detection using computational intelligence and ensemble learning

Towards an Artificial Intelligence Framework for Early Diagnosis and Prediction of Lung Cancer

A Staging Auxiliary Diagnosis Model for Nonsmall Cell Lung Cancer Based on the Intelligent Medical System

A Machine Learning Method for Improving Liver Cancer Staging.

Two-stage Cox-nnet: Biologically Interpretable Neural-Network Model for Prognosis Prediction and Its Application in Liver Cancer Survival Using Histopathology and Transcriptomic Data