Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces DiffEEG, a self-supervised denoising diffusion model designed to tackle the challenges of annotation scarcity and class imbalance in EEG-based seizure detection. By leveraging a 1D U-Net architecture and multi-head self-attention, DiffEEG is pre-trained on a large dataset of unlabeled EEG segments and fine-tuned using reinforcement learning to optimize for F1-score, particularly focusing on rare seizure events. The model achieves notable performance metrics, including 81% accuracy and 85% weighted F1 for binary seizure detection, demonstrating its potential for effective clinical deployment with minimal labeled data.
DiffEEG achieves 81% accuracy in binary seizure detection while prioritizing sensitivity to rare events, showcasing the power of self-supervised learning in EEG analysis.
Deep learning for EEG-based seizure detection faces critical challenges: severe annotation scarcity and extreme class imbalance, where ictal events comprise less than 10\% of clinical recordings. We present DiffEEG, a 9.6M-parameter self-supervised foundation model that addresses both limitations through denoising diffusion pre-training and reinforcement learning (RL)-based fine-tuning. Pre-trained on 1.3M unlabeled segments from the Temple University Hospital Seizure Corpus (TUHSZ), DiffEEG learns generic neural representations via a 1D U-Net with multi-head self-attention. For downstream adaptation, a reinforced decision layer employs policy gradient optimization to directly maximize F1-score, prioritizing sensitivity to rare seizure events over overall accuracy. Under strict patient-wise evaluation (279 patients, Leave-One-Fold-Out), DiffEEG achieves 61\% accuracy and 59\% F1 for 4-class seizure subtyping, and 81\% accuracy with 85\% weighted F1 for binary detection, maintaining clinically viable seizure recall (59\%) despite extreme imbalance (6.7\% prevalence). Segment-level evaluation establishes an upper bound of 97.6\% accuracy, confirming strong architectural capacity. DiffEEG demonstrates that diffusion-based pre-training combined with metric-aware reinforcement learning enables clinically deployable seizure monitoring with minimal labeled data requirements.