Metaheuristic-based gallstone classification using rotational forest explained with SHAP

Cholelithiasis, commonly known as Gallstone disease, occurs when hardened deposits form in the gallbladder or bile ducts. It affects millions of people worldwide and is especially common in women. While many people may not experience any symptoms, symptomatic cases can present with acute cholecystitis and other complications such as pancreatitis and even gallbladder cancer. However, this disease presents a clinical challenge due to its variable symptoms and risk of serious complications. Therefore, early prediction of gallstones is essential for timely intervention.

Thus, our study presents a novel approach for predicting gallstones. In this study, we have presented a Rotational Forest (RoF) classifier optimized using the Bald Eagle Search (BES) algorithm for gallstone prediction based on a tabular dataset. Our research has been conducted across two frameworks: using RoF alone and using RoF with the BES algorithm.

While using RoF alone, an accuracy of 78% and an AUC of 0.867 was obtained using all features. An accuracy of 75.78% and an AUC of 0.860 were obtained for RoF with the BES algorithm using only 17 features. SHapley Additive exPlanations (SHAP) and Local Interpretable Model-agnostic Explanations (LIME) analysis has distinguished CRP, Vitamin D, Obesity, HGB, and BM as the most dominant features.

Likewise, we have also compared our work with other novel works and validated the performance of our model for the prediction of gallstones.