基于脑电信号及共空间模式的抑郁症多分类算法研究

Research on multiclass classification algorithm of depression based on EEG signal and common spatial pattern

目的为了实现抑郁症患者的早发现、早治疗,提高抑郁症识别准确率及轻度患者召回率,对抑郁症的人工智能辅助诊断方法进行研究,提出了一种基于共空间模式的脑电信号二分类器投票算法。方法首先,提出一种基于方差分析确定最佳数据分段时长的方法进行数据预处理,并引入共空间模式算法进行特征提取。然后,结合当前二分类学习器解决多分类问题的扩展框架,设计了一种二分类器投票算法,解决了共空间模式算法在多分类问题上的局限性,实现健康对照、轻度患者、重度患者的脑电信号三分类。最后,利用抑郁症研究公开数据集MODMA,以准确率及轻度患者召回率为评价指标,通过对比实验,验证算法在性能上的优势。结果在公开数据集上,基于共空间模式的投票算法准确率高达97.55%,轻度患者召回率为91%,与两种传统的多分类共空间模式扩展策略性能相比,准确率分别提升1.32%和5.10%,轻度患者召回率分别提升9%和18%。结论基于共空间模式的投票算法能有效提高抑郁症识别准确率及轻度患者召回率,可为促进抑郁症的早发现早治疗提供算法支持。

Objective In order to realize the early detection and treatment of patients with depression,improve the recognition accuracy of depression and the recall rate of mild patients,the artificial intelligence aided diagnosis method of depression is studied,and a voting algorithm based on two-classifier of EEG signal is proposed.Methods Firstly,a method based on variance analysis to determine the optimal data segmentation time is proposed for data preprocessing,and the common space pattern algorithm is introduced for feature extraction.Then,combined with the extended framework of the current two-classifier to solve the multi classification problem,a two-classifier voting algorithm is designed to solve the limitation of the common spatial pattern algorithm in the multi classification problem,and realize the three classification of EEG signals of healthy controls,mild patients and severe patients.Finally,using the open data set MODMA of depression research,taking the accuracy rate and mild patient recall rate as evaluation indexes,the performance advantages of the algorithm are verified through comparative experiments.Results On the public dataset,the accuracy rate of the voting algorithm based on the common spatial pattern is as high as 97.55%,and the mild patient recall rate is 91%.Compared with the performance of the two traditional multi classification common spatial pattern extension strategies,the accuracy rate is improved by 1.32%and 5.10%,and the mild patient recall rate is improved by 9%and 18%.Conclusions The voting algorithm based on the common space pattern can effectively improve the recognition accuracy of depression and the recall rate of mild patients,and can provide algorithm support for promoting the early detection and treatment of depression.