Mortality Prediction of Various Cancer Patients via Relevant Feature Analysis and Machine Learning

Caner Bozkurt, Tunç Aşuroğlu*

*Corresponding author for this work

Research output: Contribution to journalArticleScientificpeer-review

5 Citations (Scopus)

Abstract

Breast, lung, prostate, and stomach cancers are the most frequent cancer types globally. Early-stage detection and diagnosis of these cancers pose a challenge in the literature. When dealing with cancer patients, physicians must select among various treatment methods that have a risk factor. Since the risks of treatment may outweigh the benefits, treatment schedule is critical in clinical decision making. Manually deciding which medications and treatments are going to be successful takes a lot of expertise and can be hard. In this paper, we offer a computational solution to predict the mortality of various types of cancer patients. The solution is based on the analysis of diagnosis, medication, and treatment parameters that can be easily acquired from electronic healthcare systems. A classification-based approach introduced to predict the mortality outcome of cancer patients. Several classifiers evaluated on the Medical Information Mart in Intensive Care IV (MIMIC-IV) dataset. Diagnosis, medication, and treatment features extracted for breast, lung, prostate, and stomach cancer patients and relevant feature selection done with Logistic Regression. Best F1 scores were 0.74 for breast, 0.73 for lung, 0.82 for prostate, and 0.79 for stomach cancer. Best AUROC scores were 0.94 for breast, 0.91 for lung, 0.96 for prostate, and 0.88 for stomach cancer. In addition, using relevant features, results were very similar to the baseline for each cancer type. Using less features and a robust machine-learning model, the proposed approach can be easily implemented in hospitals when there are limited data and resources available.
Original languageEnglish
Article number264
Number of pages14
JournalSN Computer Science
Volume4
Issue number3
DOIs
Publication statusPublished - May 2023
MoE publication typeA1 Journal article-refereed

Fingerprint

Dive into the research topics of 'Mortality Prediction of Various Cancer Patients via Relevant Feature Analysis and Machine Learning'. Together they form a unique fingerprint.

Cite this