Swin-VasMamba: A Topologically Constrained Model For 3D Vascular Segmentation

Accurate 3D vascular segmentation is essential for diagnosing and treating vascular diseases. This task remains challenging due to the complexity of the 3D data and the morphological diversity of blood vessels. In recent years, state space models (SSMs) have received a great attention for its good performance while preserving global receptive field and consuming less computing resources and time. Inspired by this, we propose a model called Swin-VasMamba for 3D vascular segmentation. It consists of a network called CMU-Net and a topologically constrained loss function called dsh loss. We compare our model with several other advanced segmentation models based on CNN, Transformer and Mamba. The results show that Swin-VasMamba achieves a state-of-the-art performance, with the highest Dice coefficient of 0.880, the lowest 95th-percentile of Hausdorff Distance (HD95) of 0.673, and the lowest Average Surface Distance (ASD) of 0.159 on a benchmark dataset.