Comparing the Performance of Feature Selection Methods for Predicting Gastric Cancer

Abstract

Background: Gastric cancer (GC) is a leading cause of cancer-related deaths, emphasizing the importance of timely diagnosis for effective treatment. Machine learning models have shown promise in assisting with GC diagnosis. Objectives: This study aimed at comparing the performance of various feature selection methods in identifying influential factors related to GC based on lifestyle using machine learning models. The ultimate goal was to enhance early detection and treatment of the disease. Methods: The data of patients from Shahid Ayatollah Modarres Hospital and Shohadaye Tajrish Hospital between 2013 and 2021 were utilized. Three feature selection methods (filter, wrapper, and filter-wrapper) were employed. The k-fold method validated each model. Four classifiers k Nearest Neighbor (kNN), Decision Tree (DT), Random Forest (RF), and Gradient-Boosted Decision Trees (GBDT) compared their outputs based on feature selection methods. Results: The filter-wrapper method outperformed others, achieving an area under the ROC curve and F1 score of 95.8% and 94.7%, respectively. GBDT also performed well. The wrapper and RF classifiers achieved an area under the ROC curve and F1 scores of 95.7% and 93.6%, respectively, after the filter-wrapper method. Without feature selection methods, the RF classifier had an area under the ROC curve and F1 scores of 95.6% and 91.7%, respectively, surpassing other classifiers. Conclusions: This study suggests that appropriate feature selection methods for identifying influential factors related to GC based on lifestyle can facilitate early diagnosis and treatment. The filter-wrapper method demonstrated the best performance in this regard.