Back

Cardiac age prediction using graph neural networks

Inacio, M. H. d. A.; Shah, M.; Jafari, M.; Shehata, N.; Meng, Q.; Bai, W.; Gandy, A.; Glocker, B.; O'Regan, D. P.

2023-04-20 cardiovascular medicine
10.1101/2023.04.19.23287590 medRxiv
Show abstract

The function of the human heart is characterised by complex patterns of motion that change throughout our lifespan due to accumulated damage across biological scales. Understanding the drivers of cardiac ageing is key to developing strategies for attenuating age-related processes. The motion of the surface of the heart can be conceived as a graph of connected points in space moving through time. Here we develop a generalisable framework for modelling three-dimensional motion as a graph and apply it to a task of predicting biological age. Using sequences of segmented cardiac imaging from 5064 participants in UK Biobank we train a graph neural network (GNN) to learn motion traits that predict healthy ageing. The GNN outperformed (mean absolute error, MAE = 4.74 years) a comparator dense neural network and boosting methods (MAE = 4.90 years and 5.08 years, respectively). We produce human-intelligible explanations of the predictions and using the trained model we also assess the effect of hypertension on biological age. This work shows how graph representations of complex motion can efficiently predict biologically meaningful outcomes.

Matching journals

The top 3 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.