Lung cancer classification is one of the biggest challenges in modern medicine, and it is one of the deadliest types of cancer in the world. Building accurate classification models that can accurately predict different lung cancer types can be particularly challenging with a limited amount of data, irrelevant features, and imbalanced distribution in the datasets. To address this challenge, this study utilizes a two-stage framework where the data is balanced using the synthetic minority oversampling technique (SMOTE), and the second stage uses the preprocessed dataset on three machine learning (ML) techniques comprising of Decision Tree (DT), Extreme Gradient Boosting (XGB), and Logistic Regression (LR) with a soft voting technique. The SMOTE technique plays a significant role in improving the performance of all the models by balancing the minority class and mitigating the bias introduced by the class imbalance. Across all the experiments, the proposed model achieves an accuracy of 95.16%, precision of 95.04, and AUC score of 0.98, outperforming the results of all the individual models and all benchmarks taken for this study.

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

Enhanced Lung Cancer Classification Using SMOTE and Soft Voting Ensemble of Decision Tree, XGBoost, and Logistic Regression

  • Sasanka Sekhar Dalai,
  • Bharat Jyoti Ranjan Sahu,
  • Ibanga Kpereobong Friday,
  • Subrat Kumar Nayak,
  • Nrusingha Tripathy

摘要

Lung cancer classification is one of the biggest challenges in modern medicine, and it is one of the deadliest types of cancer in the world. Building accurate classification models that can accurately predict different lung cancer types can be particularly challenging with a limited amount of data, irrelevant features, and imbalanced distribution in the datasets. To address this challenge, this study utilizes a two-stage framework where the data is balanced using the synthetic minority oversampling technique (SMOTE), and the second stage uses the preprocessed dataset on three machine learning (ML) techniques comprising of Decision Tree (DT), Extreme Gradient Boosting (XGB), and Logistic Regression (LR) with a soft voting technique. The SMOTE technique plays a significant role in improving the performance of all the models by balancing the minority class and mitigating the bias introduced by the class imbalance. Across all the experiments, the proposed model achieves an accuracy of 95.16%, precision of 95.04, and AUC score of 0.98, outperforming the results of all the individual models and all benchmarks taken for this study.