Skip to main content
JMIR Medical Informatics logoLink to JMIR Medical Informatics
. 2020 Apr 30;8(4):e15516. doi: 10.2196/15516

Machine Learning Models for the Prediction of Postpartum Depression: Application and Comparison Based on a Cohort Study

Weina Zhang 1, Han Liu 2, Vincent Michael Bernard Silenzio 3, Peiyuan Qiu 4, Wenjie Gong 1,✉,#
Editor: Christian Lovis
Reviewed by: Haomiao Jin, Mina Bjelogrlic, Balaji Polepalli Ramesh
PMCID: PMC7226048  PMID: 32352387

Abstract

Background

Postpartum depression (PPD) is a serious public health problem. Building a predictive model for PPD using data during pregnancy can facilitate earlier identification and intervention.

Objective

The aims of this study are to compare the effects of four different machine learning models using data during pregnancy to predict PPD and explore which factors in the model are the most important for PPD prediction.

Methods

Information on the pregnancy period from a cohort of 508 women, including demographics, social environmental factors, and mental health, was used as predictors in the models. The Edinburgh Postnatal Depression Scale score within 42 days after delivery was used as the outcome indicator. Using two feature selection methods (expert consultation and random forest-based filter feature selection [FFS-RF]) and two algorithms (support vector machine [SVM] and random forest [RF]), we developed four different machine learning PPD prediction models and compared their prediction effects.

Results

There was no significant difference in the effectiveness of the two feature selection methods in terms of model prediction performance, but 10 fewer factors were selected with the FFS-RF than with the expert consultation method. The model based on SVM and FFS-RF had the best prediction effects (sensitivity=0.69, area under the curve=0.78). In the feature importance ranking output by the RF algorithm, psychological elasticity, depression during the third trimester, and income level were the most important predictors.

Conclusions

In contrast to the expert consultation method, FFS-RF was important in dimension reduction. When the sample size is small, the SVM algorithm is suitable for predicting PPD. In the prevention of PPD, more attention should be paid to the psychological resilience of mothers.

Keywords: depression, postpartum, machine learning, support vector machine, random forest, prediction model

Introduction

Postpartum depression (PPD) is a serious public health problem that affects 10% to 20% of pregnant women [1-3]. PPD not only adversely affects the physical and mental health of mothers, it is detrimental to the growth and development of infants. In extreme cases even suicide and infanticide may occur [4]. Establishing an effective PPD prediction model that can be used in pregnancy may enable earlier identification, thus, helping health care providers offer more effective management to at-risk patients [5]. Previous studies have explored this possibility and demonstrated its feasibility [6,7].

Machine learning (ML) may be useful in making accurate predictions based on data from multiple sources and has been applied in prediction studies in recent years [8]. There are many predictive factors for PPD including demographics, psychology, and environment [5,9,10]. Assessing risk factors during pregnancy can allow enough time for subsequent interventions. The expert consultation method has often been used to generate guidelines for PPD detection, based on expert opinion and clinical experience. In contrast, ML approaches rely on the use of empirical data to generate prediction models. The key to building good ML models is in the rigorous selection of appropriate features and algorithms. There are two approaches to address the important challenge of feature selection in ML: filter and wrapper [11]. A random forest-based filter feature selection (FFS-RF) algorithm can use the importance score of a so-called random forest (RF) of variables as the evaluation criterion for feature selection, which will identify the subsets of data features that may be most relevant to accurately predict the targeted outcome variable(s) of interest. Such strategies to identify the most relevant data features have proven to be effective ways to explore the risk factors for some diseases [12]. There are two main algorithms used in depression prediction studies, namely, the support vector machine (SVM) and RF algorithms [8]. Depression prediction studies using these two methods have achieved relatively good results [13-15]. SVM is an example of supervised learning. It focuses on minimizing structural risks within the set of available data [16]. It has great advantages in solving high-dimensional modeling problems and performs well in situations that have relatively less available sample data [17]. In contrast, RF models are built using a decision tree as the basic classifier. RF approaches have high classification accuracy, strong inductive capacity, a simple parameter adjustment process, fast calculation speed, relatively low sensitivity to missing data values, and the ability to output feature importance [12,18].

Comparison between those ML methods concerning PPD has not been studied. This study is based on data drawn from a large, ongoing cohort study of pregnant women in the Hunan province of south central China. In this paper we combined the two feature selection methods and the two ML algorithms described above to assess four PPD prediction models using data during pregnancy to compare the effect of PPD prediction models, pick the optimal predictive model, and provide a reference for the development of ML in PPD.

Methods

Sampling

This study was part of a larger cohort study. All the data included here is original and previously unpublished. Researchers in the study collected the following measures at a series of 7 visits conducted in the first trimester through 6 weeks postpartum: depression (using the Edinburgh Postnatal Depression Scale [EPDS]), social environment, and psychological and biological factors associated with depression. The study was approved by the institutional review board of the institute of clinical pharmacology of Central South University (ChiCTR-ROC-16009255).

