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

Feature selection and transformation by machine learning reduce variable numbers and improve prediction for heart failure readmission or death

Author

Listed:
  • Saqib E Awan
  • Mohammed Bennamoun
  • Ferdous Sohel
  • Frank M Sanfilippo
  • Benjamin J Chow
  • Girish Dwivedi

Abstract

Background: The prediction of readmission or death after a hospital discharge for heart failure (HF) remains a major challenge. Modern healthcare systems, electronic health records, and machine learning (ML) techniques allow us to mine data to select the most significant variables (allowing for reduction in the number of variables) without compromising the performance of models used for prediction of readmission and death. Moreover, ML methods based on transformation of variables may potentially further improve the performance. Objective: To use ML techniques to determine the most relevant and also transform variables for the prediction of 30-day readmission or death in HF patients. Methods: We identified all Western Australian patients aged 65 years and above admitted for HF between 2003–2008 in linked administrative data. We evaluated variables associated with HF readmission or death using standard statistical and ML based selection techniques. We also tested the new variables produced by transformation of the original variables. We developed multi-layer perceptron prediction models and compared their predictive performance using metrics such as Area Under the receiver operating characteristic Curve (AUC), sensitivity and specificity. Results: Following hospital discharge, the proportion of 30-day readmissions or death was 23.7% in our cohort of 10,757 HF patients. The prediction model developed by us using a smaller set of variables (n = 8) had comparable performance (AUC 0.62) to the traditional model (n = 47, AUC 0.62). Transformation of the original 47 variables further improved (p

Suggested Citation

  • Saqib E Awan & Mohammed Bennamoun & Ferdous Sohel & Frank M Sanfilippo & Benjamin J Chow & Girish Dwivedi, 2019. "Feature selection and transformation by machine learning reduce variable numbers and improve prediction for heart failure readmission or death," PLOS ONE, Public Library of Science, vol. 14(6), pages 1-13, June.
  • Handle: RePEc:plo:pone00:0218760
    DOI: 10.1371/journal.pone.0218760
    as

    Download full text from publisher

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

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

    File URL: https://libkey.io/10.1371/journal.pone.0218760?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
    ---><---

    Citations

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


    Cited by:

    1. Saurabh Shukla & Mohd Fadzil Hassan & Muhammad Khalid Khan & Low Tang Jung & Azlan Awang, 2019. "An analytical model to minimize the latency in healthcare internet-of-things in fog computing environment," PLOS ONE, Public Library of Science, vol. 14(11), pages 1-31, November.

    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:0218760. 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.

    We have no bibliographic references for this item. You can help adding them by using 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.