Lung cancer is a common cancer that reveals notable death statistics worldwide. Due to the uncontrollable spreading rate of abnormal cells in the lung, lung cancer has become the leading cause of cancer death, with an approximate estimation of 1.8 million. In this context, early diagnosis has been highlighted to increase the survival rate. To address this challenge, this study evaluated two machine learning models, namely SVM and XGBoost for predicting risk levels. The models were trained and evaluated using a lung cancer dataset which consists of 1000 samples with 26 clinical parameters. The dataset underwent pre-processing steps and exploratory data analysis to efficiently handle missing values, outliers, and feature selection. After pre-processing, the dataset was split into training and test sets. Then, the predictive models were trained by using the training set, and the performances were assessed through the test set. Hyperparameter tuning was applied to optimize the model performance and avoid overfitting. Both SVM and XGBoost give promising results for all performance parameters assessed including accuracy, precision, recall, F1-score, and ROC-AUC. These findings demonstrate the potential of machine learning in early lung cancer detection.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Multiclass Classification of Lung Cancer with SVM and XGBoost with 20+ Features

  • Zahereel Ishwar Abdul Khalib,
  • Nur Hafizah Ghazali,
  • S. M. Wong

摘要

Lung cancer is a common cancer that reveals notable death statistics worldwide. Due to the uncontrollable spreading rate of abnormal cells in the lung, lung cancer has become the leading cause of cancer death, with an approximate estimation of 1.8 million. In this context, early diagnosis has been highlighted to increase the survival rate. To address this challenge, this study evaluated two machine learning models, namely SVM and XGBoost for predicting risk levels. The models were trained and evaluated using a lung cancer dataset which consists of 1000 samples with 26 clinical parameters. The dataset underwent pre-processing steps and exploratory data analysis to efficiently handle missing values, outliers, and feature selection. After pre-processing, the dataset was split into training and test sets. Then, the predictive models were trained by using the training set, and the performances were assessed through the test set. Hyperparameter tuning was applied to optimize the model performance and avoid overfitting. Both SVM and XGBoost give promising results for all performance parameters assessed including accuracy, precision, recall, F1-score, and ROC-AUC. These findings demonstrate the potential of machine learning in early lung cancer detection.