# Robust learning from corrupted EEG with dynamic spatial filtering

arXiv: https://arxiv.org/abs/2105.12916 · HTML: https://mlsanity.com/paper/2105.12916/

## Key facts

- Cited by 14 papers in the map (data through 14 Oct 2026).
- Builds on 29 of its 130 references that are in the map.
- First posted on arXiv 27 May 2021.

## Builds on (most cited first, 29 of 29)

- [Attention Is All You Need](https://mlsanity.com/paper/1706.03762/) (2017)
- [PyTorch: An Imperative Style, High-Performance Deep Learning Library](https://mlsanity.com/paper/1912.01703/) (2019)
- [Squeeze-and-Excitation Networks](https://mlsanity.com/paper/1709.01507/) (2017)
- [Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification](https://mlsanity.com/paper/1502.01852/) (2015)
- [CBAM: Convolutional Block Attention Module](https://mlsanity.com/paper/1807.06521/) (2018)
- [Geometric deep learning: going beyond Euclidean data](https://mlsanity.com/paper/1611.08097/) (2016)
- [EEGNet: A Compact Convolutional Network for EEG-based Brain-Computer Interfaces](https://mlsanity.com/paper/1611.08024/) (2016)
- [DeepSleepNet: a Model for Automatic Sleep Stage Scoring based on Raw Single-Channel EEG](https://mlsanity.com/paper/1703.04046/) (2017)
- [Deep learning-based electroencephalography analysis: a systematic review](https://mlsanity.com/paper/1901.05498/) (2019)
- [Uncovering the structure of clinical EEG signals with self-supervised learning](https://mlsanity.com/paper/2007.16104/) (2020)
- [SeqSleepNet: End-to-End Hierarchical Recurrent Neural Network for Sequence-to-Sequence Automatic Sleep Staging](https://mlsanity.com/paper/1809.10932/) (2018)
- [XSleepNet: Multi-View Sequential Model for Automatic Sleep Staging](https://mlsanity.com/paper/2007.05492/) (2020)
- [Subject-Aware Contrastive Learning for Biosignals](https://mlsanity.com/paper/2007.04871/) (2020)
- [A deep learning architecture for temporal sleep stage classification using multivariate and multimodal time series](https://mlsanity.com/paper/1707.03321/) (2017)
- [Learning Representations from EEG with Deep Recurrent-Convolutional Neural Networks](https://mlsanity.com/paper/1511.06448/) (2015)
- [Machine-Learning-Based Diagnostics of EEG Pathology](https://mlsanity.com/paper/2002.05115/) (2020)
- [Dreem Open Datasets: Multi-Scored Sleep Datasets to compare Human and Automated sleep staging](https://mlsanity.com/paper/1911.03221/) (2019)
- [RobustSleepNet: Transfer learning for automated sleep staging at scale](https://mlsanity.com/paper/2101.02452/) (2021)
- [Manifold-regression to predict from MEG/EEG brain signals without source modeling](https://mlsanity.com/paper/1906.02687/) (2019)
- [Learning Robust Features using Deep Learning for Automatic Seizure Detection](https://mlsanity.com/paper/1608.00220/) (2016)
- [Learning from Heterogeneous EEG Signals with Differentiable Channel Reordering](https://mlsanity.com/paper/2010.13694/) (2020)
- [Feature Learning from Incomplete EEG with Denoising Autoencoder](https://mlsanity.com/paper/1410.0818/) (2014)
- [Deep learning with convolutional neural networks for decoding and visualization of EEG pathology](https://mlsanity.com/paper/1708.08012/) (2017)
- [Invariant Representations for Noisy Speech Recognition](https://mlsanity.com/paper/1612.01928/) (2016)
- [Learning Noise-Invariant Representations for Robust Speech Recognition](https://mlsanity.com/paper/1807.06610/) (2018)
- [Spatial Attention for Far-field Speech Recognition with Deep Beamforming Neural Networks](https://mlsanity.com/paper/1911.02115/) (2019)
- [Brain-Computer Interface with Corrupted EEG Data: A Tensor Completion Approach](https://mlsanity.com/paper/1806.05017/) (2018)
- [Real-Time Sleep Staging using Deep Learning on a Smartphone for a Wearable EEG](https://mlsanity.com/paper/1811.10111/) (2018)
- [MYND: Unsupervised Evaluation of Novel BCI Control Strategies on Consumer Hardware](https://mlsanity.com/paper/2002.11754/) (2020)

## Built on by (most cited first, 14 of 14)

- [Data augmentation for learning predictive models on EEG: a systematic comparison](https://mlsanity.com/paper/2206.14483/) (2022)
- [CoRe-Sleep: A Multimodal Fusion Framework for Time Series Robust to Imperfect Modalities](https://mlsanity.com/paper/2304.06485/) (2023)
- [Interpretable and Robust AI in EEG Systems: A Survey](https://mlsanity.com/paper/2304.10755/) (2023)
- [EEG to fMRI Synthesis Benefits from Attentional Graphs of Electrode Relationships](https://mlsanity.com/paper/2203.03481/) (2022)
- [Explainable artificial intelligence approaches for brain-computer interfaces: a review and design space](https://mlsanity.com/paper/2312.13033/) (2023)
- [Geometric Neural Network based on Phase Space for BCI-EEG decoding](https://mlsanity.com/paper/2403.05645/) (2024)
- [A Learnable Counter-condition Analysis Framework for Functional Connectivity-based Neurological Disorder Diagnosis](https://mlsanity.com/paper/2310.03964/) (2023)
- [A Modulation Layer to Increase Neural Network Robustness Against Data Quality Issues](https://mlsanity.com/paper/2107.08574/) (2021)
- [A distributed neural network architecture for dynamic sensor selection with application to bandwidth-constrained body-sensor networks](https://mlsanity.com/paper/2308.08379/) (2023)
- [Masked EEG Modeling for Driving Intention Prediction](https://mlsanity.com/paper/2408.07083/) (2024)
- [Data Normalization Strategies for EEG Deep Learning](https://mlsanity.com/paper/2506.22455/) (2025)
- [Cross-Subject Domain Adaptation for Classifying Working Memory Load with Multi-Frame EEG Images](https://mlsanity.com/paper/2106.06769/) (2021)
- [Scope and Arbitration in Machine Learning Clinical EEG Classification](https://mlsanity.com/paper/2303.06386/) (2023)
- [Window Stacking Meta-Models for Clinical EEG Classification](https://mlsanity.com/paper/2401.10283/) (2024)

## Abstract (arXiv)

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 consumer-grade 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.