Participants were recruited from two maternity and child care centers in the cities of Changsha and Yiyang in the Hunan province. The former is a major provincial teaching hospital located in Changsha, a city with approximately 8.15 million residents. Yiyang city is a less economically developed area of Hunan province, with approximately 4.39 million residents. Researchers sought to recruit women in the obstetric clinics of the two hospitals from September 2016, to February 2017. The following inclusion criteria were used for participants: woman, age ≥18 years, and gestation period ≤13 weeks (pregnancy weeks are estimated based on the first day of the last menstrual period). All participants signed informed consent. In total, 1126 women were recruited.

Measures

The following tools were used to collect data.

  1. A purpose-built questionnaire, designed for this study and optimized through a pilot survey, was used to collect information including age, education, monthly income level, occupation, marital satisfaction, first pregnancy, folic acid intake, premenstrual syndrome, history of mental health concerns, family history of mental illness, mother's menopausal symptoms, childhood experiences, and life events.

  2. The EPDS was used to self-report maternal symptoms of depression [19]. The EPDS is a 10-item self-rated questionnaire, with each item scored from 0 to 3, with a total score ranging from 0 to 30. The Chinese language EPDS used in this study was translated by Wang Yuqiong [20]. The EPDS is the most common PPD screening tool [21,22]. The critical value was 9.5.

  3. The Brief Resilience Scale (BRS) was used to determine the level of psychological resilience. The BRS is a 6-item questionnaire that reflects the respondent’s ability to bounce back or recover from stress. The score is the average score of each item. A higher score indicates a stronger strain and adaptability [23].

  4. The Pittsburgh Sleep Quality Index (PSQI) is a comprehensive scale that reflects the sleep quality of subjects. It is composed of 7 dimensions: “Sleep Quality”, “Sleep Latency”, “Sleep Duration”, “Sleep Efficiency”, “Sleep Disorders”, “Use of Sleep Medications”, and “Daytime Dysfunction”. The scores of each dimension are summed to obtain the total PSQI score. Higher scores indicate worse sleep quality. According to the total score, sleep quality can be divided into different grades: 6 to 10 indicates “good sleep quality”, 11 to 15 indicates “average sleep quality”, and 16 to 21 indicates “poor sleep quality” [24]. The scale has good reliability and validity [25].

  5. The Social Support Rating Scale (SSRS), which was designed by Shuiyuan Xiao [26], was used to measure social support. The SSRS is a 10-item questionnaire with three dimensions, namely, objective support, subjective support, and use of social support. Higher total scores and higher scores for each dimension indicates a better level of social support for an individual.

  6. The Generalized Anxiety Disorder-7 (GAD-7) was developed by Spitzer [27]. The score is obtained by summing the scores of 7 items. Most current studies consider a total score of 10 or higher as indicative of anxiety [27,28].

Procedure

Seven time points were selected for depression screening, corresponding to the women’s routine obstetric examinations. We divided these into first trimester (gestational week 13 or earlier), second trimester (weeks 17-20 and 21-24), third trimester (weeks 31-32 and 35-40) and postpartum (7 days and 6 weeks postpartum). Except for the first, screening for perinatal depression by EPDS was performed twice for each trimester. If one or more of the EPDS scores was 9.5 or higher for each grouped set of visits, the participant was regarded as at risk for depression during this period. The study questionnaire, BRS, and GAD-7 were assessed during the first trimester, whereas the PSQI was used during the second trimester, and the SSRS during the third trimester. In total, 508 out of 1126 (45.12%) participants completed all screenings (Figure 1).

Figure 1.

Figure 1

Participant recruitment and response condition.

Feature Selection

Two simple and easy to implement methods were used for feature selection, namely, the expert consultation and FFS-RF methods. The expert consultation method was used to select clinically relevant factors as appropriate predictors of pre-existing or potential PPD. This was accomplished by consulting experts in the area of obstetrics and gynecology as well as mental health practitioners. The FFS-RF was used to identify proper predictors for PPD. Under this approach, features within a certain bound value range (P>.05) were selected as potential predictors and incorporated into the final prediction model.

Model Development

Of the 508 participants, 75% (381) were randomly selected for model training. Data from the remaining 127 participants was held back for use in model testing and verification. Table 1 shows the model selection scheme. Based on the expert consultation method and FFS-RF method, four PPD prediction models were generated using the SVM and RF algorithms. The parameters of the models were optimized, and the specific parameters are shown in Table 2.

Table 1.

Names of the postpartum depression prediction models.

Machine learning modeling algorithm Feature selection method

