A spatiotemporal fused network considering electrode spatial topology and time-window transition for MDD detection

TL;DR

This paper introduces SET-TIME, a novel spatiotemporal neural network that incorporates electrode spatial topology and time-window transition information to improve EEG-based MDD detection accuracy.

Contribution

The study proposes a new spatiotemporal fused network with domain adaptation for enhanced cross-subject MDD detection from EEG signals.

Findings

Achieved 92% and 94% detection accuracy on two public datasets.

Outperformed state-of-the-art methods in MDD detection.

Validated the effectiveness of multiple modules through ablation experiments.

Abstract

Recently, researchers have begun to experiment with deep learning-based methods for detecting major depressive disor-der (MDD) using electroencephalogram (EEG) signals in search of a more objective means of diagnosis. However, exist-ing spatiotemporal feature extraction methods only consider the functional correlation between multiple electrodes and temporal correlation of EEG signals, ignoring the spatial posi-tion connection information between electrodes and the conti-nuity between time windows, which reduces the model's fea-ture extraction capabilities. To address this issue, a Spatio-temporal fused network for MDD detection with Electrode spatial Topology and adjacent TIME-window transition in-formation (SET-TIME) is proposed in this study. SET-TIME is composed by a common feature extractor, a secondary time-correlation feature extractor, and a domain adaptation (DA) module, in which the former extractor is used to obtain the temporal and spatial features, and the latter extractor can mine the correlation between multiple time windows, and the DA module is adopted to enhance cross-subject detection ca-pability. The experimental results of 10-fold cross-validation show that the proposed SET-TIME method outperforms the state-of-the-art (SOTA) method by achieving MDD detection accuracies of 92.00% and 94.00% on the public datasets PRED+CT and MODMA, respectively. Ablation experiments demonstrate the effectiveness of the multiple modules in SET-TIME, which assist in MDD detection by exploring the intrin-sic spatiotemporal information of EEG signals.