We train a deep convolutional neural network on a heterogeneous set of 8522 routine EEGs from the Massachusetts General Hospital. We explore several strategies for optimizing model performance, including accounting for age and sleep stage.
The area under the receiver operating characteristic curve (AUC) on an independent test set (n = 851) is 0.917 marginally improved by including age (AUC = 0.924), and both age and sleep stages (AUC = 0.925), though not statistically significant.
The model architecture generalizes well to an independent dataset. Adding age and sleep stage to the model does not significantly improve performance.
Insights learned from misclassified examples, and minimal improvement by adding sleep stage and age suggest fruitful directions for further research.
我们在马萨诸塞州总医院的8522个常规EEG的异构集上训练深度卷积神经网络。我们探索了几种优化模型性能的策略,包括年龄和睡眠阶段。
独立测试集(n=851)上的受试者工作特征曲线(AUC)下面积通过包括年龄(AUC=0.924)略有改善,年龄和睡眠阶段(AUC=0.925),虽然没有统计学意义。
模型架构可以很好地推广到独立的数据集。将年龄和睡眠阶段添加到模型中不会显着提高性能。
从错误分类的例子中学到的见解,通过增加睡眠阶段和年龄,最小的改善为进一步的研究提供了有益的方向。