Swin-VasMamba: A Topologically Constrained Model For 3D Vascular Segmentation
Ziyu Liu, Jiaxuan Li, Xiangjian He, Qing Xu, Xin Chen, Shoujun Zhou
Abstract
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.
BibTeX
@inproceedings{icassp2025_swinvasmambaatop,
title = {Swin-VasMamba: A Topologically Constrained Model For 3D Vascular Segmentation},
author = {Ziyu Liu and Jiaxuan Li and Xiangjian He and Qing Xu and Xin Chen and Shoujun Zhou},
booktitle = {ICASSP 2025},
year = {2025}
}