用生成扩散模型预测多通道脑电未来趋势,实现癫痫发作早期预警。
EEG-DIF: Early Warning of Epileptic Seizures through Generative Diffusion Model-based Multi-channel EEG Signals Forecasting
- 将多信号预测转为图像修复任务,捕捉脑电信号时空关联。
- 在公开数据集上实现0.89的癫痫发作预警准确率。
- 适合临床辅助诊断与癫痫研究者使用。
多通道脑电图(EEG)常用于癫痫等疾病的诊断与评估。当前基于深度学习的算法多聚焦于实时信号分类,缺乏对未来趋势的预测能力。由于多通道EEG可视为大脑不同位置传感器采集的时空信号,如何构建其时空特征表示以支持未来趋势预测成为关键问题。本文提出基于生成扩散模型的多信号预测算法EEG-DIF,将多信号预测任务转化为图像补全问题,从而全面表征多通道脑电的时空相关性与未来演化模式。我们采用公开癫痫EEG数据集验证EEG-DIF,结果表明该方法能准确同步预测多通道脑电未来趋势。基于生成数据的癫痫发作早期预警准确率达0.89。总体而言,EEG-DIF为多通道脑电表征提供新范式,并为癫痫早期预警提供创新算法,有助于优化临床诊断流程。代码已开源:https://github.com/JZK00/EEG-DIF。
原文摘要 · Abstract (English)
Multi-channel EEG signals are commonly used for the diagnosis and assessment of diseases such as epilepsy. Currently, various EEG diagnostic algorithms based on deep learning have been developed. However, most research efforts focus solely on diagnosing and classifying current signal data but do not consider the prediction of future trends for early warning. Additionally, since multi-channel EEG can be essentially regarded as the spatio-temporal signal data received by detectors at different locations in the brain, how to construct spatio-temporal information representations of EEG signals to facilitate future trend prediction for multi-channel EEG becomes an important problem. This study proposes a multi-signal prediction algorithm based on generative diffusion models (EEG-DIF), which transforms the multi-signal forecasting task into an image completion task, allowing for comprehensive representation and learning of the spatio-temporal correlations and future developmental patterns of multi-channel EEG signals. Here, we employ a publicly available epilepsy EEG dataset to construct and validate the EEG-DIF. The results demonstrate that our method can accurately predict future trends for multi-channel EEG signals simultaneously. Furthermore, the early warning accuracy for epilepsy seizures based on the generated EEG data reaches 0.89. In general, EEG-DIF provides a novel approach for characterizing multi-channel EEG signals and an innovative early warning algorithm for epilepsy seizures, aiding in optimizing and enhancing the clinical diagnosis process. The code is available at https://github.com/JZK00/EEG-DIF.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。