A Generative Framework for Predictive Modeling of Multiple Chronic Conditions Using Graph Variational Autoencoder and Bandit-Optimized Graph Neural Network

Julian Carvajal Rico,Adel Alaeddini,Syed Hasib Akhter Faruqui,Susan P Fisher-Hoch,Joseph B Mccormick

2024-09-21

Abstract:Predicting the emergence of multiple chronic conditions (MCC) is crucial for early intervention and personalized healthcare, as MCC significantly impacts patient outcomes and healthcare costs. Graph neural networks (GNNs) are effective methods for modeling complex graph data, such as those found in MCC. However, a significant challenge with GNNs is their reliance on an existing graph structure, which is not readily available for MCC. To address this challenge, we propose a novel generative framework for GNNs that constructs a representative underlying graph structure by utilizing the distribution of the data to enhance predictive analytics for MCC. Our framework employs a graph variational autoencoder (GVAE) to capture the complex relationships in patient data. This allows for a comprehensive understanding of individual health trajectories and facilitates the creation of diverse patient stochastic similarity graphs while preserving the original feature set. These variations of patient stochastic similarity graphs, generated from the GVAE decoder, are then processed by a GNN using a novel Laplacian regularization technique to refine the graph structure over time and improves the prediction accuracy of MCC. A contextual Bandit is designed to evaluate the stochastically generated graphs and identify the best-performing graph for the GNN model iteratively until model convergence. We validate the performance of the proposed contextual Bandit algorithm against $\varepsilon$-Greedy and multi-armed Bandit algorithms on a large cohort (n = 1,592) of patients with MCC. These advancements highlight the potential of the proposed approach to transform predictive healthcare analytics, enabling a more personalized and proactive approach to MCC management.

Machine Learning

What problem does this paper attempt to address?

The problem that this paper attempts to solve is: **How to effectively predict the occurrence of multiple chronic conditions (MCC) for the purpose of early intervention and personalized medicine?** Specifically, in response to the challenge faced by graph neural networks (GNNs) in dealing with multiple chronic condition prediction - namely, the lack of readily - available graph - structured data - the paper proposes a new generative framework aimed at enhancing predictive analysis by constructing representative graph structures. ### Problem Background Multiple chronic conditions (such as diabetes, hypertension, obesity, etc.) have a significant impact on patient health and medical costs. Therefore, accurate prediction of the occurrence of these chronic conditions is crucial for early intervention and personalized medicine. However, traditional graph neural networks (GNNs) rely on existing graph - structured data, which is not easily obtainable in multiple chronic condition prediction. ### Core Problems of the Paper 1. **Lack of Readily - Available Graph - Structured Data**: The electronic health records (EHRs) of multiple chronic conditions usually do not contain graph structures that can be directly used for GNNs. 2. **How to Optimize Graph Structures to Improve Prediction Accuracy**: Even if there is a graph structure, how to ensure its quality and continuously optimize it remains a challenge. ### Solutions To solve the above problems, the paper proposes a generative framework that combines graph variational auto - encoder (GVAE), Laplacian - regularized graph neural network (LR - GNN), and the contextual Bandit algorithm. The main contributions of this framework are as follows: 1. **Generative Framework**: Use GVAE to capture complex relationships from patient data and generate diverse patient random similarity graphs while retaining the original feature set. 2. **Laplacian Regularization**: Integrate the random similarity graphs generated by GVAE into LR - GNN, and optimize the graph structure by iteratively applying Laplacian regularization techniques, thereby significantly improving the accuracy of MCC prediction. 3. **Contextual Bandit Optimization**: Design a contextual Bandit mechanism to iteratively evaluate and optimize the generated graph structures, select the best - performing graph for the GNN model until the model converges. ### Formula Representation - **Sampling and KL Divergence in GVAE**: \[ z=\mu+\exp\left(\frac{\log(\sigma^{2})}{2}\right)\cdot\epsilon,\quad\epsilon\sim N(0, 1) \] \[ \text{KL}(\mu,\log(\sigma^{2})) = -\frac{1}{2}\sum_{i = 1}^{N}\left(1+\log(\sigma_{i}^{2})-\mu_{i}^{2}-\exp(\log(\sigma_{i}^{2}))\right) \] - **Laplacian Regularization**: \[ L_{\text{reg}}=\lambda\times\text{mean}\left(\|y_{\text{pred}}-L y_{\text{pred}}\|^{2}\right) \] Through these methods, the paper provides an innovative solution that can effectively generate and optimize graph structures, thereby improving the accuracy of multiple chronic condition prediction.

A Generative Framework for Predictive Modeling of Multiple Chronic Conditions Using Graph Variational Autoencoder and Bandit-Optimized Graph Neural Network

Disease prediction with edge-variational graph convolutional networks

A Laplacian regularized graph neural network for predictive modeling of multiple chronic conditions

A novel interactive deep cascade spectral graph convolutional network with multi-relational graphs for disease prediction

Graph neural network modelling as a potentially effective method for predicting and analyzing procedures based on patients' diagnoses

Towards Probabilistic Generative Models Harnessing Graph Neural Networks for Disease-Gene Prediction

Graph Neural Networks for Heart Failure Prediction on an EHR-Based Patient Similarity Graph

GraphCare: Enhancing Healthcare Predictions with Personalized Knowledge Graphs

Scalable Modeling of Multivariate Longitudinal Data for Prediction of Chronic Kidney Disease Progression

Graph representation forecasting of patient's medical conditions: towards a digital twin

Self-explainable Graph Neural Network for Alzheimer's Disease And Related Dementias Risk Prediction

VEGN: Variant Effect Prediction with Graph Neural Networks

Comorbidity-based framework for Alzheimer's disease classification using graph neural networks

Graph neural network-based subgraph analysis for predicting adverse drug events

AMA-GCN: Adaptive Multi-layer Aggregation Graph Convolutional Network for Disease Prediction

A hybrid method of recurrent neural network and graph neural network for next-period prescription prediction

A Functional Model for Structure Learning and Parameter Estimation in Continuous Time Bayesian Network: An Application in Identifying Patterns of Multiple Chronic Conditions

A Model Predictive Control Functional Continuous Time Bayesian Network for Self-Management of Multiple Chronic Conditions

Dynamic Partitioning of Graphs Based on Multivariate Blood Glucose Data—A Graph Neural Network Model for Diabetes Prediction

Disease prediction using graph convolutional networks: Application to Autism Spectrum Disorder and Alzheimer's disease

Dynamic Functional Continuous Time Bayesian Networks for Prediction and Monitoring of the Impact of Patients’ Modifiable Lifestyle Behaviors on the Emergence of Multiple Chronic Conditions