Expert consultation method FFS-RFa
Random forest E-RFb F-RFc
Support vector machine E-SVMd F-SVMe

aFFS-RF: filter feature selection based on random forest.

bE-RF: model built using the random forest algorithm and expert consultation method.

cF-RF: model built using the random forest algorithm and Random forest-based filter feature selection method.

dE-SVM: model built using the support vector machine algorithm and expert consultation method.

eF-SVM: model built using the support vector machine algorithm and Random forest-based filter feature selection method.

Table 2.

Optimal parameters for each model.

PPDa prediction model name Parameter settings
E-RFb n_estimator=300, criterion=entropy, max_features=sqrt
E-SVMc Kernel=linear
F-RFd n_estimator=300, max_features=auto, criterion=gini
F-SVMe Kernel=linear

aPPD: postpartum depression.

bE-RF: model built using the random forest algorithm and expert consultation method.

cE-SVM: model built using the support vector machine algorithm and expert consultation method.

dF-RF: model built using the random forest algorithm and Random forest-based filter feature selection method.

eF-SVM: model built using the support vector machine algorithm and Random forest-based filter feature section method.

Evaluation of Model Effects

For the test set, we used the trained models to test and compare their prediction of PPD with real data and created a confusion matrix (Table 3). A series of indicators were obtained of each model. The following index formulas were used.

Table 3.

Confusion matrix.

Predicted Results Real Results

Positive Negative
Positive a c
Negative b d

Accuracy = Inline graphic

Misclassification rate = Inline graphic

Positive predictive value = Inline graphic

Negative predictive value = Inline graphic

Sensitivity (Sen) = Inline graphic

Specificity (Spe) = Inline graphic

Geometric mean = Inline graphic

The sensitivity and the receiver operator curve-area under the curve (ROC-AUC) were used to evaluate the effects of each model and choose the best prediction model. To select the optimal model, we first selected the model with an ROC-AUC>0.75 to confirm that it had a good comprehensive prediction effect. On this basis, we then selected the model with the highest sensitivity as the best prediction model, thus, ensuring that as many mothers as possible with a high risk of PPD would be detected.

Statistical Analysis

This study used the REDCap system to build a database and SPSS version 18.0 to clean the data. The training and test sets were analyzed by the “sklearn.model_selection.train_test_split” package. The RF data were analyzed by the “sklearn.ensemble.randomforestclassifiers” package. The SVM data were analyzed by the “sklearn.svm.SVC” package. Cross-validation was performed using the “sklearn.cross_validation” package. All these packages were available in the Python 3.6 software.

Results

Candidate Predictors

Multimedia Appendix 1 shows the 25 candidate predictors of the subjects with and without PPD. Among the 508 subjects, 173 (34.1%) were regarded as having PPD. The average age of the pregnant women was 28.64 years (SD 4.344). The average BRS score was 3.10 (SD 0.371). The average individual monthly income of the women and their spouses was between 2000 and 5000 yuan (US $393-785). Most of the subjects had a bachelor's degree. Of the 173 women with PPD, 116 (67.1%) had positive EPDS screening results in the third trimester. Multimedia Appendix 1 shows the results of the single-factor analysis (P<.05).

Feature Selection

The predictive features obtained by the expert consultation and FFS-RF methods are shown in Textbox 1. This study included a total of 25 features: 17 were selected as predictive characteristics by expert consultation method and 7 were selected by FFS-RF.

Selected features of the two methods of feature selection in descending order.

Expert consultation method

  • Age

  • Education

  • Monthly income level

  • Husband’s education

  • Husband’s monthly income level

  • Marital satisfaction

  • Sexual, psychological, or physical spousal abuse

  • Childhood abuse history

  • Premenstrual syndrome-mood instability

  • Premenstrual syndrome-sleep changes

  • Depression history of woman

  • Depression history of family members

  • Other mental illness history of woman

  • Other mental illness history of family members

  • Mother’s menopausal symptoms

  • Level of psychological resilience

  • Depressive symptoms in the third trimester

Random forest-based filter feature selection

  • Level of psychological resilience

  • Depressive symptoms in first trimester

  • Monthly income level

  • Husband’s monthly income level

  • Husband’s education

  • Education

  • Mother’s menopausal symptoms

Model Effects

PPD prediction models were established using the RF and SVM modeling applied to the training data set, using the feature sets constructed through our two feature selection methods. The optimal parameters of each model are shown in Table 2. After five-fold cross-validation, we found that when n_estimator=200, max_features=sqrt, and criterion=entropy, the model built using the RF algorithm and expert consultation method (E-RF) had the best sensitivity. When n_estimator=200, criterion=gini, and max_features=auto, the model built using the RF algorithm and FFS-RF method (F-RF) had the best sensitivity. Therefore, the software default setting was max_features=auto. With the SVM algorithm, regardless of the feature selection strategy, the kernel function with the highest model sensitivity was a linear kernel function.

