Computer-aided detection of breast cancer on the Wisconsin dataset: An artificial neural networks approach

被引:54
作者
Alshayeji, Mohammad H. [1 ]
Ellethy, Hanem [2 ]
Abed, Saed [1 ]
Gupta, Renu [3 ]
机构
[1] Kuwait Univ, Coll Engn & Petr, Comp Engn Dept, POB 5969, Safat 13060, Kuwait
[2] Univ Queensland, Queensland Brain Inst, Brisbane, Qld, Australia
[3] Kuwait Univ, Fac Med, Dept Radiol, Kuwait, Kuwait
关键词
Artificial neural networks (ANN); Breast cancer classification; Wisconsin breast cancer dataset (WBCD); Wisconsin diagnostic breast cancer (WDBC); MACHINE LEARNING TECHNIQUES; DIAGNOSIS; CLASSIFICATION; MAMMOGRAMS; RULES; SEGMENTATION; PREDICTION; SYSTEM;
D O I
10.1016/j.bspc.2021.103141
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
The early detection of breast cancer (BC) has a significant impact on reducing the disease's mortality rate. As an effective cost- and time-saving tool, computer-aided diagnosis (CAD) systems have been developed in this research field, aiding clinicians and radiologists' decision-making process by offering highly accurate information. In this study, a shallow artificial neural network (ANN) model with one hidden layer is used to diagnose and predict BC using the Wisconsin breast cancer dataset (WBCD) and the Wisconsin diagnostic breast cancer (WDBC) dataset without employing feature optimization or selection algorithms. The datasets are divided into 80% for training and 20% for testing using five-fold cross-validation. The model's effectiveness and efficiency are evaluated based on sensitivity, specificity, precision, accuracy, and F1 score, along with the area under the receiver-operating characteristic curve (AUC). In the task of classifying benign and malignant tumours using the WBCD, the shallow ANN model showed promising performance with an average accuracy of 99.85%, specificity of 99.72%, sensitivity of 100%, precision of 99.69%, and F1 score of 99.84%. For BC detection using WDBC, it achieved an average accuracy of 99.47%, specificity of 99.53%, sensitivity of 99.59%, precision of 98.71%, and F1 score of 99.13%. The AUC of the proposed model was 99.86% and 99.56% for the WBCD and the WDBC dataset, respectively, illustrating the model's discrimination capacity. Moreover, the ANN model outperformed state-of-the-art models that integrate feature optimization and selection algorithms to classify BC tumours using WBCD and WDBC. Hence, the shallow ANN model presented here demonstrates significant potential for diagnosing BC using WBCD and WDBC without the need for feature optimization or selection algorithms.
引用
收藏
页数:10
相关论文
共 58 条
[1]  
Aalaei S, 2016, IRAN J BASIC MED SCI, V19, P476
[2]   An evolutionary artificial neural networks approach for breast cancer diagnosis [J].
Abbass, HA .
ARTIFICIAL INTELLIGENCE IN MEDICINE, 2002, 25 (03) :265-281
[3]   Breast cancer classification using deep belief networks [J].
Abdel-Zaher, Ahmed M. ;
Eldeib, Ayman M. .
EXPERT SYSTEMS WITH APPLICATIONS, 2016, 46 :139-144
[4]   Thermography Based Breast Cancer Detection Using Texture Features and Support Vector Machine [J].
Acharya, U. Rajendra ;
Ng, E. Y. K. ;
Tan, Jen-Hong ;
Sree, S. Vinitha .
JOURNAL OF MEDICAL SYSTEMS, 2012, 36 (03) :1503-1510
[5]   On Breast Cancer Detection: An Application of Machine Learning Algorithms on the Wisconsin Diagnostic Dataset [J].
Agarap, Abien Fred M. .
2ND INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND SOFT COMPUTING (ICMLSC 2018), 2015, :5-9
[6]   Trees vs Neurons: Comparison between random forest and ANN for high-resolution prediction of building energy consumption [J].
Ahmad, Muhammad Waseem ;
Mourshed, Monjur ;
Rezgui, Yacine .
ENERGY AND BUILDINGS, 2017, 147 :77-89
[7]   Evaluation of deep learning detection and classification towards computer-aided diagnosis of breast lesions in digital X-ray mammograms [J].
Al-antari, Mugahed A. ;
Han, Seung-Moo ;
Kim, Tae-Seong .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2020, 196
[8]   An efficient multiple sclerosis segmentation and detection system using neural networks [J].
Alshayeji, Mohammad H. ;
Al-Rousan, Mohammad A. ;
Ellethy, Hanem ;
Abed, Sa'ed .
COMPUTERS & ELECTRICAL ENGINEERING, 2018, 71 :191-205
[9]  
[Anonymous], 2012, UCI Machine Learning Repository-SMS Spam Collection Data Set
[10]   Using Machine Learning Algorithms for Breast Cancer Risk Prediction and Diagnosis [J].
Asri, Hiba ;
Mousannif, Hajar ;
Al Moatassime, Hassan ;
Noel, Thomas .
7TH INTERNATIONAL CONFERENCE ON AMBIENT SYSTEMS, NETWORKS AND TECHNOLOGIES (ANT 2016) / THE 6TH INTERNATIONAL CONFERENCE ON SUSTAINABLE ENERGY INFORMATION TECHNOLOGY (SEIT-2016) / AFFILIATED WORKSHOPS, 2016, 83 :1064-1069