English

Modelling EHR timeseries by restricting feature interaction

Machine Learning 2019-11-18 v1 Computers and Society Machine Learning

Abstract

Time series data are prevalent in electronic health records, mostly in the form of physiological parameters such as vital signs and lab tests. The patterns of these values may be significant indicators of patients' clinical states and there might be patterns that are unknown to clinicians but are highly predictive of some outcomes. Many of these values are also missing which makes it difficult to apply existing methods like decision trees. We propose a recurrent neural network model that reduces overfitting to noisy observations by limiting interactions between features. We analyze its performance on mortality, ICD-9 and AKI prediction from observational values on the Medical Information Mart for Intensive Care III (MIMIC-III) dataset. Our models result in an improvement of 1.1% [p<0.01] in AU-ROC for mortality prediction under the MetaVision subset and 1.0% and 2.2% [p<0.01] respectively for mortality and AKI under the full MIMIC-III dataset compared to existing state-of-the-art interpolation, embedding and decay-based recurrent models.

Keywords

Cite

@article{arxiv.1911.06410,
  title  = {Modelling EHR timeseries by restricting feature interaction},
  author = {Kun Zhang and Yuan Xue and Gerardo Flores and Alvin Rajkomar and Claire Cui and Andrew M. Dai},
  journal= {arXiv preprint arXiv:1911.06410},
  year   = {2019}
}

Comments

Machine Learning for Health (ML4H) at NeurIPS 2019 - Extended Abstract

R2 v1 2026-06-23T12:16:38.598Z