A Generative Shape Compositional Framework: Towards Representative Populations of Virtual Heart Chimaeras
A Generative Shape Compositional Framework: Towards Representative Populations of Virtual Heart Chimaeras
复制标题
DOI:
10.48550/arxiv.2210.01607
复制
发表时间:
2022
期刊:
影响因子:
--
通讯作者:
Haoran Dou;S. Virtanen;N. Ravikumar;Alejandro F Frangi
中科院分区:
文献类型:
--
作者:
Haoran Dou;S. Virtanen;N. Ravikumar;Alejandro F Frangi
—Generating virtual populations of anatomy that capture sufficient variability while remaining plausible is essential for conducting in-silico trials of medical devices. However, not all anatomical shapes of interest are always available for each individual in a population. Imaging examinations and modalities employed and available can vary across individuals. Different imaging modalities may have different fields of view, be sensitive to signals from different tissues/organs, or both. Hence, missing/partially-overlapping anatomical information is often available across individuals in a population. We introduce a generative shape model for complex anatomical structures, learnable from datasets of unpaired datasets, i.e. where each substructure in the complex comes from datasets that have missing or partially overlapping substructures from disjoint subjects of the same population. The proposed generative model can synthesise complete whole-complex shape assemblies coined virtual chimaeras , as opposed to natural human chimaeras. We applied this framework to build virtual chimaeras from databases of whole-heart shape assemblies that each contribute samples for heart substructures. Specifically, we propose a graph neural network-based generative shape compositional framework which comprises two components - a part-aware generative shape model which captures the variability in shape observed for each structure of interest in the training population; and a spatial composition network which assembles/composes the structures synthesised by the former into multi-part shape assemblies (viz. virtual chimaeras ). We also propose a novel self-supervised learning scheme that enables the spatial composition network to be trained with partially overlapping data and weak labels. We trained and validated our approach using shapes of cardiac structures derived from cardiac magnetic resonance images available in the UK Biobank. When trained with complete and partially overlapping data, our approach significantly outperforms a PCA-based shape model (trained with complete data) in terms of generalisability and specificity. This demonstrates the superiority of the proposed approach as