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 Accéder directement au contenu
Communication Dans Un Congrès Année : 2023

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

Résumé

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
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

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

Identifiants

  • HAL Id : hal-03832309 , version 2

Citer

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⟩
185 Consultations
224 Téléchargements

Partager

Gmail Facebook X LinkedIn More