KCL logo

Motion Modelling and Analysis Group

School of Biomedical Engineering and Imaging Sciences,
King's College London


Incorporating Topological Knowledge Into Deep Learning Based Segmentation

(Funded by EPSRC grant EP/P001009/1)

There are many cases in which the topology of a structure to be segmented is known a priori. For example, when segmenting the left ventricle of the heart from short-axis cine MR images, it is known that the correct segmentation will be a topological closed loop in two dimensions. However, the current state-of-the-art in deep learning based segmentation makes no use of this knowledge. In this work, we explicitly incorporate topological priors into a deep learning segmentation model for the first time, based on the concept of persistent homology. These priors require no additional ground truth segmentations and so the model is suitable for use in a semi-supervised setting. We found that by including the topological priors the number of topological errors was reduced whilst maintaining high overlap with the ground truth.

Examples of topology in segmentation
Three example MRI images of the short-axis view of the heart. The estimated segmentations produced by a U-net model (a, c, e) show topological errors. The segmentations for the same model trained with our topological prior (b, d, f) have the correct topology.