Introducing Soft Topology Constraints in Deep Learning-based Segmentation using Projected Pooling Loss - Inria - Institut national de recherche en sciences et technologies du numérique
Conference Papers Year : 2023

Introducing Soft Topology Constraints in Deep Learning-based Segmentation using Projected Pooling Loss

Abstract

Deep learning methods have achieved impressive results for 3D medical image segmentation. However, when the network is only guided by voxel-level information, it may provide anatomically aberrant segmentations. When performing manual segmentations, experts heavily rely on prior anatomical knowledge. Topology is an important prior information due to its stability across patients. Recently, several losses based on persistent homology were proposed to constrain topology. Persistent homology offers a principled way to control topology. However, it is computationally expensive and complex to implement, in particular in 3D. In this paper, we propose a novel loss function to introduce topological priors in deep learning-based segmentation, which is fast to compute and easy to implement. The loss performs a projected pooling within two steps. We first focus on errors from a global perspective by using 3D MaxPooling to obtain projections of 3D data onto three planes: axial, coronal and sagittal. Then, 2D MaxPooling layers with different kernel sizes are used to extract topological features from the multi-view projections. These two steps are combined using only MaxPooling, thus ensuring the efficiency of the loss function. Our approach was evaluated in several medical image datasets (spleen, heart, hippocampus, red nucleus). It allowed reducing topological errors and, in some cases, improving voxel-level accuracy.
Fichier principal
Vignette du fichier
proceedings_version.pdf (895.33 Ko) Télécharger le fichier
Origin Files produced by the author(s)

Dates and versions

hal-03832309 , version 1 (27-10-2022)
hal-03832309 , version 2 (27-10-2022)

Identifiers

  • HAL Id : hal-03832309 , version 2

Cite

Guanghui Fu, Rosana El Jurdi, Lydia Chougar, Didier Dormont, Romain Valabregue, et al.. Introducing Soft Topology Constraints in Deep Learning-based Segmentation using Projected Pooling Loss. SPIE Medical Imaging 2023, Feb 2023, San Diego, United States. ⟨hal-03832309v2⟩
217 View
303 Download

Share

More