.png){kind=icon}
.png){kind=icon}
Return
Stroke classification based on deep reinforcement learning over stroke screening imbalanced data
.png){kind=icon}
PRE
AI
DOI：10.1016/j.compeleceng.2023.109069.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Stroke screening is a crucial measure for reducing stroke occurrence, disability, and mortality. However there are numerous risk factors and the limited number of high-risk stroke groups in all screening populations, the screening data is redundant and imbalanced. We propose an improved feature selection algorithm to identify stroke key risk factors and an oversampling MRF-SMOTE algorithm to balance data. Then, a deep reinforcement learning classification model based on the Dueling DQN (Deep Q-network) algorithm is constructed for stroke classification, with the optimized loss function. Benchmark models include KNN, SVM, random forest, and Dueling DQN. Stroke screening data is pre-processed by selecting key risk factors and oversampled by MRFSMOTE. Experiments show that the optimized Dueling DQN model can classify the preprocessed stroke screening data in terms of accuracy, AUC, precision, and F1-measure, which are 0.8982, 0.96, 0.899, and 0.8981 and improved by 4.37 %, 6.92 %, 3.78 % and 4.04 %, respectively, compared with the existing Dueling DQN.
Keywords:
Stroke classification
Stroke key risk factors selection
Deep reinforcement learning model
Imbalanced data
Journal
C
IF:
4.9
Papers:
6.7K
Citations:
1.3W