Exploring machine learning to predict depressive relapses of bipolar disorder patients

• Published in: Journal of affective disorders Vol. 295; pp. 681 - 687
• Main Authors: Rotenberg, Luisa de Siqueira, Borges-Júnior, Renato Gomes, Lafer, Beny, Salvini, Rogerio, Dias, Rodrigo da Silva
• Format: Journal Article
• Language: English
• Published: Elsevier B.V 01.12.2021
• Subjects: Artificial intelligence; Bipolar disorder; Depressive relapse; Machine learning; Mental health; Depressive relapse; Bipolar disorder; Artificial intelligence; Mental health; Machine learning
• ISSN: 0165-0327; 1573-2517
• DOI: 10.1016/j.jad.2021.08.127

•Machine learning models were able to classify and predict depressive relapse, therefore, being able to accurately predict whether a patient would have a relapse four visits prior to the actual relapse visit.•Our results support the capability of precision medicine and the use of machine learning algorithms to predict relapse in earlier visits with a reasonable performance.•Intelligent algorithms may be a valuable and efficient strategy to further support medical decision making and as a consequence improve the long-term prognosis of BD. Bipolar disorder (BD) is a chronic mood disorder characterized by recurrent episodes of mania or hypomania and depression, expressed by changes in energy levels and behavior. However, most of relapse studies use evidence-based approaches with statistical methods. With the advance of the precision medicine this study aims to use machine learning (ML) approaches as a possible predictor in depressive relapses in BD. Four accepted and well used ML algorithms (Support Vector Machines, Random Forests, Naïve Bayes, and Multilayer Perceptron) were applied to the Systematic Treatment Enhancement Program for Bipolar Disorder (STEP-BD) dataset in a cohort of 800 patients (507 patients presented depressive relapse and 293 did not), who became euthymic during the study and were followed for one year. The ML algorithms presented reasonable performance in the prediction task, ranging from 61 to 80% in the F-measure. The Random Forest algorithm obtained a higher average of performance (Relapse Group 68%; No Relapse Group 74%). The three most important mood symptoms observed in the relapse visit (Random Forest) were: interest; depression mood and energy. Social and psychological parameters such as marital status, social support system, personality traits, might be an important predictor in depressive relapses, although we did not compute this data in our study. Our findings indicate that applying precision medicine models by means of machine learning in BD studies could be feasible as a sensible approach to better support medical decision-making in the BD treatment and prevention of future relapses.