Skip to Main content Skip to Navigation
Preprints, Working Papers, ...

Robust learning from corrupted EEG with dynamic spatial filtering

Abstract : Building machine learning models using EEG recorded outside of the laboratory setting requires methods robust to noisy data and randomly missing channels. This need is particularly great when working with sparse EEG montages (1-6 channels), often encountered in consumergrade or mobile EEG devices. Neither classical machine learning models nor deep neural networks trained end-to-end on EEG are typically designed or tested for robustness to corruption, and especially to randomly missing channels. While some studies have proposed strategies for using data with missing channels, these approaches are not practical when sparse montages are used and computing power is limited (e.g., wearables, cell phones). To tackle this problem, we propose dynamic spatial filtering (DSF), a multi-head attention module that can be plugged in before the first layer of a neural network to handle missing EEG channels by learning to focus on good channels and to ignore bad ones. We tested DSF on public EEG data encompassing ∼4,000 recordings with simulated channel corruption and on a private dataset of ∼100 at-home recordings of mobile EEG with natural corruption. Our proposed approach achieves the same performance as baseline models when no noise is applied, but outperforms baselines by as much as 29.4% accuracy when significant channel corruption is present. Moreover, DSF outputs are interpretable, making it possible to monitor channel importance in real-time. This approach has the potential to enable the analysis of EEG in challenging settings where channel corruption hampers the reading of brain signals.
Complete list of metadata

https://hal.ird.fr/ird-03274947
Contributor : Hubert Banville Connect in order to contact the contributor
Submitted on : Wednesday, June 30, 2021 - 3:44:01 PM
Last modification on : Wednesday, September 29, 2021 - 6:06:01 PM
Long-term archiving on: : Friday, October 1, 2021 - 6:45:37 PM

File

DL_on_corrupted_sparse_EEG.pdf
Files produced by the author(s)

Identifiers

  • HAL Id : ird-03274947, version 1

Citation

Hubert Banville, Sean Wood, Chris Aimone, Denis-Alexander Engemann, Alexandre Gramfort. Robust learning from corrupted EEG with dynamic spatial filtering. 2021. ⟨ird-03274947⟩

Share

Metrics

Record views

40

Files downloads

47