论文标题

一种机器学习方法来自动分类八种睡眠障碍

A Machine Learning Approach to Automatic Classification of Eight Sleep Disorders

论文作者

Zhuang, Dylan, Rao, Ivey, Ibrahim, Ali K

论文摘要

在这项研究中,我们试图回答以下基础研究问题:机器学习模型是否能够高精度对所有类型的睡眠障碍进行分类?在睡眠障碍信号的不同方式中,有些比其他人更重要吗?原始信号将深度学习模型用作输入时是否可以改善它们的性能?先前的研究表明,大多数睡眠障碍属于八类。为了研究机器学习模型的性能,将多肌仪记录分类为八类睡眠病理学,我们选择了环状交替的模式睡眠数据库。我们开发了一个多通道深度学习模型,其中将一组卷积神经网络应用于六个不同模式的原始信号渠道,包括三个EEG信号的频道和EMG,ECG和EOG信号的一个通道。为了将DL模型的性能与其他模型进行比较,我们设计了一种模型,该模型采用光谱特征而不是原始信号作为其输入。我们首先使用RF算法研究了信号模态的“重要性”问题。我们发现,在四种信号方式中,ECG对重要特征和EMG第二做出了贡献。然后,我们研究了建议的机器学习模型的准确性。我们验证了将原始信号作为其输入的多通道DL-R模型的表现优于所有其他模型,其灵敏度和特异性得分均高于95%。这种准确性表现与那些已发表的结果相当,这些结果涉及较少类型的睡眠障碍。我们采用了两种流行的热图生成技术,我们证实了DL模型的出色性能是由于CNN网络从原始信号中提取有效功能的能力。

In this research, we attempt to answer the following basic research questions: Is a machine learning model able to classify all types of sleep disorders with high accuracy? Among the different modalities of sleep disorder signals, are some more important than others? Do raw signals improve the performance of a deep learning model when they are used as inputs? Prior research showed that most sleep disorders belong to eight categories. To study the performance of machine learning models in classifying polysomnography recordings into the eight categories of sleep pathologies, we selected the Cyclic Alternating Pattern Sleep Database. We developed a multi-channel Deep Learning model where a set of Convolutional Neural Networks were applied to six channels of raw signals of different modalities, including three channels of EEG signals and one channel each of EMG, ECG , and EOG signals. To compare the performance of the DL model with other models, we designed a model that took spectral features, instead of raw signals, as its inputs. We first studied the "importance" issue of signal modalities using the RF algorithm. We found that ECG contributed most to the important features and EMG second, among the four signal modalities. We then studied the accuracy performance of the proposed machine learning models. We verified that the multi-channel DL-R model, which took raw signals as its inputs, outperformed all other models, with its sensitivity and specificity scores both being above 95 %. This accuracy performance is on a par with those published results which dealt with fewer types of sleep disorders. We adopted two popular heatmap-generating techniques, with which we confirmed that the DL model's superior performance was owing to the CNN network's ability to extract potent features from raw signals.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源