Cutting-Edge Net of Biomedical Research and INnovation (CenBRAIN), School of Engineering, Westlake University, Hangzhou, Zhejiang, China.
Department of Neurosurgery, Second Affiliated Hospital, Zhejiang University School of Medicine, Hangzhou, Zhejiang, China.
Sci Rep. 2021 Dec 6;11(1):23498. doi: 10.1038/s41598-021-02798-8.
Deep learning techniques have led to significant advancements in seizure prediction research. However, corresponding used benchmarks are not uniform in published results. Moreover, inappropriate training and evaluation processes used in various work create overfitted models, making prediction performance fluctuate or unreliable. In this study, we analyzed the various data preparation methods, dataset partition methods in related works, and explained the corresponding impacts to the prediction algorithms. Then we applied a robust processing procedure that considers the appropriate sampling parameters and the leave-one-out cross-validation method to avoid possible overfitting and provide prerequisites for ease benchmarking. Moreover, a deep learning architecture takes advantage of a one-dimension convolutional neural network and a bi-directional long short-term memory network is proposed for seizure prediction. The architecture achieves 77.6% accuracy, 82.7% sensitivity, and 72.4% specificity, and it outperforms the indicators of other prior-art works. The proposed model is also hardware friendly; it has 6.274 k parameters and requires only 12.825 M floating-point operations, which is advantageous for memory and power constrained device implementations.
深度学习技术在癫痫预测研究中取得了重大进展。然而,发表的结果中相应的基准并不统一。此外,在不同的工作中使用不恰当的训练和评估过程会创建过拟合的模型,从而导致预测性能波动或不可靠。在这项研究中,我们分析了相关工作中各种数据准备方法和数据集划分方法,并解释了它们对预测算法的相应影响。然后,我们应用了一种稳健的处理程序,考虑了适当的采样参数和留一交叉验证方法,以避免可能的过拟合,并为易于基准测试提供前提条件。此外,我们提出了一种利用一维卷积神经网络和双向长短时记忆网络的深度学习架构,用于癫痫预测。该架构的准确率为 77.6%,灵敏度为 82.7%,特异性为 72.4%,优于其他现有技术的指标。所提出的模型也对硬件友好,它有 6274 个参数,只需要 12.825M 的浮点运算,有利于在内存和功耗受限的设备上实现。