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.
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 1 journal accounts for 50% of the predicted probability mass.