Detection of breast cancer using machine learning and explainable artificial intelligence.

Journal: Scientific reports
Published Date:

Abstract

Breast cancer is characterized by the proliferation of abnormal breast cells that eventually turn into malignant tumors. These cancer cells can metastasize to be life-threatening and fatal. An intricate mix of environmental factors and individual genetic composition can lead to the formation of this deadly carcinoma. Improvements in the diagnosis and treatment of cancer are essential given the rising incidence of breast cancer. Over the past few decades, machine learning has helped provide accurate medical diagnosis results. Therefore, this study used diagnostic characteristics of patients and multiple machine learning classifiers to identify breast cancer. Incorporating explainable artificial intelligence techniques revealed the underlying factors for the model predictions, adding a layer of transparency and interpretability. Out of the algorithms, random forest showed the best result, an F1-score of 84%. The stacked ensemble model, which combines the strengths of different models, obtained an F1-score performance of 83%. The research emphasized the results obtained by explainers such as SHAP (SHapley Additive exPlanations), LIME (Local Interpretable Model-agnostic Explanations), ELI5 (Explain Like I'm Five), Anchor and QLattice (Quantum Lattice) to decipher the findings. Interpretable algorithms can be applied in the medical sector to assist practitioners in predicting breast cancer, reducing diagnostic errors, and improving clinical decision-making.

Authors

  • Tharunya Arravalli
    Department of Electronics and Communication Engineering, Manipal Institute of Technology, Manipal Academy of Higher Education, Manipal, Karnataka, 576104, India.
  • Krishnaraj Chadaga
    Department of Computer Science and Engineering, Manipal Institute of Technology, Manipal Academy of Higher Education, Manipal 576104, India.
  • H Muralikrishna
    Department of Electronics and Communication Engineering, Manipal Institute of Technology, Manipal Academy of Higher Education, Manipal, Karnataka, 576104, India. murali.h@manipal.edu.
  • Niranjana Sampathila
    Department of Biomedical Engineering, Manipal Institute of Technology, Manipal Academy of Higher Education, Manipal, India. Electronic address: niranjana.s@manipal.edu.
  • D Cenitta
    Manipal Institute of Technology, Manipal Academy of Higher Education, Manipal, 576104, India.
  • Rajagopala Chadaga
    Department of Mechanical and Industrial Engineering, Manipal Institute of Technology, Manipal Academy of Higher Education, Manipal, Karnataka, India.
  • K S Swathi
    Department of Health Care and Hospital Management, Prasanna School of Public Health, Manipal Academy of Higher Education, Manipal, 576104, India.