Attention-Based Convolutional Recurrent Deep Neural Networks for the Prediction of Response to Repetitive Transcranial Magnetic Stimulation for Major Depressive Disorder

被引:15
|
作者
Shahabi, Mohsen Sadat [1 ]
Shalbaf, Ahmad [1 ]
Nobakhsh, Behrooz [1 ]
Rostami, Reza [2 ]
Kazemi, Reza [3 ]
机构
[1] Shahid Beheshti Univ Med Sci, Sch Med, Dept Biomed Engn & Med Phys, Tehran, Iran
[2] Univ Tehran, Dept Psychol, Tehran, Iran
[3] Inst Cognit Sci Studies, Dept Cognit Psychol, Tehran, Iran
关键词
Transfer learning; convolutional neural networks; long-short term memory; attention mechanism; treatment outcome prediction; repetitive transcranial magnetic stimulation; THETA CONNECTIVITY; BRAIN; EEG; COMPLEXITY; DIAGNOSIS; GRAPH; RTMS;
D O I
10.1142/S0129065723500077
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Repetitive Transcranial Magnetic Stimulation (rTMS) is proposed as an effective treatment for major depressive disorder (MDD). However, because of the suboptimal treatment outcome of rTMS, the prediction of response to this technique is a crucial task. We developed a deep learning (DL) model to classify responders (R) and non-responders (NR). With this aim, we assessed the pre-treatment EEG signal of 34 MDD patients and extracted effective connectivity (EC) among all electrodes in four frequency bands of EEG signal. Two-dimensional EC maps are put together to create a rich connectivity image and a sequence of these images is fed to the DL model. Then, the DL framework was constructed based on transfer learning (TL) models which are pre-trained convolutional neural networks (CNN) named VGG16, Xception, and EfficientNetB0. Then, long short-term memory (LSTM) cells are equipped with an attention mechanism added on top of TL models to fully exploit the spatiotemporal information of EEG signal. Using leave-one subject out cross validation (LOSO CV), Xception-BLSTM-Attention acquired the highest performance with 98.86% of accuracy and 97.73% of specificity. Fusion of these models as an ensemble model based on optimized majority voting gained 99.32% accuracy and 98.34% of specificity. Therefore, the ensemble of TL-LSTM-Attention models can predict accurately the treatment outcome.
引用
收藏
页数:13
相关论文
共 50 条
  • [1] A convolutional recurrent neural network with attention for response prediction to repetitive transcranial magnetic stimulation in major depressive disorder
    Mohsen Sadat Shahabi
    Ahmad Shalbaf
    Reza Rostami
    Reza Kazemi
    Scientific Reports, 13
  • [2] A convolutional recurrent neural network with attention for response prediction to repetitive transcranial magnetic stimulation in major depressive disorder
    Shahabi, Mohsen Sadat
    Shalbaf, Ahmad
    Rostami, Reza
    Kazemi, Reza
    SCIENTIFIC REPORTS, 2023, 13 (01)
  • [3] Prediction of response to repetitive transcranial magnetic stimulation for major depressive disorder using hybrid Convolutional recurrent neural networks and raw Electroencephalogram Signal
    Mohsen Sadat Shahabi
    Ahmad Shalbaf
    Reza Rostami
    Cognitive Neurodynamics, 2023, 17 : 909 - 920
  • [4] Prediction of response to repetitive transcranial magnetic stimulation for major depressive disorder using hybrid Convolutional recurrent neural networks and raw Electroencephalogram Signal
    Shahabi, Mohsen Sadat
    Shalbaf, Ahmad
    Rostami, Reza
    COGNITIVE NEURODYNAMICS, 2023, 17 (04) : 909 - 920
  • [5] Benzodiazepine use and response to repetitive transcranial magnetic stimulation in Major Depressive Disorder
    Fitzgerald, Paul B.
    Daskalakis, Zafiris J.
    Hoy, Kate E.
    BRAIN STIMULATION, 2020, 13 (03) : 694 - 695
  • [6] Left handedness and response to repetitive transcranial magnetic stimulation in major depressive disorder
    Fitzgerald, Paul B.
    Hoy, Kate E.
    Daskalakis, Zafiris J.
    WORLD JOURNAL OF BIOLOGICAL PSYCHIATRY, 2021, 22 (04): : 310 - 314
  • [7] Repetitive transcranial magnetic stimulation for major depressive disorder: A review
    Daskalakis, Z. Jeff
    Levinson, Andrea J.
    Fitzgerald, Paul B.
    CANADIAN JOURNAL OF PSYCHIATRY-REVUE CANADIENNE DE PSYCHIATRIE, 2008, 53 (09): : 555 - 566
  • [8] The effect of repetitive and Deep Transcranial Magnetic Stimulation on quantitative electroencephalography in major depressive disorder
    Ilhan, Reyhan
    Arikan, Mehmet Kemal
    FRONTIERS IN PSYCHIATRY, 2025, 15
  • [9] Cortical Thickness and Response to Repetitive Transcranial Magnetic Stimulation in Youth with Major Depressive Disorder
    McLellan, Quinn
    Kirton, Adam
    Wilkes, T. Christopher
    Schwartz, Kelly Dean
    Ma, Keon
    Jaworska, Natalia
    Langevin, Lisa Marie
    Swansburg, Rose
    Zewdie, Ephrem
    Ciechanski, Patrice
    Damji, Omar
    MacMaster, Frank
    BIOLOGICAL PSYCHIATRY, 2017, 81 (10) : S401 - S401
  • [10] Predictors of response to repetitive transcranial magnetic stimulation (rTMS) in the treatment of major depressive disorder
    Beuzon, G.
    Timour, Q.
    Saoud, M.
    ENCEPHALE-REVUE DE PSYCHIATRIE CLINIQUE BIOLOGIQUE ET THERAPEUTIQUE, 2017, 43 (01): : 3 - 9