Determining the Major Contributing Features to Predict Breast Cancer Imposing ML Algorithms with LIME and SHAP
Prakash Paudel, Ruby Saud, Satish Kumar Karna, Mohan Bhandari · 2023
Significant advancements have been achieved recently in the use of artificial intelligence (AI) to solve healthcare-related problems. A potential method for identifying and treating many illnesses is machine learning (ML). The “Breast Cancer Dataset” was used in this work to perform classification tasks using ML techniques including Support Vector Machine (SVM), Random Forest (RF) and MultiLayer Perceptron (MLP) for identifying the forms of breast cancer, either Benign or Malignant. The SVM, RF, and MLP achieved impressive PR scores exceeding 0.98. However, MLP exhibited superior performance among these algorithms, achieving an exceptional AUC value of 99.71. Conversely, SVM slightly outperformed the other algorithms in terms of training accuracy, with a value of 97.90%. This work used shapley additive explanations (SHAP) and local interpretable model-agnostic explanations (LIME) to shed light on how these categorization algorithms made judgments. These interpretability methodologies showed that the majority of characteristics consistently predicted the presence of malignant or benign cancer types. This study demonstrates how ML algorithms' decision-making procedures may be understood, possibly promoting the use of ML in clinical settings for medical diagnosis.