The model evaluation index is shown in Table 4, and the ROC curves for the four PPD models are shown in Figures 2-5. The SVM models had a slightly lower classification rate as well as a significantly higher sensitivity than the RF models. No significant differences in the specificity of each prediction model were observed. Both the positive predictive and negative predictive values of the SVM models were significantly higher than those of the RF models. With regard to feature selection, the geometric mean value for the expert consultation method was slightly higher than that of the FFS-RF. The ROC-AUC value under the SVM was slightly higher than under the RF. In summary, among the four models tested, F-SVM was the optimal model.

Table 4.

Test data sets for each model evaluation index.

Items E-RFa E-SVMb F-RFc F-SVMd
Misclassification rate 0.28 0.20 0.27 0.22
Sensitivity 0.48 0.68 0.48 0.69
Specificity 0.86 0.87 0.86 0.83
Positive predictive value 0.63 0.72 0.63 0.68
Negative predictive value 0.76 0.84 0.76 0.84
Geometric mean 0.84 0.76 0.64 0.76
ROC-AUCe 0.75 0.81 0.70 0.78

aE-RF: model built using the random algorithm and expert consultation method.

bE-SVM: model built using the support vector machine algorithm and expert consultation method.

cF-RF: model built using the random forest algorithm and random forest-based filter feature selection method.

dF-SVM: model built using the support vector machine algorithm and Random forest-based filter feature selection method.

eROC-AUC: receiver operating characteristic curve-area under the curve.

Figure 2.

Figure 2

The receiver operating characteristic curve of E-RF. AUC: area under the curve; ROC: receiver operating characteristic.

Figure 5.

Figure 5

The receiver operating characteristic curve of F-SVM. AUC: area under the curve; ROC: receiver operating characteristic curve.

Figure 3.

Figure 3

The receiver operating characteristic curve of E-SVM. AUC: area under the curve; ROC: receiver operating characteristic.

Figure 4.

Figure 4

The receiver operating characteristic curve of F-RF. AUC: area under the curve; ROC: receiver operating characteristic.

The features selected by the expert consultation method and FFS-RF method were put into the E-RF and F-RF models, respectively. The importance of the features was ranked as shown in Figure 6. The importance of mental elasticity in the model is significantly higher than other factors. Symptoms of depression in late pregnancy was the second most important predictor. Income levels were also important predictors of PPD. There was no significant difference in the importance of each factor to PPD. The top most important features in these two models are shown in Textbox 2.

Figure 6.

Figure 6

The relative feature importance rankings of the E-RF and the F-RF based on the two feature selection methods.

Top features according to the E-RF and F-RF in descending order.

Model built using the random forest algorithm and expert consultation method

  1. Level of psychological resilience

  2. Depressive symptoms in the third trimester

  3. Monthly income level

  4. Husband’s education

  5. Education

  6. Husband’s monthly income level

  7. Mother’s menopausal symptoms

  8. Premenstrual syndrome-mood instability

  9. Marital satisfaction

  10. Age

Model built using the random forest algorithm and random forest-based filter feature selection method

  1. Level of psychological resilience

  2. Depressive symptoms in early pregnancy

  3. Monthly income level

  4. Husband’s monthly income level

  5. Husband’s education

  6. Education

  7. Mother’s menopausal symptoms

Discussion

We compared four PPD prediction models and provided a reference for the application of ML in PPD. Compared with the expert consultation method approach, the FFS-RF method identified fewer predictive factors. We found that the F-SVM model was the best model. The strongest predictive factor was the psychological resilience of pregnant women.

Between the expert consultation method and FFS-RF method, the latter selected far fewer predictive factors. Furthermore, there was no significant difference between the two methods in terms of their effects on model performance, indicating that the FFS-RS method could reduce dimensions and improve the efficiency of the algorithmic function without changing model predictive performance. The reduction in the number of predictive factors means that the burden of collecting information is reduced, making the model easier to implement and popularize, especially in busy obstetric clinics.

