MSc Thesis/Guided Research Project: Age Prediction with Graph Neural Networks on UK BioBank data

About the dataset: The UK BioBank is a large medical dataset, containing data of about 50,000 participants. It holds images (brain MRI, abdominal MRI) as well as several thousand clinical parameters of the participants (demographic data, lifestyle data, clinical diagnoses,…). The lack of available data is a big challenge in medical applications of artificial intelligence. Therefore, the UK Biobank holds great potential to explore ML techniques on a large standardised dataset.
Methods: Graph Neural Networks (GNNs) are currently a hot topic in research. They provide a method for deep learning, directly applied on graph structured data. The goal of this project is to build a population graph, which holds extracted information from both, imaging data and clinical features, to predict the participants’ age. The advantage of building a population graph is that it allows to explore the whole dataset in one data structure. By comparing the actual biological age versus the predicted age from the network, we can draw conclusions about the participants’ health. The high dimensionality and the size of this dataset provide perfect conditions for data exploration and gives the freedom to implement own ideas and personal interests.


Your qualifications:

  • Prior knowledge in deep learning and computer vision
  • Very good programming skills in PyTorch
  • Interest in graph neural networks and geometric deep learning

If you are interested, please send us an email with your CV attached. We are looking forward to your message!


References

[1] Wu, Zonghan, et al. “A comprehensive survey on graph neural networks.” IEEE transactions on neural networks and learning systems 32.1 (2020): 4-24.
[2] Tarroni, Giacomo, et al. “Large-scale Quality control of cardiac imaging in population Studies: Application to UK Biobank.” Scientific reports 10.1 (2020): 1-11.
[3] Kipf, Thomas N., and Max Welling. “Semi-supervised classification with graph convolutional networks.” arXiv preprint arXiv:1609.02907 (2016).
[4] Cosmo, Luca, et al. “Latent-graph learning for disease prediction.” International Conference on Medical Image Computing and Computer-Assisted Intervention. Springer, Cham, 2020.
[5] Parisot, Sarah, et al. “Spectral graph convolutions for population-based disease prediction.” International conference on medical image computing and computer-assisted intervention. Springer, Cham, 2017.

Tamara Müller
Tamara Müller
PhD Student

My main research interests lie in applications of algorithms and technology in healthcare, artificial intelligence, and neuroscience.