Abstract:Medical image segmentation, which aims to automatically extract anatomical or pathological structures, plays a key role in computer-aided diagnosis and disease analysis. Despite the problem has been widely studied, existing methods are prone to topological errors. In medical imaging, the topology of the structure, such as the kidney or lung, is usually known. Preserving the topology of the structure in the segmentation process is of utmost importance for accurate image analysis. In this work, a novel learning-based segmentation model is proposed. A {\it topology-preserving segmentation network (TPSN)} is trained to give an accurate segmentation result of an input image that preserves the prescribed topology. TPSN is a deformation-based model that yields a deformation map through a UNet, which takes the medical image and a template mask as inputs. The main idea is to deform a template mask describing the prescribed topology by a diffeomorphism to segment the object in the image. The topology of the shape in the template mask is well preserved under the diffeomorphic map. The diffeomorphic property of the map is controlled by introducing a regularization term related to the Jacobian in the loss function. As such, a topology-preserving segmentation result can be guaranteed. Furthermore, a multi-scale TPSN is developed in this paper that incorporates multi-level information of images to produce more precise segmentation results. To evaluate our method, we applied the 2D TPSN on Ham10000 and 3D TPSN on KiTS21. Experimental results illustrate our method outperforms the baseline UNet segmentation model with/without connected-component analysis (CCA) by both the dice score and IoU score. Besides, results show that our method can produce reliable results even in challenging cases, where pixel-wise segmentation models by UNet and CCA fail to obtain accurate results.

Directional Connectivity-based Segmentation of Medical Images

3D Graph-S<SUP>2</SUP>Net: Shape-Aware Self-ensembling Network for Semi-supervised Segmentation with Bilateral Graph Convolution

Graph-BAS3Net: Boundary-Aware Semi-Supervised Segmentation Network with Bilateral Graph Convolution

SynergyNet: Bridging the Gap between Discrete and Continuous Representations for Precise Medical Image Segmentation

Sparse Graph Connectivity for Image Segmentation

Manifold-Aware Local Feature Modeling for Semi-Supervised Medical Image Segmentation

Dendritic Deep Learning for Medical Segmentation

Deep Active Contour Network for Medical Image Segmentation.

Dendritic Learning and Miss Region Detection-Based Deep Network for Multi-scale Medical Segmentation

A Mutual Inclusion Mechanism for Precise Boundary Segmentation in Medical Images

Depth disentanglement strategy of latent space for medical image segmentation

Topology-Preserving Segmentation Network: A Deep Learning Segmentation Framework for Connected Component

Feature interaction network based on hierarchical decoupled convolution for 3D medical image segmentation

Head and Neck Vessel Segmentation with Connective Topology Using Affinity Graph

One Network to Segment Them All: A General, Lightweight System for Accurate 3D Medical Image Segmentation

Shape-aware Semi-supervised 3D Semantic Segmentation for Medical Images

Target area distillation and section attention segmentation network for accurate 3D medical image segmentation

CylinGCN: Cylindrical structures segmentation in 3D biomedical optical imaging by a contour-based graph convolutional network

Decompose-and-Integrate Learning for Multi-class Segmentation in Medical Images

Narrowing the semantic gaps in U-Net with learnable skip connections: The case of medical image segmentation

Crossover-Net: Leveraging vertical-horizontal crossover relation for robust medical image segmentation