BuyTheDips: PathLoss for improved topology-preserving deep learning-based image segmentation - Archive ouverte HAL Accéder directement au contenu
Pré-Publication, Document De Travail Année : 2022

BuyTheDips: PathLoss for improved topology-preserving deep learning-based image segmentation

Résumé

Capturing the global topology of an image is essential for proposing an accurate segmentation of its domain. However, most of existing segmentation methods do not preserve the initial topology of the given input, which is detrimental for numerous downstream object-based tasks. This is all the more true for deep learning models which most work at local scales. In this paper, we propose a new topology-preserving deep image segmentation method which relies on a new leakage loss: the Pathloss. Our method is an extension of the BALoss [1], in which we want to improve the leakage detection for better recovering the closeness property of the image segmentation. This loss allows us to correctly localize and fix the critical points (a leakage in the boundaries) that could occur in the predictions, and is based on a shortest-path search algorithm. This way, loss minimization enforces connectivity only where it is necessary and finally provides a good localization of the boundaries of the objects in the image. Moreover, according to our research, our Pathloss learns to preserve stronger elongated structure compared to methods without using topology-preserving loss. Training with our topological loss function, our method outperforms state-ofthe-art topology-aware methods on two representative datasets of different natures: Electron Microscopy and Historical Map.
Fichier non déposé

Dates et versions

hal-03725431 , version 1 (17-07-2022)

Identifiants

  • HAL Id : hal-03725431 , version 1

Citer

Vũ Ngọc Minh Ôn, Yizi Chen, Nicolas Boutry, Jonathan Fabrizio, Clément Mallet. BuyTheDips: PathLoss for improved topology-preserving deep learning-based image segmentation. 2022. ⟨hal-03725431⟩
111 Consultations
16 Téléchargements

Partager

Gmail Facebook X LinkedIn More