Machine Learning System for the Effective Diagnosis and Survival Prediction of Breast Cancer Patients

被引:3
|
作者
Gago, Arturo [1 ]
Aguirre, Jean Marko [1 ]
Wong, Lenis [1 ]
机构
[1] Univ Peruana Ciencias Aplicadas, Program Informat Syst Engn, Lima, Peru
关键词
breast cancer; diagnosis; treatment; machine learning (ML); random forest (RF); naive bayes (NB); CLASSIFICATION;
D O I
10.3991/ijoe.v20i02.42883
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
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.
引用
收藏
页码:95 / 113
页数:19
相关论文
共 50 条
  • [1] Machine learning models in breast cancer survival prediction
    Montazeri, Mitra
    Montazeri, Mohadeseh
    Montazeri, Mahdieh
    Beigzadeh, Amin
    TECHNOLOGY AND HEALTH CARE, 2016, 24 (01) : 31 - 42
  • [2] Prediction of survival and metastasis in breast cancer patients using machine learning classifiers
    Tapak, Leili
    Shirmohammadi-Khorram, Nasrin
    Amini, Payam
    Alafchi, Behnaz
    Hamidi, Omid
    Poorolajal, Jalal
    CLINICAL EPIDEMIOLOGY AND GLOBAL HEALTH, 2019, 7 (03): : 293 - 299
  • [3] A comparison of machine learning techniques for survival prediction in breast cancer
    Leonardo Vanneschi
    Antonella Farinaccio
    Giancarlo Mauri
    Marco Antoniotti
    Paolo Provero
    Mario Giacobini
    BioData Mining, 4
  • [4] Machine Learning Techniques for Survival Time Prediction in Breast Cancer
    Mihaylov, Iliyan
    Nisheva, Maria
    Vassilev, Dimitar
    ARTIFICIAL INTELLIGENCE: METHODOLOGY, SYSTEMS, AND APPLICATIONS, AIMSA 2018, 2018, 11089 : 186 - 194
  • [5] A comparison of machine learning techniques for survival prediction in breast cancer
    Vanneschi, Leonardo
    Farinaccio, Antonella
    Mauri, Giancarlo
    Antoniotti, Mauro
    Provero, Paolo
    Giacobini, Mario
    BIODATA MINING, 2011, 4
  • [6] Recommender system: prediction/diagnosis of breast cancer using hybrid machine learning algorithm
    Shalli Rani
    Manpreet Kaur
    Munish Kumar
    Multimedia Tools and Applications, 2022, 81 : 9939 - 9948
  • [7] Recommender system: prediction/diagnosis of breast cancer using hybrid machine learning algorithm
    Rani, Shalli
    Kaur, Manpreet
    Kumar, Munish
    MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 81 (07) : 9939 - 9948
  • [8] Prediction Model of Breast Cancer Survival Months: A Machine Learning Approach
    Naser, Mohammad Y. M.
    Chambers, Destini
    Bhattacharya, Sylvia
    SOUTHEASTCON 2023, 2023, : 851 - 855
  • [9] RETRACTED: Machine Learning Algorithms for Prediction of Survival Curves in Breast Cancer Patients (Retracted Article)
    Maabreh, Roqia Saleem Awad
    Alazzam, Malik Bader
    AlGhamdi, Ahmed S.
    APPLIED BIONICS AND BIOMECHANICS, 2021, 2021
  • [10] Survival prediction in second primary breast cancer patients with machine learning: An analysis of SEER database
    Wu, Yafei
    Zhang, Yaheng
    Duan, Siyu
    Gu, Chenming
    Wei, Chongtao
    Fang, Ya
    COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2024, 254