Enabling Unsupervised Training of Deep EEG Denoisers With Intelligent Partitioning
For researchers and practitioners in wearable EEG, this method removes the need for clean reference signals, enabling practical deep learning denoising in real-world settings.
The paper tackles unsupervised EEG denoising by proposing iPSD, which partitions EEG segments into independent noisy realizations to enable self-supervised training of deep denoisers. It achieves state-of-the-art performance, especially at low SNR (-10 dB) and with EMG artifacts, with spectral fidelity orders of magnitude higher than baselines.
Denoising wearable electroencephalogram (EEG) is inherently challenging since neural activity is not only subtle but also inseparable from spectrally overlapping noise artifacts. Classical signal processing methods, relying on fixed or heuristic rules, cannot handle the time-varying pervasive artifacts in wearable EEGs. Deep learning methods, on the other hand, show promise in decomposition-free EEG denoising using highly expressive neural networks, but the training requires artifact-free EEG, which is inherently unobtainable. To address this, we propose Intelligent Partitioning for Self-supervised Denoising (iPSD). Our method eliminates the need for clean references by learning to partition an input EEG segment into independent noisy realizations with the same underlying signal. This enables self-supervision of deep learning denoisers, even in zero-shot settings where only a single EEG segment to be denoised is available. We validate iPSD through extensive experiments, including validations on wearable EEG from in-ear sensors. The results show that iPSD achieves state-of-the-art performance, most notably under extremely low signal-to-noise ratios (down to -10 dB) and challenging artifacts (e.g., EMG), with spectral fidelity orders of magnitude higher than competitive baselines.