Interpretable machine learning approach for early cancer detection
Babajide Olakunle Afeni, Peter Adetola Adetunji, Peter Adetola Adetunji, Ibrahim Olakunle Yakub, Philip Adetayo Adetunji, Philip Adetayo Adetunji · International Journal of Science and Research Archive · 2024
Early cancer detection significantly improves treatment outcomes and patient survival rates. This study explores the efficacy of various machine learning models such as Logistic Regression, Support Vector Machine (SVM), Random Forest, XGBoost, and Neural Network in predicting early-stage cancer. Employing the Local Interpretable Model-agnostic Explanations (LIME) approach, we ensure model transparency and interpretability, which are essential for clinical application. The models were evaluated on a dataset with key features including cancer history, gender, smoking status, age, BMI, genetic risk, alcohol intake, and physical activity. Among the models, Random Forest and XGBoost demonstrated superior performance, achieving the highest balanced accuracy and AUC scores. LIME visualizations revealed that cancer history and gender were the most influential features across all models, with additional contributions from smoking status, age, and BMI. The study highlights the potential of tree-based models for accurate and interpretable cancer detection, providing clinicians with actionable insights. Our findings advocate for the integration of these models into clinical practice, enabling early intervention and personalized treatment strategies. Further research is recommended to validate these models in larger and more diverse populations and to explore the inclusion of additional medical data to enhance predictive accuracy.