论文标题
农奴:使用嵌入,规则和功能的可解释的睡眠分期
SERF: Interpretable Sleep Staging using Embeddings, Rules, and Features
论文作者
论文摘要
最近基于深度学习的临床决策支持系统的准确性是有希望的。但是,缺乏模型可解释性仍然是医疗保健中人工智能广泛采用的障碍。使用睡眠作为案例研究,我们提出了一种可推广的方法,将临床解释性与黑盒深度学习得出的高精度相结合。多聚词(PSG)的临床医生确定的睡眠阶段仍然是评估睡眠质量的金标准。但是,专家的PSG手册注释昂贵且良好。我们建议使用嵌入式,规则和功能来读取PSG的农奴,可解释的睡眠分期。农奴通过从AASM手册中得出的有意义的特征来解释分类的睡眠阶段,用于睡眠和相关事件的评分。在农奴中,从卷积和复发性神经网络的混合体获得的嵌入被转移到可解释的特征空间。这些代表性的可解释功能用于训练简单的模型,例如浅决策树进行分类。模型结果将在两个公开可用的数据集上进行验证。农奴超过了可解释的睡眠分期的当前最新面积。使用梯度增强的树作为分类器,在当前最新的黑盒模型的2%以内,农奴将获得0.766 $κ$和0.870 AUC-ROC。
The accuracy of recent deep learning based clinical decision support systems is promising. However, lack of model interpretability remains an obstacle to widespread adoption of artificial intelligence in healthcare. Using sleep as a case study, we propose a generalizable method to combine clinical interpretability with high accuracy derived from black-box deep learning. Clinician-determined sleep stages from polysomnogram (PSG) remain the gold standard for evaluating sleep quality. However, PSG manual annotation by experts is expensive and time-prohibitive. We propose SERF, interpretable Sleep staging using Embeddings, Rules, and Features to read PSG. SERF provides interpretation of classified sleep stages through meaningful features derived from the AASM Manual for the Scoring of Sleep and Associated Events. In SERF, the embeddings obtained from a hybrid of convolutional and recurrent neural networks are transposed to the interpretable feature space. These representative interpretable features are used to train simple models like a shallow decision tree for classification. Model results are validated on two publicly available datasets. SERF surpasses the current state-of-the-art for interpretable sleep staging by 2%. Using Gradient Boosted Trees as the classifier, SERF obtains 0.766 $κ$ and 0.870 AUC-ROC, within 2% of the current state-of-the-art black-box models.