arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1902.01012cs.LGq-bio.QMstat.ML
收藏

使用EEG信号和机器学习的癫痫发作类型分类:设定基准

Seizure Type Classification using EEG signals and Machine Learning: Setting a benchmark

IBM Research Australia(IBM澳大利亚研究院);Google Health(谷歌健康)

Subhrajit Roy, Umar Asif, Jianbin Tang, Stefan Harrer

摘要:

癫痫发作类型的准确分类在癫痫患者的治疗和疾病管理中起着至关重要的作用。癫痫发作类型不仅影响药物的选择,而且影响患者可以安全从事的活动范围。随着人工智能使能的自动癫痫发作检测的最新进展,下一个前沿是癫痫发作类型的自动分类。在这一点上,在本文中,我们探讨了应用机器学习算法的多类癫痫发作类型分类。我们使用了最近发布的TUH EEG癫痫发作语料库(V1.4.0和V1.5.2),并进行了彻底的搜索空间探索,以评估各种预处理技术,机器学习算法和相应的超参数组合在此任务上的性能。我们表明,我们的算法可以达到加权的$F1$得分高达0.901的逐例交叉验证和0.561的逐例交叉验证,从而为头皮EEG为基础的多类癫痫发作类型分类的基准。

英文摘要:

Accurate classification of seizure types plays a crucial role in the treatment and disease management of epileptic patients. Epileptic seizure types not only impact the choice of drugs but also the range of activities a patient can safely engage in. With recent advances being made towards artificial intelligence enabled automatic seizure detection, the next frontier is the automatic classification of seizure types. On that note, in this paper, we explore the application of machine learning algorithms for multi-class seizure type classification. We used the recently released TUH EEG seizure corpus (V1.4.0 and V1.5.2) and conducted a thorough search space exploration to evaluate the performance of a combination of various pre-processing techniques, machine learning algorithms, and corresponding hyperparameters on this task. We show that our algorithms can reach a weighted $F1$ score of up to 0.901 for seizure-wise cross validation and 0.561 for patient-wise cross validation thereby setting a benchmark for scalp EEG based multi-class seizure type classification.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