Development and Validation of a Deep-Learning Model for Differential Treatment Benefit Prediction for Adults with Major Depressive Disorder Deployed in the Artificial Intelligence in Depression Medication Enhancement (AIDME) Study

๐Ÿ“… 2024-06-07
๐Ÿ›๏ธ arXiv.org
๐Ÿ“ˆ Citations: 2
โœจ Influential: 0
๐Ÿ“„ PDF
๐Ÿค– AI Summary
Current pharmacotherapy for major depressive disorder (MDD) in adults relies heavily on a trial-and-error approach, leading to delayed remission and suboptimal outcomes. Method: We developed a clinical-grade deep learning model that individually predicts remission probability across ten first-line antidepressants. Trained on harmonized, multicenter randomized clinical trial data, the model integrates 25 clinical and demographic features and employs Bayesian optimization to enhance generalizability. Contribution/Results: To our knowledge, this is the first AI tool enabling concurrent comparative efficacy prediction across ten antidepressants. It achieves an AUC of 0.65 (p = 0.01) on an independent test setโ€”significantly outperforming conventional baselines. Both simulation and real-world validation demonstrate improved population-level remission rates. The model has undergone bias-aware evaluation, exhibits strong real-world deployability, and is currently being evaluated in the AIDME randomized controlled trial to advance precision psychiatry in MDD treatment.

Technology Category

Application Category

๐Ÿ“ Abstract
INTRODUCTION: The pharmacological treatment of Major Depressive Disorder (MDD) relies on a trial-and-error approach. We introduce an artificial intelligence (AI) model aiming to personalize treatment and improve outcomes, which was deployed in the Artificial Intelligence in Depression Medication Enhancement (AIDME) Study. OBJECTIVES: 1) Develop a model capable of predicting probabilities of remission across multiple pharmacological treatments for adults with at least moderate major depression. 2) Validate model predictions and examine them for amplification of harmful biases. METHODS: Data from previous clinical trials of antidepressant medications were standardized into a common framework and included 9,042 adults with moderate to severe major depression. Feature selection retained 25 clinical and demographic variables. Using Bayesian optimization, a deep learning model was trained on the training set, refined using the validation set, and tested once on the held-out test set. RESULTS: In the evaluation on the held-out test set, the model demonstrated achieved an AUC of 0.65. The model outperformed a null model on the test set (p = 0.01). The model demonstrated clinical utility, achieving an absolute improvement in population remission rate in hypothetical and actual improvement testing. While the model did identify one drug (escitalopram) as generally outperforming the other drugs (consistent with the input data), there was otherwise significant variation in drug rankings. On bias testing, the model did not amplify potentially harmful biases. CONCLUSIONS: We demonstrate the first model capable of predicting outcomes for 10 different treatment options for patients with MDD, intended to be used at or near the start of treatment to personalize treatment. The model was put into clinical practice during the AIDME randomized controlled trial whose results are reported separately.
Problem

Research questions and friction points this paper is trying to address.

Predict remission probabilities for MDD treatments
Validate model accuracy and bias mitigation
Personalize depression medication using AI
Innovation

Methods, ideas, or system contributions that make the work stand out.

Deep-learning model predicts depression treatment outcomes
Bayesian optimization for personalized medication selection
Clinical utility tested in AIDME randomized trial
๐Ÿ”Ž Similar Papers
No similar papers found.
D
D. Benrimoh
McGill University, Department of Psychiatry
C
C. Armstrong
Aifred Health, Montreal, Canada
J
J. Mehltretter
Aifred Health, Montreal, Canada
R
R. Fratila
Aifred Health, Montreal, Canada
K
K. Perlman
McGill University, Department of Psychiatry, Aifred Health, Montreal, Canada
S
S. Israel
Aifred Health, Montreal, Canada
A
A. Kapelner
Department of Mathematics, Queens College, CUNY, Queens, NY
S
Sagar V Parikh
University of Michigan, Ann Arbor
J
J. F. Karp
Department of Psychiatry, College of Medicine-Tucson, University of Arizona. Tucson, AZ
K
K. Heller
Google, Mountain View, California
G
G. Turecki
McGill University, Department of Psychiatry