A GA-based feature selection and parameter optimization of an ANN in diagnosing breast cancer

被引:79
作者
Ahmad, Fadzil [1 ,2 ]
Isa, Nor Ashidi Mat [1 ]
Hussain, Zakaria [2 ]
Osman, Muhammad Khusairi [2 ]
Sulaiman, Siti Noraini [2 ]
机构
[1] Univ Sains Malaysia, Sch Elect & Elect Engn, Imaging & Intelligent Syst Res Team ISRT, Nibong Tebal 14300, Pulau Pinang, Malaysia
[2] Univ Teknol Mara UiTM, Fac Elect Engn, Permatang Pauh 13500, Pulau Pinang, Malaysia
关键词
Genetic algorithm; Artificial neural network; Multilayer perceptron; Feature selection; Backpropagation; Classification accuracy; ARTIFICIAL NEURAL-NETWORKS; SUPPORT VECTOR MACHINES; GENETIC ALGORITHM; CLASSIFICATION RULES; GLOBAL OPTIMIZATION; SYSTEM;
D O I
10.1007/s10044-014-0375-9
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Breast cancer is the most common cancer diagnosed and cause of death among women worldwide. There is evidence that early detection and treatment can increase the survival rate of breast cancer patients. The traditional method for diagnosing the disease relies on human experiences to identify the presence of certain pattern from the database. It is prone to human error, time consuming and labour intensive. Therefore, this work proposes an automatic breast cancer diagnosis technique using a genetic algorithm (GA) for simultaneous feature selection and parameter optimization of an artificial neural network (ANN). The proposed algorithm is implemented with three different variations of the backpropagation technique namely the resilient back-propagation (GAANN_RP), Levenberg-Marquardt (GAANN_LM) and gradient descent with momentum (GAANN_GD) for fine tuning of the weight of ANN, and their performances are compared. Besides, the effect of the feature selection and manual determination of the hidden node size has also been investigated. Interestingly, one of the proposed algorithms called GAANN_RP produces the best and on average, 99.24 and 98.29 % correct classification, respectively, on the Wisconsin breast cancer dataset, which is comparable with the results gathered from other works found in the literature.
引用
收藏
页码:861 / 870
页数:10
相关论文
共 53 条
[1]   An evolutionary artificial neural networks approach for breast cancer diagnosis [J].
Abbass, HA .
ARTIFICIAL INTELLIGENCE IN MEDICINE, 2002, 25 (03) :265-281
[2]   Supervised fuzzy clustering for the identification of fuzzy classifiers [J].
Abonyi, J ;
Szeifert, F .
PATTERN RECOGNITION LETTERS, 2003, 24 (14) :2195-2207
[3]   Support vector machines combined with feature selection for breast cancer diagnosis [J].
Akay, Mehmet Fatih .
EXPERT SYSTEMS WITH APPLICATIONS, 2009, 36 (02) :3240-3247
[4]  
Albrecht AA, 2002, ICONIP'02: PROCEEDINGS OF THE 9TH INTERNATIONAL CONFERENCE ON NEURAL INFORMATION PROCESSING, P184
[5]   A multi-objective memetic and hybrid methodology for optimizing the parameters and performance of artificial neural networks [J].
Almeida, Leandro M. ;
Ludermir, Teresa B. .
NEUROCOMPUTING, 2010, 73 (7-9) :1438-1450
[6]  
[Anonymous], 2024, P INT SCI CONFERENCE
[7]  
[Anonymous], P IEEE EMBC CMBEC
[8]   Empirical study of feature selection methods based on individual feature evaluation for classification problems [J].
Arauzo-Azofra, Antonio ;
Aznarte, Jose Luis ;
Benitez, Jose M. .
EXPERT SYSTEMS WITH APPLICATIONS, 2011, 38 (07) :8170-8177
[9]   G-Prop: Global optimization of multilayer perceptrons using GAs [J].
Castillo, PA ;
Merelo, JJ ;
Prieto, A ;
Rivas, V ;
Romero, G .
NEUROCOMPUTING, 2000, 35 :149-163
[10]  
Esugasini S, 2005, LECT NOTES COMPUTER, V3682, P166, DOI [10.1007/11552451_17, DOI 10.1007/11552451_17]