计算机工程与应用 ›› 2020, Vol. 56 ›› Issue (10): 268-272.DOI: 10.3778/j.issn.1002-8331.1901-0261

• 工程与应用 • 上一篇    下一篇

优化极限学习机及其在脑卒中TCD数据分类应用

耿银凤,张雪英,李凤莲,胡风云,贾文辉,王超   

  1. 1.太原理工大学 信息与计算机学院,太原 030024
    2.山西省人民医院 神经内科,太原 030012
  • 出版日期:2020-05-15 发布日期:2020-05-13

Optimized Extreme Learning Machine and Its Application in Classification of Stroke TCD Data

GENG Yinfeng, ZHANG Xueying, LI Fenglian, HU Fengyun, JIA Wenhui, WANG Chao   

  1. 1.College of Information and Computer, Taiyuan University of Technology, Taiyuan 030024, China
    2.Department of Neurology, Shanxi Province People’s Hospital, Taiyuan 030012, China
  • Online:2020-05-15 Published:2020-05-13

摘要:

为提高脑卒中经颅多普勒(Transcranial Doppler,TCD)数据分类的效率和准确率,应用蝙蝠算法(Bat Algorithm,BA)优化极限学习机(Extreme Learning Machine,ELM)模型进行脑卒中分类预测。在训练ELM模型时,隐含层输入权值矩阵和隐含层阈值矩阵元素产生的随机性影响了模型性能。为此,利用BA对ELM参数中的输入权值矩阵和隐含层阈值矩阵进行了优化,并用BA-ELM模型对实验所用的TCD数据集进行分类。实验结果表明,BA-ELM模型的分类准确率比ELM提高了22.77%,能有效进行脑卒中预测。

关键词: 蝙蝠算法, 极限学习机, 经颅多普勒, 脑卒中

Abstract:

In order to improve the efficiency and accuracy of Transcranial Doppler(TCD)data classification for stroke prediction, an Extreme Learning Machine(ELM) optimized by Bat Algorithm(BA) model is proposed. The element values of input weight matrix and threshold matrix in hidden layer are randomly set in the process of training the ELM model, which badly affect the classification performance of the network. To solve this problem, BA is used to optimize the parameters mentioned above. The proposed BA-ELM model is further used to classify the stroke patient TCD data in the experiment. Experimental results indicate that, the accuracy of the BA-ELM model is improved by 22.77% compared with the typical ELM model for classifying TCD data set, so the proposed BA-ELM model can effectively be used for stroke prediction.

Key words: bat algorithm, extreme learning machine, transcranial Doppler, stroke