.png){kind=icon}
Return
Data Augmentation for Seizure Prediction With Generative Diffusion Model
.png){kind=icon}
PRE
AI
DOI：10.1109/TCDS.2024.3489357.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Data augmentation (DA) can significantly strengthen the electroencephalogram (EEG)-based seizure prediction methods. However, existing DA approaches are just the linear transformations of original data and cannot explore the feature space to increase diversity effectively. Therefore, we propose a novel diffusion-based DA method called DiffEEG. DiffEEG can fully explore data distribution and generate samples with high diversity, offering extra information to classifiers. It involves two processes: the diffusion process and the denoised process. In the diffusion process, the model incrementally adds noise with different scales to EEG input and converts it into random noise. In this way, the representation of data can be learned. In the denoised process, the model utilizes learned knowledge to sample synthetic data from random noise input by gradually removing noise. The randomness of input noise and the precise representation enable the synthetic samples to possess diversity while ensuring the consistency of feature space. We compared DiffEEG with original, down-sampling, sliding windows and recombination methods, and integrated them into five representative classifiers. The experiments demonstrate the effectiveness and generality of our method. With the contribution of DiffEEG, the multiscale CNN achieves state-of-the-art performance, with an average sensitivity, FPR, AUC of 95.4%, 0.051/h, 0.932 on the CHB-MIT database and 93.6%, 0.121/h, 0.822 on the Kaggle database.
Keywords:
Data augmentation (DA)
deep learning
diffusion model
seizure prediction
Journal
IF:
4.9
Papers:
1.0K
Citations:
3.5K
Organization
Cited Papers
ShareSaveAn Overview of EEG-based Machine Learning Methods in Seizure Prediction and Opportunities for Neurologists in this Field.png)
Neuroscience
NeuroscienceIF0
PREAI
ShareSaveShareSave
ShareSaveShareSave
View more