Gallstone Classification Using Random Forest Optimized by Sand Cat Swarm Optimization Algorithm with SHAP and DiCE-Based Interpretability

Citation

Sarker, Proshenjit and Tiang, Jun Jiat and Nahid, Abdullah-Al (2025) Gallstone Classification Using Random Forest Optimized by Sand Cat Swarm Optimization Algorithm with SHAP and DiCE-Based Interpretability. Sensors, 25 (17). p. 5489. ISSN 1424-8220

[img] Text
sensors-25-05489-v2.pdf - Published Version
Restricted to Repository staff only

Download (1MB)

Abstract

Gallstone disease affects approximately 10–20% of the global adult population, with early diagnosis being essential for effective treatment and management. While image-based machine learning (ML) models have shown high accuracy in gallstone detection, tabular data approaches remain less explored. In this study, we have proposed a Random Forest (RF) classifier optimized using the Sand Cat Swarm Optimization (SCSO) algorithm for gallstone prediction based on a tabular dataset. Our experiments have been conducted across four frameworks: only RF without cross-validation (CV), RF with CV, RF-SCSO without CV, and RF-SCSO with CV. Only RF without CV model has achieved 81.25%, 79.07%, 85%, and 73.91% accuracy, F-score, precision, and recall, respectively, using all 38 features, while the RF with CV has obtained a 10-fold cross-validation accuracy of 78.42% using the same feature set. With SCSO-based feature reduction, the RF-SCSO without and with CV models have delivered a comparable accuracy of 79.17% and 78.32%, respectively, using only 13 features, indicating effective dimensionality reduction. SHAP analysis has identified CRP, Vitamin D, and AAST as the most influential features, and DiCE has further illustrated the model’s behavior by highlighting corrective counterfactuals for misclassified instances. These findings demonstrate the potential of interpretable, feature-optimized ML models for gallstone diagnosis using structured clinical data.

Item Type: Article
Uncontrolled Keywords: Gallstone, machine learning, random forest classifier, Sand Cat Swarm Optimization, SHAP, DiCE
Subjects: Q Science > QA Mathematics > QA71-90 Instruments and machines
Divisions: Faculty of Artificial Intelligence & Engineering (FAIE)
Depositing User: Ms Suzilawati Abu Samah
Date Deposited: 30 Sep 2025 04:08
Last Modified: 05 Oct 2025 05:24
URII: http://shdl.mmu.edu.my/id/eprint/14565

Downloads

Downloads per month over past year

View ItemEdit (login required)