An Evolutionary Self-organizing Cost-Sensitive Radial Basis Function Neural Network to Deal with Imbalanced Data in Medical Diagnosis

被引:6
作者
Wu, Jia-Chao [1 ]
Shen, Jiang [1 ]
Xu, Man [2 ]
Liu, Fu-Sheng [1 ]
机构
[1] Tianjin Univ, Coll Management & Econ, Tianjin 300072, Peoples R China
[2] Nankai Univ, Business Sch, Tianjin 300071, Peoples R China
基金
中国国家自然科学基金;
关键词
Imbalanced data; Medical diagnosis; Radial basis function neural network; Cost sensitive; Genetic algorithm; Particle swarm optimization; GENETIC ALGORITHM; CLASSIFICATION; SMOTE; APPROXIMATION; PREDICTION; SELECTION; CLUSTER;
D O I
10.2991/ijcis.d.201012.005
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Class imbalance is a common issue in medical diagnosis. Although standard radial basis function neural network (RBF-NN) has achieved remarkably high performance on balanced data, its ability to classify imbalanced data is still limited. So far as we know, cost-sensitive learning is an advanced imbalanced data processing method. However, few studies have focused on the combination of RBF-NN and cost sensitivity. From our knowledge, only one paper has proposed a cost-sensitive RBF-NN for software defect prediction. However, the authors implemented a fixed RBF-NN structure. In this paper, a novel cost-sensitive RBF-NN that optimizes structure and parameters simultaneously is proposed to handle medical imbalanced data. Genetic algorithm (GA) and improved particle swarm optimization (IPSO) are used to optimize the structure and parameters of cost-sensitive RBF-NN respectively, and the optimization of cost-sensitive RBF-NN based on dynamic structure is realized. A cost-sensitive function determined adaptively by the sample distribution as the objective function of RBF-NN, so that it can adapt to datasets with different sample distributions. Experimental results show that the proposed cost-sensitive RBF-NN outperforms other state-of-the-art representative algorithms for live imbalanced medical diagnostic datasets in term of accuracy and area under curve (AUC). It can improve the accuracy of medical diagnosis and reduce the error rate of medical decisions. (C) 2020 The Authors. Published by Atlantis Press B.V.
引用
收藏
页码:1608 / 1618
页数:11
相关论文
共 53 条
[1]   Utilizing hybrid functional fuzzy wavelet neural networks with a teaching learning-based optimization algorithm for medical disease diagnosis [J].
Alneamy, Jamal Salahaldeen Majeed ;
Alnaish, Zakaria A. Hameed ;
Hashim, S. Z. Mohd ;
Alnaish, Rahma A. Hamed .
COMPUTERS IN BIOLOGY AND MEDICINE, 2019, 112
[2]   Applied improved RBF neural network model for predicting the broiler output energies [J].
Amini, Sherwin ;
Taki, Morteza ;
Rohani, Abbas .
APPLIED SOFT COMPUTING, 2020, 87
[3]  
[Anonymous], 2016, SCI WORLD J
[4]   Software defect prediction using cost-sensitive neural network [J].
Arar, Omer Faruk ;
Ayan, Kursat .
APPLIED SOFT COMPUTING, 2015, 33 :263-277
[5]  
Batista G. E. A. P. A., 2004, ACM SIGKDD Explor Newsl, V6, P20, DOI [10.1145/1007730.1007735, DOI 10.1145/1007730.1007735]
[6]   Ensembles for feature selection: A review and future trends [J].
Bolon-Canedo, Veronica ;
Alonso-Betanzos, Amparo .
INFORMATION FUSION, 2019, 52 :1-12
[7]   Novel Cost-Sensitive Approach to Improve the Multilayer Perceptron Performance on Imbalanced Data [J].
Castro, Cristiano L. ;
Braga, Antonio P. .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2013, 24 (06) :888-899
[8]   A method for resampling imbalanced datasets in binary classification tasks for real-world problems [J].
Cateni, Silvia ;
Colla, Valentina ;
Vannucci, Marco .
NEUROCOMPUTING, 2014, 135 :32-41
[9]   SMOTE: Synthetic minority over-sampling technique [J].
Chawla, Nitesh V. ;
Bowyer, Kevin W. ;
Hall, Lawrence O. ;
Kegelmeyer, W. Philip .
2002, American Association for Artificial Intelligence (16)
[10]  
Domingos P., 1999, C KNOWL DISC DAT MIN