Development of Artificial Intelligence for Determining Major Depressive Disorder Based on Resting-State EEG and Single-Pulse Transcranial Magnetic Stimulation-Evoked EEG Indices

Abstract

Depression is the disorder with the greatest socioeconomic burdens. Its diagnosis is still based on an operational diagnosis derived from symptoms, and no objective diagnostic indicators exist. Thus, the present study aimed to develop an artificial intelligence (AI) model to aid in the diagnosis of depression from electroencephalography (EEG) data by applying machine learning to resting-state EEG and transcranial magnetic stimulation (TMS)-evoked EEG acquired from patients with depression and healthy controls. Resting-state EEG and single-pulse TMS-EEG were acquired from 60 patients and 60 healthy controls. Power spectrum analysis, phase synchronization analysis, and phase-amplitude coupling analysis were conducted on EEG data to extract feature candidates to apply different types of machine learning algorithms. Furthermore, to address the limitation of the sample size, dimensionality reduction was performed in a manner to increase the quality of information by featuring robust neurophysiological metrics that showed significant differences between the two groups. Then, nine different machine learning models were applied to the data. For the EEG data, we created models combining four modalities, including (1) resting-state EEG, (2) pre-stimulus TMS-EEG, (3) post-stimulus TMS-EEG, and (4) differences between pre- and post-stimulus TMS-EEG, and evaluated their performance. We found that the best estimation performance (a mean area under the curve of 0.922) was obtained using receiver operating characteristic curve analysis when linear discriminant analysis (LDA) was applied to the combination of the four feature sets. This study showed that by using TMS-EEG neurophysiological indices as features, it is possible to develop a depression decision-support AI algorithm that exhibits high discrimination accuracy.