An enhanced machine learning approach with stacking ensemble learner for accurate liver cancer diagnosis using feature selection and gene expression data
Amena Mahmoud, Eiko Takaoka
Abstract
Liver cancer is a significant global health concern, necessitating accurate and timely diagnosis for effective treatment. Machine learning approaches have emerged as promising tools for improving liver cancer classification using gene expression data in recent years. This study presents an advanced machine learning approach for liver cancer diagnosis using gene expression data, combining feature selection techniques with a stacking ensemble learning model. Our method addresses the challenges of high dimensionality and complex patterns in genomic data to improve diagnostic accuracy and interpretability. We employed a feature selection process to identify the most relevant gene expressions associated with liver cancer. This approach reduced the dimensionality of the data while preserving crucial biological information. The selected features were then used to train a stacking ensemble model, which combined multiple base learners, including Multi-Layer Perceptron (MLP), Random Forest (RF) model, K-nearest neighbor (KNN) model, and Support vector machine (SVM), with a meta-learner Extreme Gradient Boosting (Xgboost) model to make final predictions. The stacking ensemble achieved an accuracy of (97%), outperforming individual machine learning algorithms and traditional diagnostic methods. Furthermore, the model demonstrated high sensitivity (96.8%) and specificity (98.1%), crucial for early detection and minimizing false positives. ➢ Propose an enhanced machine learning with stacking ensemble learner for accurate liver cancer diagnosis ➢ Use feature selection techniques to identify key genetic markers while maintaining integrity in essential biological information. ➢ Combine multi-layer perceptron, random forest, K-nearest neighbor, and Support vector machine with XGBoost meta-learning for enhanced diagnostic precision. ➢ Show exceptional reliability in terms of sensitivity and specificity in cancer detection through clinical validation. ➢ Improve diagnostic capabilities by reducing false positives and supporting early intervention strategies.