The SVM was chosen as the better algorithm, as it showed higher sensitivity than the RF algorithm (E-SVM=0.67, F-SVM=0.69, E-RF=0.48, F-RF=0.48). SVM had a clear advantage over RF in processing our research data, and the smaller sample size may be the main reason for this finding. Previous research on depression suggested that sample size is a key factor affecting the performance of ML models. When the sample size is small, SVM can avoid overfitting while providing efficient computing time and produces better prediction results in depression [29,30]. Our results also support this view. Therefore, we believe that when the data set is small, SVM is more practical than RF in prediction research for PPD. Several previous studies used the SVM algorithm to make PPD predictions. Jiménez [13] collected data on postpartum women from seven Spanish hospitals and used the EPDS score as the outcome indicator to train a PPD prediction model based on SVM. Sriraam [15] used social media as a data source and, based on the mental health data of 173 mothers, a SVM-based PPD prediction model was established. De Choudhury [31] developed a SVM model to identify high-risk emotions and behaviors predictive of PPD using the content of Twitter posts. As these studies either target different populations or use different methods to detect the occurrence of PPD, the model prediction effects cannot be easily compared. However, the results of the optimal F-SVM model in our study are within range (sensitivity=0.69, ROC-AUC=0.78) and consistent with the findings of previous studies (sensitivity=0.56-0.78, ROC-AUC=0.63-0.81) [13,15,31]. Due to the negative effects of PPD on mothers and infants [32,33], such as the negative effects on the physical and mental health of mothers, the closeness of the mother-infant bond, and infant development, it is important to have a model with high sensitivity while maintaining a high ROC-AUC value. The selection of indicators in evaluating depression prediction models varies across studies. For example, Sriraam [15] and De Choudhury [31] emphasized the accuracy of the model's prediction of PPD. Jiménez [13] emphasized model sensitivity and specificity. The balance between them is the geometric mean. The ROC-AUC is also widely used to evaluate the comprehensive performance of a model [14,15]. Our evaluation criteria provide a reference for prediction research for screening purposes, but the approach may be different in research studies.

We found that the top 3 most important predictors in the models were psychological resilience, depression during the third trimester, and monthly income level. First, psychological resilience is the most important factor in the prediction of PPD, which can be attributed to the protective effect of psychological elasticity. Pregnancy and childbirth are a challenging time for women emotionally and physiologically, and the mother's body and mind are under greater stress [15]. Previous research has shown that psychological resilience as an important regulatory process can enable people to recover from and adapt to stress and life events, reducing the occurrence of adverse outcomes [34-36]. Our results also support the findings of Lu [37], who found that the level of psychological elasticity was negatively correlated with the occurrence of PPD. Second, the results regarding depression in the third trimester are consistent with most previous studies. Depression in the third trimester is associated with PPD [9,38,39]. A review by Robertson [5] mentioned that “depression and anxiety during pregnancy are the strongest predictors of PPD”. Mora's [40] research suggests that depression in the third trimester may continue to develop into the postpartum period. Third, the monthly income levels remain important factors affecting PPD, which supports Rhonda's [41] findings that mothers with low income levels faced obstacles in using mental health resources and were more likely to be frustrated. Epidemiological studies of PPD worldwide have also found that the incidence in developing countries is higher than that in developed countries [42].

The identification of these predictors also reveals the different aspects of PPD risk factors. A pregnant woman's psychological elasticity may reflect her personality traits. Depression in the third trimester may be a special symptom accompanying pregnancy. The income of a pregnant woman and her partner reflects the stability and coping resources available to them. It indicates that PPD risk should be assessed based on a combination of individual long-term, short-term, and environmental characteristics.

This study has several limitations. First, there was potential selection bias. Women who were not lost to follow-up might have had a greater awareness of mental health services. Second, the 50% loss to follow-up and small sample size may have negatively affected the applicability of the PPD model, indicating that more extensive validation is required. Third, a larger number of potential predictive factors would have been useful. Further studies should develop different PPD models using other ML algorithms and data from different sources as well as incorporating additional cultural factors to expand the application of the PPD models.

Acknowledgments

We acknowledge the people who have contributed to the field of this study, including professor KK Cheng from University of Birmingham, Liu Lu from Central South University. This project is funded by the National Natural Science Foundation of China (Grant No 81402690, 81773446), the National Natural Science Foundation of Hunan Province (Grant No 2019JJ40351), and the Graduate Research and Innovation Project of Central South University (Grant No 1053320183626).

Abbreviations

BRS

Brief Resilience Scale

E-RF

model built using the random forest algorithm and expert consultation method

E-SVM

model built using the support vector machine algorithm and expert consultation method

EPDS

Edinburgh Postnatal Depression Scale

F-RF

model built using the random forest algorithm and Random forest-based filter feature selection method

F-SVM

model built using the support vector machine algorithm and Random forest-based filter feature selection method

FFS-RF

random forest-based filter feature selection

GAD-7

Generalized Anxiety Disorder-7

ML

machine learning

PPD

postpartum depression

PSQI

Pittsburgh Sleep Quality Index

RF

random forest

ROC-AUC

receiver operator curve-area under the curve

SSRS

Social Support Rating Scale

SVM

support vector machine.

Appendix

Multimedia Appendix 1

Comparison of candidate predictors in the sample of pregnant women (N=508).

Multimedia Appendix 2

Comparison of demographic characteristics, including data sets of 618 pregnant women lost in the cohort and 508 mothers who left the cohort study after childbirth.

Multimedia Appendix 3

