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
相关论文
共 68 条
  • [21] Erguzel T.T., 2016, P SAI INT SYST C, P391
  • [22] Neural Network Based Response Prediction of rTMS in Major Depressive Disorder Using QEEG Cordance
    Erguzel, Turker Tekin
    Ozekes, Serhat
    Gultekin, Selahattin
    Tarhan, Nevzat
    Sayar, Gokben Hizli
    Bayram, Ali
    [J]. PSYCHIATRY INVESTIGATION, 2015, 12 (01) : 61 - 65
  • [23] Ant Colony Optimization Based Feature Selection Method for QEEG Data Classification
    Erguzel, Turker Tekin
    Ozekes, Serhat
    Gultekin, Selahattin
    Tarhan, Nevzat
    [J]. PSYCHIATRY INVESTIGATION, 2014, 11 (03) : 243 - 250
  • [24] Motor Intention Decoding from the Upper Limb by Graph Convolutional Network Based on Functional Connectivity
    Feng, Naishi
    Hu, Fo
    Wang, Hong
    Zhou, Bin
    [J]. INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2021, 31 (12)
  • [25] Functional and Effective Connectivity: A Review
    Friston, Karl J.
    [J]. BRAIN CONNECTIVITY, 2011, 1 (01) : 13 - 36
  • [26] Graph theory analysis of directed functional brain networks in major depressive disorder based on EEG signal
    Hasanzadeh, Fatemeh
    Mohebbi, Maryam
    Rostami, Reza
    [J]. JOURNAL OF NEURAL ENGINEERING, 2020, 17 (02)
  • [27] Prediction of rTMS treatment response in major depressive disorder using machine learning techniques and nonlinear features of EEG signal
    Hasanzadeh, Fatemeh
    Mohebbi, Maryam
    Rostami, Reza
    [J]. JOURNAL OF AFFECTIVE DISORDERS, 2019, 256 : 132 - 142
  • [28] A Swallowing Decoder Based on Deep Transfer Learning: AlexNet Classification of the Intracranial Electrocorticogram
    Hashimoto, Hiroaki
    Kameda, Seiji
    Maezawa, Hitoshi
    Oshino, Satoru
    Tani, Naoki
    Khoo, Hui Ming
    Yanagisawa, Takufumi
    Yoshimine, Toshiki
    Kishima, Haruhiko
    Hirata, Masayuki
    [J]. INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2021, 31 (11)
  • [29] Baseline effective connectivity predicts response to repetitive transcranial magnetic stimulation in patients with treatment-resistant depression
    Iwabuchi, S. J.
    Auer, D. P.
    Lankappa, S. T.
    Palaniyappan, L.
    [J]. EUROPEAN NEUROPSYCHOPHARMACOLOGY, 2019, 29 (05) : 681 - 690
  • [30] Detection of Trees on Street-View Images Using a Convolutional Neural Network
    Jodas, Danilo Samuel
    Yojo, Takashi
    Brazolin, Sergio
    Velasco, Giuliana Del Nero
    Papa, Joao Paulo
    [J]. INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2022, 32 (01)