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

Abstract

Background and goals: One of the biggest difficulties facing healthcare systems today is the prevalence of multiple chronic diseases (MCC). Mortality and the development of new chronic illnesses are more likely in those with MCC. Pre-existing diseases and risk factors specific to the patient have an impact on the complex stochastic process that guides the evolution of MCC. This study's goal is to use a brand-new Graph Neural Network (GNN) model to examine the connections between specific chronic illnesses, patient-level risk factors, and pre-existing conditions.

Methods: We propose a graph neural network model to analyze the relationship between five chronic conditions (diabetes, obesity, cognitive impairment, hyperlipidemia, and hypertension). The proposed model adds a graph Laplacian regularization term to the loss function, which aims to improve the parameter learning process and accuracy of the GNN based on the graph structure. For validation, we used historical data from the Cameron County Hispanic Cohort (CCHC).

Results: Evaluating the Laplacian regularized GNN on data from 600 patients, we expanded our analysis from two chronic conditions to five chronic conditions. The proposed model consistently surpassed a baseline GNN model, achieving an average accuracy of ≥89% across all combinations. In contrast, the performance of the standard model declined more markedly with the addition of more chronic conditions. The Laplacian regularization provided consistent predictions for adjacent nodes, beneficial in cases with shared attributes among nodes.

Conclusions: The incorporation of Laplacian regularization in our GNN model is essential, resulting in enhanced node categorization and better predictive performance by harnessing the graph structure. This study underscores the significance of considering graph structure when designing neural networks for graph data. Future research might further explore and refine this regularization method for various tasks using graph-structured data.

Keywords: Graph neural network; Laplacian regularization; Multiple chronic conditions; Risk factors.