Development and external validation of a diagnostic model for differentiating major depressive disorder from bipolar disorder
摘要
Misdiagnosing bipolar disorder (BD) as major depressive disorder (MDD) can lead to poor treatment outcomes. This study aims to develop and validate a machine learning-based model to effectively differentiate between MDD and BD using electronic medical record (EMR) data.
MethodsThis retrospective study enrolled 584 patients with BD and 1,179 patients with MDD from two medical centers between January 2022 and August 2024. Feature selection was performed using both Least Absolute Shrinkage and Selection Operator (LASSO) regression and the Boruta algorithm. Six machine learning (ML) algorithms were used to construct the model. SHapley Additive exPlanations (SHAP) analysis was conducted to improve model interpretability.
ResultsAmong the six machine learning models constructed based on these features, the RF model demonstrated the best overall performance, achieving the highest (AUC = 0.863) in the internal validation set and also showing moderate discriminative ability (AUC = 0.710) in the external validation set. SHAP analysis identified illness duration, creatine kinase, and age of onset as the most important predictive features.
ConclusionsThe 7-predictor RF model demonstrated moderate discriminative performance, showing potential as an auxiliary decision support tool for distinguishing hospitalized BD from MDD patients. However, the reliance on the International Classification of Diseases, 10th Revision (ICD-10) criteria may result in the misclassification of latent BD, thereby limiting the model’s accuracy and generalizability. Furthermore, future work should validate the model’s generalization ability in multi-center samples and develop intuitive decision support tools to enhance clinical utility.
Clinical trial registrationNot applicable. This is a retrospective observational study that does not involve any clinical intervention; therefore, clinical trial registration was not required.