A Geometric Deep Learning Framework for Generation of Virtual Left Ventricles as Graphs

Soodeh Kalaie, Andy Bulpitt, Alejandro F. Frangi, Ali Gooya

Research output: Contribution to journalConference articlepeer-review

Abstract

Generative statistical models have a wide range of applications in the modelling of anatomies. In-silico clinical trials of medical devices, for instance, require the development of virtual populations of anatomy that capture enough variability while remaining plausible. Model construction and use are heavily influenced by the correspondence problem and establishing shape matching over a large number of training data. This study focuses on generating virtual cohorts of left ventricle geometries resembling different-sized shape populations, suitable for in-silico experiments. We present an unsupervised data-driven probabilistic generative model for shapes. This framework incorporates an attention-based shape matching procedure using graph neural networks, coupled with a β−VAE generation model, eliminating the need for initial shape correspondence. Left ventricle shapes derived from cardiac magnetic resonance images available in the UK Biobank are utilized for training and validating the framework. We investigate our method's generative capabilities in terms of generalisation and specificity and show that it is able to synthesise virtual populations of realistic shapes with volumetric measurements in line with actual clinical indices. Moreover, results show our method outperforms joint registration-PCA-based models.

Original languageEnglish
Pages (from-to)426-442
Number of pages17
JournalProceedings of Machine Learning Research
Volume227
Publication statusPublished - 2023
Event6th International Conference on Medical Imaging with Deep Learning, MIDL 2023 - Nashville, United States
Duration: 10 Jul 202312 Jul 2023

Keywords

  • Attention
  • Generative modelling
  • Geometric Deep Learning
  • Graph Neural Networks
  • Shape matching
  • Virtual populations

Fingerprint

Dive into the research topics of 'A Geometric Deep Learning Framework for Generation of Virtual Left Ventricles as Graphs'. Together they form a unique fingerprint.

Cite this