Obesity Prediction with EHR Data: A deep learning approach with interpretable elements

Abstract

Childhood obesity is a major public health challenge. Early prediction and identification of the children at a high risk of developing childhood obesity may help in engaging earlier and more effective interventions to prevent and manage obesity. Most existing predictive tools for childhood obesity primarily rely on traditional regression-type methods using only a few hand-picked features and without exploiting longitudinal patterns of children data. Deep learning methods allow the use of high-dimensional longitudinal datasets. In this paper, we present a deep learning model designed for predicting future obesity patterns from generally available items on children medical history. To do this, we use a large unaugmented electronic health records dataset from a large pediatric health system. We adopt a general LSTM network architecture which are known to better represent the longitudinal data. We train our proposed model on both dynamic and static EHR data. Our model is used to predict obesity for ages between 2-20 years. We compared the performance of our LSTM model with other machine learning methods that aggregate over sequential data and ignore temporality. To add interpretability, we have additionally included an attention layer to calculate the attention scores for the timestamps and rank features of each timestamp.

19 pages, 4 Tables, 7 figures

References in corpus (6)

• ADADELTA: An Adaptive Learning Rate Method
• Patient2Vec: A Personalized Interpretable Deep Representation of the Longitudinal Electronic Health Record
• Obesity Prediction with EHR Data: A deep learning approach with interpretable elements
• CarePre: An Intelligent Clinical Decision Assistance System
• Deepr: A Convolutional Net for Medical Records
• Multi-modal Predictive Models of Diabetes Progression

Cited by in corpus (7)

• Obesity Prediction with EHR Data: A deep learning approach with interpretable elements
• A review of Generative Adversarial Networks for Electronic Health Records: applications, evaluation measures and data sources
• Comparison of Missing Data Imputation Methods using the Framingham Heart study dataset
• Time-series Imputation and Prediction with Bi-Directional Generative Adversarial Networks
• CURENet: Combining Unified Representations for Efficient Chronic Disease Prediction
• Effective Predictive Modeling for Emergency Department Visits and Evaluating Exogenous Variables Impact: Using Explainable Meta-learning Gradient Boosting
• HyMaTE: A Hybrid Mamba and Transformer Model for EHR Representation Learning