StackLiverNet: A Novel Stacked Ensemble Model for Accurate and Interpretable Liver Disease Detection

TL;DR

StackLiverNet is an interpretable, high-accuracy ensemble model for liver disease detection that combines advanced preprocessing, feature selection, and explainability techniques to improve clinical applicability.

Contribution

This study introduces StackLiverNet, a novel stacked ensemble model with optimized classifiers and interpretability methods, addressing previous issues of accuracy and transparency in liver disease diagnosis.

Findings

Achieved 99.89% testing accuracy

Demonstrated high interpretability with LIME and SHAP

Efficient training and inference suitable for clinical use

Abstract

Liver diseases are a serious health concern in the world, which requires precise and timely diagnosis to enhance the survival chances of patients. The current literature implemented numerous machine learning and deep learning models to classify liver diseases, but most of them had some issues like high misclassification error, poor interpretability, prohibitive computational expense, and lack of good preprocessing strategies. In order to address these drawbacks, we introduced StackLiverNet in this study; an interpretable stacked ensemble model tailored to the liver disease detection task. The framework uses advanced data preprocessing and feature selection technique to increase model robustness and predictive ability. Random undersampling is performed to deal with class imbalance and make the training balanced. StackLiverNet is an ensemble of several hyperparameter-optimized base classifiers, whose complementary advantages are used through a LightGBM meta-model. The provided model demonstrates excellent performance, with the testing accuracy of 99.89%, Cohen Kappa of 0.9974, and AUC of 0.9993, having only 5 misclassifications, and efficient training and inference speeds that are amenable to clinical practice (training time 4.2783 seconds, inference time 0.1106 seconds). Besides, Local Interpretable Model-Agnostic Explanations (LIME) are applied to generate transparent explanations of individual predictions, revealing high concentrations of Alkaline Phosphatase and moderate SGOT as important observations of liver disease. Also, SHAP was used to rank features by their global contribution to predictions, while the Morris method confirmed the most influential features through sensitivity analysis.