基于优化极限学习机的CVD预测模型研究-《郑州大学学报(工学版)》

文章信息/Info

Author(s):: Lu Peng1; Li Qihang1; Shang Liga 3; Li Xinjian 1; Zhang Wei 1; 1. School of Electrical Engineering, Zhengzhou University; 2. Henan Provincial Collaborative Innovation Center for Internet Medical and Health Services; 3. Primary and Secondary School Health Care Center in Dongcheng District, Beijing

Keywords:: Cardiovascular diseases; Risk prediction; extreme learning machine; particle swarm

摘要:: 利用机器学习算法,改变传统心血管疾病(CVD)预测模型的严格数理化公式,以增加危险因素的纳入,降低数据格式的要求.首先提出利用基于单隐层前馈神经网络(SLFNs) 的极限学习机(ELM) 算法建立CVD预测模型,进一步通过五阶段连续变异方式建立增强领导粒子的粒子群算法(ELPSO),以粒子群(PSO) 算法的优化策略,!对SLFNsJ的隐层单元参数进行优化,通过对UCI数据库stalog(heart)数据集和heart disease 分析结果显示,所提ELPSO-ELM模型测试正确率分别达到85.71%,84.00%,AUC(ROC曲线下面积)分别达到0.9024、0.8423,高于传统CVD预测模型，同时放松了数据线性化约束，能纳入更多的复杂危险因素。线性化约束!能纳入更多的复杂危险因素

Abstract:: Abstract: In order to increase the risk factors that could be accepted and reduce the data format requirements in cardiovascular disease (CVD) predicition models,machine learning algorithms were used to change the strict mathematical formulas of traditional CVD prediction models.Frstly,a CVD prediction model by extreme learning machine (ELM)algorithm based on single hidden layer feed-forward neural network (SLFNs) was proposed. Further more,an enhanced leader particle swarm optimization(ELPSO) through a five-staged successive mutation method was used ,and the optimized strategy of PSO was also used to optimize the SLFNs hidden layer units parameters.The analysis results on Statlog(Heart) dataset and Heart Disease Dataset of UCI database indicated that the test accuracy of proposed ELPSO-ELM model could reach 85.71%and 84.00%respectively, the AUC(The area under the ROC curve )could reach 0.9024 and 0.8423 respectively.They were higher than conventional CVD prediction models.The proposed models relaxed the linear constraints of data format and more complex risk factors could be accepted.