Deformable registration and generative modelling of aortic anatomies by auto-decoders and neural ODEs

TL;DR

This paper presents AD-SVFD, a deep learning model that performs deformable registration and generative modeling of aortic anatomies using auto-decoders and neural ODEs, enabling efficient shape analysis and synthesis.

Contribution

The work introduces a novel auto-decoder based neural ODE framework for vascular shape registration and generation, improving efficiency and generalization across shape cohorts.

Findings

High accuracy in aortic shape registration

Efficient inference with low-dimensional codes

Generative capability for synthetic anatomies

Abstract

This work introduces AD-SVFD, a deep learning model for the deformable registration of vascular shapes to a pre-defined reference and for the generation of synthetic anatomies. AD-SVFD operates by representing each geometry as a weighted point cloud and models ambient space deformations as solutions at unit time of ODEs, whose time-independent right-hand sides are expressed through artificial neural networks. The model parameters are optimized by minimizing the Chamfer Distance between the deformed and reference point clouds, while backward integration of the ODE defines the inverse transformation. A distinctive feature of AD-SVFD is its auto-decoder structure, that enables generalization across shape cohorts and favors efficient weight sharing. In particular, each anatomy is associated with a low-dimensional code that acts as a self-conditioning field and that is jointly optimized with the network parameters during training. At inference, only the latent codes are fine-tuned, substantially reducing computational overheads. Furthermore, the use of implicit shape representations enables generative applications: new anatomies can be synthesized by suitably sampling from the latent space and applying the corresponding inverse transformations to the reference geometry. Numerical experiments, conducted on healthy aortic anatomies, showcase the high-quality results of AD-SVFD, which yields extremely accurate approximations at competitive computational costs.