Back

Graph-based Fusion Modeling and Explanation for Disease Trajectory Prediction

Tariq, A.; Tang, S.; Sakhi, H.; Celi, L. A. G.; Newsome, J.; Rubin, D.; Trivedi, H.; Gicchoya, J. W.; Patel, B.; Banerjee, I.

2022-10-28 health informatics
10.1101/2022.10.25.22281469 medRxiv
Show abstract

We propose a relational graph to incorporate clinical similarity between patients while building personalized clinical event predictors with a focus on hospitalized COVID-19 patients. Our graph formation process fuses heterogeneous data, i.e., chest X-rays as node features and non-imaging EHR for edge formation. While node represents a snap-shot in time for a single patient, weighted edge structure encodes complex clinical patterns among patients. While age and gender have been used in the past for patient graph formation, our method incorporates complex clinical history while avoiding manual feature selection. The model learns from the patients own data as well as patterns among clinically-similar patients. Our visualization study investigates the effects of neighborhood of a node on its predictiveness and showcases the models tendency to focus on edge-connected patients with highly suggestive clinical features common with the node. The proposed model generalizes well by allowing edge formation process to adapt to an external cohort.

Matching journals

The top 6 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.