Attention-based neural networks for clinical prediction modelling on electronic health records.

Journal: BMC medical research methodology
Published Date:

Abstract

BACKGROUND: Deep learning models have had a lot of success in various fields. However, on structured data they have struggled. Here we apply four state-of-the-art supervised deep learning models using the attention mechanism and compare against logistic regression and XGBoost using discrimination, calibration and clinical utility.

Authors

  • Egill A Fridgeirsson
    Department of Medical Informatics, Erasmus University Medical Center, Rotterdam, The Netherlands.
  • David Sontag
    1 Department of Computer Science, New York University , New York, New York.
  • Peter Rijnbeek
    Department of Medical Informatics, Erasmus Medical Center, Rotterdam, The Netherlands.