Related Experiment Video
Updated: Aug 11, 2025
Implementation of a Real-Time Psychosis Risk Detection and Alerting System Based on Electronic Health Records using CogStack
Published on: May 15, 2020
An optimization for postpartum depression risk assessment and preventive intervention strategy based machine learning
Hao Liu1, Anran Dai1, Zhou Zhou2
1School of Basic Medicine and Clinical Pharmacy, China Pharmaceutical University, Nanjing 211198, China; Department of Clinical Pharmacology, Nanjing First Hospital, Nanjing Medical University, Nanjing 210006, China.
Background:
Postpartum depression (PPD) is one of the most common psychiatric disorders for women after delivery. The establishment of an effective PPD prediction model helps to distinguish high-risk groups, and verifying whether such high-risk groups can benefit from drug intervention is very important for clinical guidance.
Methods:
We collected data of parturients that underwent a cesarean delivery. The Control group was divided into a training cohort and a testing cohort. Six different ML models were constructed and we compared their prediction performance in the testing cohort. For model interpretation, we introduced SHapley Additive exPlanations (SHAP). Then, training cohort, ketamine group and dexmedetomidine (DEX) group were classified as high or low risk for PPD by the model. A 1:1 propensity score matching (PSM) was performed to compare the incidence of PPD between two groups in different risk cohorts.
Results:
Extreme gradient enhancement (XGB) had the best recognition effect, with an area under the receiver operating characteristic curve (AUROC) of 0.789 (95 % CI 0.742-0.836) in the training cohort and 0.744 (95 % CI 0.655-0.823) in the testing cohort, respectively. A threshold of 21.5 % PPD risk probability was determined. After PSM, the results showed that the incidence of PPD in the two intervention groups was significantly different from the control group in the high-risk cohort (P < 0.001) but not in the low-risk cohort (P > 0.001).
Conclusion:
Our study demonstrated that the XGB algorithm provided a more accurate in prediction of PPD risk, and it was beneficial to receive early intervention for the high-risk groups distinguished by the model.
