1. Comparative analysis of machine learning versus traditional method for early detection of parental depression symptoms in the NICU.
- Author
-
Sadjadpour F, Hosseinichimeh N, Abedi V, and Soghier LM
- Subjects
- Humans, Female, Male, Infant, Newborn, Adult, Early Diagnosis, Logistic Models, Risk Factors, Machine Learning, Depression diagnosis, Parents psychology, Intensive Care Units, Neonatal
- Abstract
Introduction: Neonatal intensive care unit (NICU) admission is a stressful experience for parents. NICU parents are twice at risk of depression symptoms compared to the general birthing population. Parental mental health problems have harmful long-term effects on both parents and infants. Timely screening and treatment can reduce these negative consequences., Objective: Our objective is to compare the performance of the traditional logistic regression with other machine learning (ML) models in identifying parents who are more likely to have depression symptoms to prioritize screening of at-risk parents. We used data obtained from parents of infants discharged from the NICU at Children's National Hospital ( n = 300) from 2016 to 2017. This dataset includes a comprehensive list of demographic characteristics, depression and stress symptoms, social support, and parent/infant factors., Study Design: Our study design optimized eight ML algorithms - Logistic Regression, Support Vector Machine, Decision Tree, Random Forest, XGBoost, Naïve Bayes, K-Nearest Neighbor, and Artificial Neural Network - to identify the main risk factors associated with parental depression. We compared models based on the area under the receiver operating characteristic curve (AUC), positive predicted value (PPV), sensitivity, and F-score., Results: The results showed that all eight models achieved an AUC above 0.8, suggesting that the logistic regression-based model's performance is comparable to other common ML models., Conclusion: Logistic regression is effective in identifying parents at risk of depression for targeted screening with a performance comparable to common ML-based models., Competing Interests: The authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest., (Copyright © 2024 Sadjadpour, Hosseinichimeh, Abedi and Soghier.)
- Published
- 2024
- Full Text
- View/download PDF