MBrain: A Multi-channel Self-Supervised Learning Framework for Brain Signals (2306.13102v1)
Abstract: Brain signals are important quantitative data for understanding physiological activities and diseases of human brain. Most existing studies pay attention to supervised learning methods, which, however, require high-cost clinical labels. In addition, the huge difference in the clinical patterns of brain signals measured by invasive (e.g., SEEG) and non-invasive (e.g., EEG) methods leads to the lack of a unified method. To handle the above issues, we propose to study the self-supervised learning (SSL) framework for brain signals that can be applied to pre-train either SEEG or EEG data. Intuitively, brain signals, generated by the firing of neurons, are transmitted among different connecting structures in human brain. Inspired by this, we propose MBrain to learn implicit spatial and temporal correlations between different channels (i.e., contacts of the electrode, corresponding to different brain areas) as the cornerstone for uniformly modeling different types of brain signals. Specifically, we represent the spatial correlation by a graph structure, which is built with proposed multi-channel CPC. We theoretically prove that optimizing the goal of multi-channel CPC can lead to a better predictive representation and apply the instantaneou-time-shift prediction task based on it. Then we capture the temporal correlation by designing the delayed-time-shift prediction task. Finally, replace-discriminative-learning task is proposed to preserve the characteristics of each channel. Extensive experiments of seizure detection on both EEG and SEEG large-scale real-world datasets demonstrate that our model outperforms several state-of-the-art time series SSL and unsupervised models, and has the ability to be deployed to clinical practice.
- EEG signal analysis for diagnosing neurological disorders using discrete wavelet transform and intelligent techniques. Sensors 20, 9 (2020), 2505.
- The great time series classification bake off: A review and experimental evaluation of recent algorithmic advances. Data mining and knowledge discovery 31, 3 (2017), 606–660.
- Uncovering the structure of clinical EEG signals with self-supervised learning. Journal of Neural Engineering 18, 4 (2021), 046020.
- Language models are few-shot learners. In NeurIPS. 1877–1901.
- BrainNet: Epileptic wave detection from SEEG with hierarchical graph diffusion learning. In KDD. 2741–2751.
- Accurate EEG-based emotion recognition on combined features using deep convolutional neural networks. IEEE Access 7 (2019), 44317–44328.
- A simple framework for contrastive learning of visual representations. In ICML. 1597–1607.
- Wei Chen and Ke Shi. 2021. Multi-scale attention convolutional neural network for time series classification. Neural Networks 136 (2021), 126–140.
- Xinlei Chen and Kaiming He. 2021. Exploring simple siamese representation learning. In CVPR. 15745–15753.
- Deep learning for electroencephalogram (EEG) classification tasks: A review. Journal of Neural Engineering 16, 3 (2019), 031001.
- Spontaneous travelling cortical waves gate perception in behaving primates. Nature 587, 7834 (2020), 432–436.
- ROCKET: Exceptionally fast and accurate time series classification using random convolutional kernels. Data Mining and Knowledge Discovery 34, 5 (2020), 1454–1495.
- Minirocket: A very fast (almost) deterministic transform for time series classification. In KDD. 248–257.
- BERT: Pre-training of deep bidirectional Transformers for language understanding. arXiv preprint arXiv:1810.04805 (2018).
- Unsupervised visual representation learning by context prediction. In ICCV. 1422–1430.
- Differential entropy feature for EEG-based emotion classification. In NER. 81–84.
- Time-series representation learning via temporal and contextual contrasting. In IJCAI. 2352–2359.
- Unsupervised scalable representation learning for multivariate time series. In NeurIPS.
- Clive WJ Granger. 1969. Investigating causal relations by econometric models and cross-spectral methods. Econometrica: journal of the Econometric Society (1969), 424–438.
- Sepp Hochreiter and Jürgen Schmidhuber. 1997. Long short-term memory. Neural Computation 9, 8 (1997), 1735–1780.
- Lina Elsherif Ismail and Waldemar Karwowski. 2020. Applications of EEG indices for the quantification of human cognitive performance: A systematic review and bibliometric analysis. PloS one 15, 12 (2020), e0242857.
- Diederik P Kingma and Jimmy Ba. 2015. Adam: A method for stochastic optimization. In ICLR.
- Cognitive analysis of working memory load from eeg, by a deep recurrent neural network. In ICASSP. 2576–2580.
- Christopher W Lynn and Danielle S Bassett. 2019. The physics of brain network structure, function and control. Nature Reviews Physics 1, 5 (2019), 318–332.
- Deep anomaly detection of seizures with paired Stereoelectroencephalography and video recordings. Scientific Reports 11, 1 (2021), 1–11.
- Shuffle and learn: Unsupervised learning using temporal order verification. In ECCV. 527–544.
- Contrastive representation learning for Electroencephalogram classification. In PMLR. 238–253.
- Unified deep supervised domain adaptation and generalization. In ICCV. 5715–5725.
- Representation learning with contrastive predictive coding. arXiv preprint arXiv:1807.03748 (2018).
- 11 - Magnetoelectric composites for medical application. In Composite Magnetoelectrics. Woodhead Publishing, 297–327.
- Pytorch: An imperative style, high-performance deep learning library. In NeurIPS.
- Intracranial electroencephalographic seizure-onset patterns: Effect of underlying pathology. Brain 137, 1 (2014), 183–196.
- Predicting the spatiotemporal diversity of seizure propagation and termination in human focal epilepsy. Nature communications 9, 1 (2018), 1–15.
- Machine learning for predicting epileptic seizures using EEG signals: A review. IEEE Reviews in Biomedical Engineering 14 (2020), 139–155.
- Patrick Schäfer. 2015. The BOSS is concerned with time series classification in the presence of noise. Data Mining and Knowledge Discovery 29, 6 (2015), 1505–1530.
- The Temple University Hospital seizure detection corpus. Frontiers in Neuroinformatics 12 (2018), 83.
- Discrete graph structure learning for forecasting multiple time series. In ICLR.
- Epileptic seizures detection using deep learning techniques: A review. International Journal of Environmental Research and Public Health 18, 11 (2021), 5780.
- EEG emotion recognition using dynamical graph convolutional neural networks. IEEE Transactions on Affective Computing 11, 3 (2020), 532–541.
- Monash university, uea, ucr time series regression archive. arXiv preprint arXiv:2006.10996 (2020).
- Self-supervised graph neural networks for improved Electroencephalographic seizure analysis. In ICLR.
- Attention is all you need. In NeurIPS.
- Connectivity strength-weighted sparse group representation-based brain network construction for M CI classification. Human brain mapping 38, 5 (2017), 2370–2383.
- A multi-view deep learning framework for EEG seizure detection. IEEE Journal of Biomedical and Health Informatics 23, 1 (2019), 83–94.
- TS2Vec: Towards universal representation of time series. arXiv preprint arXiv:2106.10466 (2021).
- Graph Transformer networks. In NeurIPS.
- A Transformer-based framework for multivariate time series representation learning. In KDD. 2114–2124.
- A survey on deep learning-based non-invasive brain signals: Recent advances and new frontiers. Journal of Neural Engineering 18, 3 (2021), 031002.
- Graph-guided network for irregularly sampled multivariate time series. In ICLR.
- Zhe Zhu. 2017. Change detection using landsat time series: A review of frequencies, preprocessing, algorithms, and applications. ISPRS Journal of Photogrammetry and Remote Sensing 130 (2017), 370–384.