2023
doi: 10.12720/jait.14.6.1410-1424
Enhancing Prediction Accuracy in Gastric Cancer Using High-Confidence Machine Learning Models for Class Imbalance
Danish Jamil1,2,*
,
Sellappan Palaniappan1 ,
Muhammad Naseem2 ,
Asiah Lokman1
- 1Department of Information Technology, Malaysia University of Science and Technology, Kuala Lumpur, Malaysia
- 2Department of Software Engineering, Sir Syed University of Engineering and Technology, Karachi, Pakistan
- Manuscript receivedMay 23, 2023
- revisedJune 25, 2023
- acceptedJuly 7, 2023
- publishedDecember 19, 2023
Abstract
Gastric Cancer (GC) diagnosis and prognosis present significant challenges in the clinical industry. To address the issue of low prediction accuracy resulting from imbalanced positive and negative GC cases, this study proposes a medical Decision Support System (DSS) based on supervised Machine Learning (ML) methods. Four ML models, including Naïve Bayes (NB), Logistic Regression (LR), and Multilayer Perceptron (MLP), were employed in this study. The impact of data imbalance on GC prediction was assessed through two procedures. Among the ML models, the MLP model demonstrated the best performance in weighted GC prediction, achieving a sensitivity of 0.930 and a Positive Predictive Value (PPV) of 0.932 for balanced predictions, and a sensitivity of 0.918 and a PPV of 0.908 for unbalanced predictions. The NB model showed promise in handling the data imbalance issue, achieving a sensitivity of 0.722 and a PPV of 0.420 on the unbalanced dataset. Additionally, a DSS was developed specifically for the NB and LR models to improve prediction accuracy. The proposed method significantly improved the sensitivity of optimistic GC case prediction, with the Naïve Bayes model achieving a sensitivity of 0.936 and the Logistic Regression model achieving a sensitivity of 0.8306. These improvements enhance the reliability and efficiency of GC diagnostics, offering valuable decision support in healthcare. This research provides insights into addressing class imbalance in GC likelihood prediction and has potential implications for clinical practice.
Keywords
- class imbalance
- gastric cancer
- decision support system
- machine learning
- prediction accuracy
- naive bayes
- logistic regression
- medical diagnostics
- positive predictive value
How to Cite
Danish Jamil, Sellappan Palaniappan, Muhammad Naseem, and Asiah Lokman, "Enhancing Prediction Accuracy in Gastric Cancer Using High-Confidence Machine Learning Models for Class Imbalance," Journal of Advances in Information Technology, vol. 14, no. 6, pp. 1410-1424, 2023. https://doi.org/10.12720/jait.14.6.1410-1424
Copyright & License
Copyright © 2023 by the authors. This is an open access article distributed under the Creative Commons Attribution License which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited (CC BY 4.0).
