Abstract:Background: Recent evidence suggests that integration of multi-modal data improves performance in machine learning prediction of depression treatment outcomes. Here, we compared the predictive performance of three machine learning classifiers using differing combinations of sociodemographic characteristics, baseline clinical self-reports, cognitive tests, and structural magnetic resonance imaging (MRI) features to predict treatment outcomes in late-life depression (LLD). Methods: Data were combined from two clinical trials conducted with depressed adults aged 60 and older, including response to escitalopram ( N = 32, NCT01902004) and Tai Chi ( N = 35, NCT02460666). Remission was defined as a score of 6 or less on the 24-item Hamilton Rating Scale for Depression (HAMD) at the end of 24 weeks of treatment. Features subsets were constructed from baseline sociodemographic and clinical features, gray matter volumes (GMVs), or both. Three classification algorithms were compared: (1) Support Vector Machine-Radial Bias Function (SVMRBF), (2) Random Forest (RF), and (3) Logistic Regression (LR). A repeated 5-fold cross-validation approach with a wrapper-based feature selection method was used for model fitting. Model performance metrics included Area under the ROC Curve (AUC) and Matthews correlation coefficient (MCC). Cross-validated performance significance was tested by permutation analysis. Classifiers were compared by Cochran's Q and post-hoc pairwise comparisons using McNemar's Chi-Square test with Bonferroni correction. Results: For the RF and SVMRBF algorithms, the combined feature set outperformed the clinical and GMV feature sets with a final cross-validated AUC of 0.83 ± 0.11 and 0.80 ± 0.11, respectively. Both classifiers passed permutation analysis. The LR algorithm performed best using GMV features alone (AUC 0.79 ± 0.14) but failed to pass permutation analysis using any feature set. Performance of the three classifiers differed significantly for all three features sets. Important predictive features of treatment response included anterior and posterior cingulate volumes, depression characteristics, and self-reported health-related quality scores. Conclusion: This preliminary exploration into the use of ML and multi-modal data to identify predictors of general treatment response in LLD indicates that integration of clinical and structural MRI features significantly increases predictive capability. Identified features are among those previously implicated in geriatric depression, encouraging future work in this arena.

Using Machine Learning to Predict Antidepressant Treatment Outcome from Electronic Health Records

Using Electronic Health Records and Machine Learning to Predict Postpartum Depression.

Development and validation of a machine learning algorithm for predicting the risk of postpartum depression among pregnant women

Subphenotyping Depression Using Machine Learning and Electronic Health Records

AI-assisted prediction of differential response to antidepressant classes using electronic health records

Combining machine learning algorithms for prediction of antidepressant treatment response

Machine Learning Prediction of Treatment Outcome in Late-Life Depression

Predicting therapy outcome in a digital mental health intervention for depression and anxiety: A machine learning approach

Machine learning in the prediction of depression treatment outcomes: a systematic review and meta-analysis

Toward personalizing treatment for depression: predicting diagnosis and severity

Improving Cardiovascular Risk Prediction Through Machine Learning Modelling of Irregularly Repeated Electronic Health Records

Prediction and diagnosis of depression using machine learning with electronic health records data: a systematic review

Machine learning to predict clinical remission in depressed patients after acute phase selective serotonin reuptake inhibitor treatment

Prediction of Antidepressant Treatment Response and Remission Using an Ensemble Machine Learning Framework

Ensemble Learning for Early‐Response Prediction of Antidepressant Treatment in Major Depressive Disorder

The promise of machine learning in predicting treatment outcomes in psychiatry

Electroencephalography for predicting antidepressant treatment success: A systematic review and meta-analysis

Efficiently identifying individuals at high risk for treatment resistance in major depressive disorder using electronic health records

Using Machine Learning and Electronic Health Record (EHR) Data for the Early Prediction of Alzheimer's Disease and Related Dementias.

Machine Learning on Early Diagnosis of Depression

Machine learning in major depression: From classification to treatment outcome prediction