IDEAS home Printed from https://ideas.repec.org/a/plo/pone00/0177811.html
   My bibliography  Save this article

Predicting congenital heart defects: A comparison of three data mining methods

Author

Listed:
  • Yanhong Luo
  • Zhi Li
  • Husheng Guo
  • Hongyan Cao
  • Chunying Song
  • Xingping Guo
  • Yanbo Zhang

Abstract

Congenital heart defects (CHD) is one of the most common birth defects in China. Many studies have examined risk factors for CHD, but their predictive abilities have not been evaluated. In particular, few studies have attempted to predict risks of CHD from, necessarily unbalanced, population-based cross-sectional data. Therefore, we developed and validated machine learning models for predicting, before and during pregnancy, women’s risks of bearing children with CHD. We compared the results of these models in a large-scale, comprehensive population-based retrospective cross-sectional epidemiological survey of birth defects in six counties in Shanxi Province, China, covering 2006 to 2008. This contained 78 cases of CHD among 33831 live births. We constructed nine synthetic variables to use in the models: maternal age, annual per capita income, family history, maternal history of illness, nutrition and folic acid deficiency, maternal illness in pregnancy, medication use in pregnancy, environmental risk factors in pregnancy, and unhealthy maternal lifestyle in pregnancy. The machine learning algorithms Weighted Support Vector Machine (WSVM) and Weighted Random Forest (WRF) were trained on, and a logistic regression (Logit) was fitted to, two-thirds of the data. Their predictive abilities were then tested in the remaining data. True positive rate (TPR), true negative rate (TNR), accuracy (ACC), area under the curves (AUC), G-means, and Weighted accuracy (WTacc) were used to compare the classification performance of the models. Median values, from repeating the data partitioning 1000 times, were used in all comparisons. The TPR and TNR of the three classifiers were above 0.65 and 0.93, respectively, better than any reported in the literature. TPR, wtACC, AUC and G were highest for WSVM, showing that it performed best. All three models are precise enough to identify groups at high risk of CHD. They should all be considered for future investigations of other birth defects and diseases.

Suggested Citation

  • Yanhong Luo & Zhi Li & Husheng Guo & Hongyan Cao & Chunying Song & Xingping Guo & Yanbo Zhang, 2017. "Predicting congenital heart defects: A comparison of three data mining methods," PLOS ONE, Public Library of Science, vol. 12(5), pages 1-14, May.
  • Handle: RePEc:plo:pone00:0177811
    DOI: 10.1371/journal.pone.0177811
    as

    Download full text from publisher

    File URL: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0177811
    Download Restriction: no

    File URL: https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0177811&type=printable
    Download Restriction: no

    File URL: https://libkey.io/10.1371/journal.pone.0177811?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    References listed on IDEAS

    as
    1. Rafael Pino-Mejias & Mercedes Carrasco-Mairena & Antonio Pascual-Acosta & Maria-Dolores Cubiles-De-La-Vega & Joaquin Munoz-Garcia, 2008. "A comparison of classification models to identify the Fragile X Syndrome," Journal of Applied Statistics, Taylor & Francis Journals, vol. 35(3), pages 233-244.
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Yafei Wu & Ya Fang, 2020. "Stroke Prediction with Machine Learning Methods among Older Chinese," IJERPH, MDPI, vol. 17(6), pages 1-11, March.

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Nader Fallah & Arnold Mitnitski & Kenneth Rockwood, 2011. "Applying neural network Poisson regression to predict cognitive score changes," Journal of Applied Statistics, Taylor & Francis Journals, vol. 38(9), pages 2051-2062, November.
    2. Alexander Herr, 2010. "Statistics for Categorical Surveys—A New Strategy for Multivariate Classification and Determining Variable Importance," Sustainability, MDPI, vol. 2(2), pages 1-18, February.
    3. Jian Zhou & Xibing Li & Hani Mitri, 2015. "Comparative performance of six supervised learning methods for the development of models of hard rock pillar stability prediction," Natural Hazards: Journal of the International Society for the Prevention and Mitigation of Natural Hazards, Springer;International Society for the Prevention and Mitigation of Natural Hazards, vol. 79(1), pages 291-316, October.

    More about this item

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:plo:pone00:0177811. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.