Definitions and coding of analyzed variables.

Footnotes

Authors' Contributions: WZ, as the first author, developed the initial manuscript. She helped with recruitment of the participants and collected the data. Authors WZ and HL performed the statistical analysis. Authors HL and VS contributed substantially to the revision and refinement of the final manuscript study. Authors WG and PQ guided the overall design of the study and supervised the model development and manuscript. WG and PQ contributed equally to this paper.

Conflicts of Interest: None declared.

References

  • 1.VanderKruik R, Barreix M, Chou D, Allen T, Say L, Cohen LS, Maternal Morbidity Working Group The global prevalence of postpartum psychosis: a systematic review. BMC Psychiatry. 2017 Dec 28;17(1):272. doi: 10.1186/s12888-017-1427-7. https://bmcpsychiatry.biomedcentral.com/articles/10.1186/s12888-017-1427-7. [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 2.Gavin NI, Gaynes BN, Lohr KN, Meltzer-Brody S, Gartlehner G, Swinson T. Perinatal depression: a systematic review of prevalence and incidence. Obstet Gynecol. 2005 Nov;106(5 Pt 1):1071–83. doi: 10.1097/01.AOG.0000183597.31630.db. [DOI] [PubMed] [Google Scholar]
  • 3.Fisher J, Cabral de Mello M, Patel V, Rahman A, Tran T, Holton S, Holmes W. Prevalence and determinants of common perinatal mental disorders in women in low- and lower-middle-income countries: a systematic review. Bull World Health Organ. 2012 Feb 01;90(2):139G–149G. doi: 10.2471/BLT.11.091850. http://europepmc.org/abstract/MED/22423165. [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 4.Muzik M, Borovska S. Perinatal depression: implications for child mental health. Ment Health Fam Med. 2010 Dec;7(4):239–47. http://europepmc.org/abstract/MED/22477948. [PMC free article] [PubMed] [Google Scholar]
  • 5.Robertson E, Grace S, Wallington T, Stewart DE. Antenatal risk factors for postpartum depression: a synthesis of recent literature. Gen Hosp Psychiatry. 2004;26(4):289–95. doi: 10.1016/j.genhosppsych.2004.02.006. [DOI] [PubMed] [Google Scholar]
  • 6.Righetti-Veltema M, Conne-Perréard E, Bousquet A, Manzano J. Risk factors and predictive signs of postpartum depression. J Affect Disord. 1998 Jun;49(3):167–80. doi: 10.1016/s0165-0327(97)00110-9. [DOI] [PubMed] [Google Scholar]
  • 7.Fergusson DM, Horwood LJ, Thorpe K. Changes in depression during and following pregnancy. ALSPAC study team. Study of pregnancy and children. Paediatr Perinat Epidemiol. 1996 Jul;10(3):279–93. doi: 10.1111/j.1365-3016.1996.tb00051.x. [DOI] [PubMed] [Google Scholar]
  • 8.Shatte ABR, Hutchinson DM, Teague SJ. Machine learning in mental health: a scoping review of methods and applications. Psychol Med. 2019 Jul;49(9):1426–1448. doi: 10.1017/S0033291719000151. [DOI] [PubMed] [Google Scholar]
  • 9.Beck CT. Predictors of postpartum depression: an update. Nurs Res. 2001;50(5):275–85. doi: 10.1097/00006199-200109000-00004. [DOI] [PubMed] [Google Scholar]
  • 10.Johnstone SJ, Boyce PM, Hickey AR, Morris-Yatees AD, Harris MG. Obstetric risk factors for postnatal depression in urban and rural community samples. Aust N Z J Psychiatry. 2001 Feb;35(1):69–74. doi: 10.1046/j.1440-1614.2001.00862.x. [DOI] [PubMed] [Google Scholar]
  • 11.Peters J, Verhoest NE, Samson R, Van Meirvenne M, Cockx L, De Baets B. Uncertainty propagation in vegetation distribution models based on ensemble classifiers. Ecological Modelling. 2009 Mar;220(6):791–804. doi: 10.1016/j.ecolmodel.2008.12.022. [DOI] [Google Scholar]
  • 12.Yao Dengju, Yang Jing. Research on feature selection and classification method based on random forest for medical datasets. Harbin Engineering University. 2017 May 23; http://cdmd.cnki.com.cn/Article/CDMD-10217-1017245762.htm. [Google Scholar]
  • 13.Jiménez-Serrano S, Tortajada S, García-Gómez JM. A mobile health application to predict postpartum depression based on machine learning. Telemed J E Health. 2015 Jul;21(7):567–74. doi: 10.1089/tmj.2014.0113. [DOI] [PubMed] [Google Scholar]
  • 14.Jin H, Wu S, Di Capua P. Development of a clinical forecasting model to predict comorbid depression among diabetes patients and an application in depression screening policy making. Prev Chronic Dis. 2015 Sep 03;12:E142. doi: 10.5888/pcd12.150047. https://www.cdc.gov/pcd/issues/2015/15_0047.htm. [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 15.Natarajan S, Prabhakar A, Ramanan N, Baglione A, Connelly K, Siek K. Boosting for postpartum depression prediction. 2017 IEEE/ACM International Conference on Connected Health: Applications, Systems and Engineering Technologies (CHASE); 2017-07-17 to 2017-07-19; Philadelphia. 2017. Jul 17, pp. 232–240. [DOI] [Google Scholar]
  • 16.Andrew Am. An Introduction To Support Vector Machines And Other Kernel-based Learning Methods. Cambridge, United Kingdom: Cambridge University Press; 2001. Feb, pp. 103–115. [Google Scholar]
  • 17.Boser BE, Guyon IM, Vapnik VN. A training algorithm for optimal margin classifiers. COLT '92: Proceedings of the fifth annual workshop on Computational learning theory; 1992-07-27 to 1992-07-29; Pittsburgh, Pennsylvania. 1992. Jul 27, pp. 144–152. [DOI] [Google Scholar]
  • 18.Hapfelmeier A, Hothorn T, Ulm K, Strobl C. A new variable importance measure for random forests with missing data. Stat Comput. 2012 Aug 28;24(1):21–34. doi: 10.1007/s11222-012-9349-1. [DOI] [Google Scholar]
  • 19.Costafreda SG, Chu C, Ashburner J, Fu CHY. Prognostic and diagnostic potential of the structural neuroanatomy of depression. PLoS One. 2009 Jul 27;4(7):e6353. doi: 10.1371/journal.pone.0006353. http://dx.plos.org/10.1371/journal.pone.0006353. [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 20.Wang Y, Guo X, Lau Y, Chan KS, Yin L, Chen J. Psychometric evaluation of the Mainland Chinese version of the Edinburgh Postnatal Depression Scale. Int J Nurs Stud. 2009 Jun;46(6):813–23. doi: 10.1016/j.ijnurstu.2009.01.010. [DOI] [PubMed] [Google Scholar]
  • 21.Hewitt CE, Gilbody SM, Mann R, Brealey S. Instruments to identify post-natal depression: Which methods have been the most extensively validated, in what setting and in which language? Int J Psychiatry Clin Pract. 2010 Mar;14(1):72–6. doi: 10.3109/13651500903198020. [DOI] [PubMed] [Google Scholar]
  • 22.Cox JL, Holden JM, Sagovsky R. Detection of postnatal depression. Development of the 10-item Edinburgh Postnatal Depression Scale. Br J Psychiatry. 1987 Jun;150:782–6. doi: 10.1192/bjp.150.6.782. [DOI] [PubMed] [Google Scholar]
  • 23.Smith BW, Dalen J, Wiggins K, Tooley E, Christopher P, Bernard J. The brief resilience scale: assessing the ability to bounce back. Int J Behav Med. 2008 Sep;15(3):194–200. doi: 10.1080/10705500802222972. [DOI] [PubMed] [Google Scholar]
  • 24.Buysse DJ, Reynolds CF, Monk TH, Berman SR, Kupfer DJ. The Pittsburgh Sleep Quality Index: a new instrument for psychiatric practice and research. Psychiatry Res. 1989 May;28(2):193–213. doi: 10.1016/0165-1781(89)90047-4. [DOI] [PubMed] [Google Scholar]
  • 25.Lu T, Yan L, Ping X, Zhang G, Wu D. Analysis on reliability and validity of the Pittsburgh sleep quality index. Chongqing Med. 2014 Feb 18;2014(3):260–263. [Google Scholar]
  • 26.Xiao SY. The theoretical basis and research application of social support rating scale. Journal of Clinical Psychological Medicine. 1993 Nov 17;:98–100. http://www.cnki.com.cn/Article/CJFDTotal-LCJS402.019.htm. [Google Scholar]
  • 27.Spitzer RL, Kroenke K, Williams JBW, Löwe B. A brief measure for assessing generalized anxiety disorder: the GAD-7. Arch Intern Med. 2006 May 22;166(10):1092–7. doi: 10.1001/archinte.166.10.1092. [DOI] [PubMed] [Google Scholar]
  • 28.Li W, Lukai. Rongjing D, Dayi H, Sheng L. GW25-e4488 The value of Chinese version GAD-7 and PHQ-9 to screen anxiety and depression in cardiovascular outpatients. Journal of the American College of Cardiology. 2014 Oct;64(16):C222. doi: 10.1016/j.jacc.2014.06.1038. [DOI] [Google Scholar]
  • 29.Patel MJ, Khalaf A, Aizenstein HJ. Studying depression using imaging and machine learning methods. Neuroimage Clin. 2016;10:115–23. doi: 10.1016/j.nicl.2015.11.003. https://linkinghub.elsevier.com/retrieve/pii/S2213-1582(15)30020-6. [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 30.Malki K, Koritskaya E, Harris F, Bryson K, Herbster M, Tosto MG. Epigenetic differences in monozygotic twins discordant for major depressive disorder. Transl Psychiatry. 2016 Jun 14;6(6):e839–e839. doi: 10.1038/tp.2016.101. http://europepmc.org/abstract/MED/27300265. [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 31.Choudhury MD, Count S, Horvitz E. Predicting postpartum changes in emotion and behavior via social media. CHI '13: Proceedings of the SIGCHI Conference on Human Factors in Computing Systems; 2013-04-27 to 2013-05-02; Paris, France. 2013. Apr 27, pp. 27–2013. [DOI] [Google Scholar]
  • 32.Glasheen C, Richardson GA, Fabio A. A systematic review of the effects of postnatal maternal anxiety on children. Arch Womens Ment Health. 2010 Feb;13(1):61–74. doi: 10.1007/s00737-009-0109-y. http://europepmc.org/abstract/MED/19789953. [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 33.Martini J, Petzoldt J, Einsle F, Beesdo-Baum K, Höfler M, Wittchen H. Risk factors and course patterns of anxiety and depressive disorders during pregnancy and after delivery: a prospective-longitudinal study. J Affect Disord. 2015 Apr 01;175:385–95. doi: 10.1016/j.jad.2015.01.012. [DOI] [PubMed] [Google Scholar]
  • 34.Lutha SS, Cicchetti D. The construct of resilience: implications for interventions and social policies. Dev Psychopathol. 2000;12(4):857–85. doi: 10.1017/s0954579400004156. http://europepmc.org/abstract/MED/11202047. [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 35.Zautra AJ, Hall JS, Murray KE. Handbook Of Adult Resilience. New York, NY: The Guilford Press; 2010. Resilience: A new definition of health for people and communities. [Google Scholar]
  • 36.Fletcher D, Sarkar M. Psychological resilience. European Psychologist. 2013 Jan;18(1):12–23. doi: 10.1027/1016-9040/a000124. [DOI] [Google Scholar]
  • 37.Lu Q, Ding Q, Wang YL, Wang Y. Moderating effect of psychological resilience between prenatal perceived stress and postnatal depression among perinatal women. Chinese Nursing Research. 2019 Jul 8;33(11):1906–1910. doi: 10.12102/j.issn.1009-6493.2019.11.019. [DOI] [Google Scholar]
  • 38.O'hara MW, Swain AM. Rates and risk of postpartum depression—a meta-analysis. International Review of Psychiatry. 2009 Jul 11;8(1):37–54. doi: 10.3109/09540269609037816. [DOI] [Google Scholar]
  • 39.Josefsson A, Angelsiöö L, Berg G, Ekström C, Gunnervik C, Nordin C, Sydsjö G. Obstetric, somatic, and demographic risk factors for postpartum depressive symptoms. Obstetrics & Gynecology. 2002;99(2):223–228. doi: 10.1097/00006250-200202000-00011. [DOI] [PubMed] [Google Scholar]
  • 40.Sutter-Dallay AL, Cosnefroy O, Glatigny-Dallay E, Verdoux H, Rascle N. Evolution of perinatal depressive symptoms from pregnancy to two years postpartum in a low-risk sample: the MATQUID cohort. J Affect Disord. 2012 Jun;139(1):23–9. doi: 10.1016/j.jad.2011.08.018. [DOI] [PubMed] [Google Scholar]
  • 41.Boyd RC, Mogul M, Newman D, Coyne JC. Screening and referral for postpartum depression among low-income women: a qualitative perspective from community health workers. Depress Res Treat. 2011;2011:320605. doi: 10.1155/2011/320605. doi: 10.1155/2011/320605. [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 42.Gulamani SS, Premji SS, Kanji Z, Azam SI. A review of postpartum depression, preterm birth, and culture. J Perinat Neonatal Nurs. 2013;27(1):52–9; quiz 60. doi: 10.1097/JPN.0b013e31827fcf24. [DOI] [PubMed] [Google Scholar]

Associated Data

This section collects any data citations, data availability statements, or supplementary materials included in this article.

Supplementary Materials

Multimedia Appendix 1

Comparison of candidate predictors in the sample of pregnant women (N=508).

Multimedia Appendix 2

Comparison of demographic characteristics, including data sets of 618 pregnant women lost in the cohort and 508 mothers who left the cohort study after childbirth.

Multimedia Appendix 3

Definitions and coding of analyzed variables.


Articles from JMIR Medical Informatics are provided here courtesy of JMIR Publications Inc.

RESOURCES