Machine Learning System for the Effective Diagnosis and Survival Prediction of Breast Cancer Patients
DOI:
https://doi.org/10.3991/ijoe.v20i02.42883Keywords:
breast cancer, diagnosis, machine learning, treatment, random forest, naive bayesAbstract
Breast cancer is one of the most significant global health challenges. Effective diagnosis and prognosis prediction are crucial for improving patient outcomes in the case of this disease. As machine learning (ML) has significantly improved prediction models in many disciplines, the goal of this study is to develop a ML system for medical specialists that can accurately predict tumor diagnosis and patient survival for breast cancer patients. For the training of diagnosis and survival prediction, five algorithmic models—decision tree (DT), random forest (RF), naive bayes (NB), support vector machines (SVMs), and gradient boosting—were trained with 569 records from the Breast Cancer Wisconsin dataset and 1,980 records from the Breast Cancer Gene Expression Profiles dataset. The results showed that the NB model exhibited better performance for tumor diagnosis, achieving an accuracy of 95.0%, while RF presented the best results for patient survival, with an accuracy of 76.0%. A survey of medical experts’ experience with the resulting system showed high scores in reliability, performance, satisfaction, usability, and efficiency, confirming that ML systems have the potential to improve breast cancer patient outcomes.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2023 Arturo Gago, Jean Aguirre, Lenis Wong
This work is licensed under a Creative Commons Attribution 4.0 International License.
The submitting author warrants that the submission is original and that she/he is the author of the submission together with the named co-authors; to the extend the submission incorporates text passages, figures, data or other material from the work of others, the submitting author has obtained any necessary permission.
Articles in this journal are published under the Creative Commons Attribution Licence (CC-BY What does this mean?). This is to get more legal certainty about what readers can do with published articles, and thus a wider dissemination and archiving, which in turn makes publishing with this journal more valuable for you, the authors.
By submitting an article the author grants to this journal the non-exclusive right to publish it. The author retains the copyright and the publishing rights for his article without any restrictions.