Topology-Preserving Image Segmentation with Spatial-Aware Persistent Feature Matching

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wen, Bo, Zhang, Haochen, Bartsch, Dirk-Uwe G., Freeman, William R., Nguyen, Truong Q., An, Cheolhong
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866908615673118720
author Wen, Bo
Zhang, Haochen
Bartsch, Dirk-Uwe G.
Freeman, William R.
Nguyen, Truong Q.
An, Cheolhong
author_facet Wen, Bo
Zhang, Haochen
Bartsch, Dirk-Uwe G.
Freeman, William R.
Nguyen, Truong Q.
An, Cheolhong
contents Topological correctness is critical for segmentation of tubular structures, which pervade in biomedical images. Existing topological segmentation loss functions are primarily based on the persistent homology of the image. They match the persistent features from the segmentation with the persistent features from the ground truth and minimize the difference between them. However, these methods suffer from an ambiguous matching problem since the matching only relies on the information in the topological space. In this work, we propose an effective and efficient Spatial-Aware Topological Loss Function that further leverages the information in the original spatial domain of the image to assist the matching of persistent features. Extensive experiments on images of various types of tubular structures show that the proposed method has superior performance in improving the topological accuracy of the segmentation compared with state-of-the-art methods. Code is available at https://github.com/JRC-VPLab/SATLoss.
format Preprint
id arxiv_https___arxiv_org_abs_2412_02076
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Topology-Preserving Image Segmentation with Spatial-Aware Persistent Feature Matching
Wen, Bo
Zhang, Haochen
Bartsch, Dirk-Uwe G.
Freeman, William R.
Nguyen, Truong Q.
An, Cheolhong
Computer Vision and Pattern Recognition
Topological correctness is critical for segmentation of tubular structures, which pervade in biomedical images. Existing topological segmentation loss functions are primarily based on the persistent homology of the image. They match the persistent features from the segmentation with the persistent features from the ground truth and minimize the difference between them. However, these methods suffer from an ambiguous matching problem since the matching only relies on the information in the topological space. In this work, we propose an effective and efficient Spatial-Aware Topological Loss Function that further leverages the information in the original spatial domain of the image to assist the matching of persistent features. Extensive experiments on images of various types of tubular structures show that the proposed method has superior performance in improving the topological accuracy of the segmentation compared with state-of-the-art methods. Code is available at https://github.com/JRC-VPLab/SATLoss.
title Topology-Preserving Image Segmentation with Spatial-Aware Persistent Feature Matching
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.02076