Topological-Aware Regularization for Semi-Supervised Intracranial Aneurysm Vessel Segmentation

Feiyang Xiao, Yichi Zhang, Xigui Li, Yuanye Zhou, Chen Jiang, Xin Guo, LIMEI HAN, Yuxin Li, Fengping Zhu, Yuan Cheng

Keywords: Intracranial Aneurysm Segmentation, Semi-Supervised Learning, Topological-Aware Regularization

Abstract:

Accurate segmentation of intracranial aneurysm and their parent vessels (IA-Vessel) from magnetic resonance angiography is a critical prerequisite for computational fluid dynamics-based rupture risk assessment. While deep learning methods can automate this laborious task, they are hindered by the high cost and scarcity of expert annotations.
Most existing semi-supervised methods focus on enforcing regional constraints while largely ignoring topological constraints, which is insensitive to subtle but critical errors like vessel adhesion or surface irregularities, which are often unsuitable for downstream applications.
To address this gap, we introduce topological-aware regularization (TAR) for by incorporating the learning of local vascular topology to ensure the precise and geometrically correct segmentation of the IA-Vessel complex using only a small amount of labeled data. Experimental results on a multi-center MRA dataset show that our framework efficiently utilizes unlabeled data and outperforms state-of-the-art semi-supervised segmentation methods.
Instead of being restricted to a fixed framework, TAR is a plug-and-play strategy that can be seamlessly integrated into various semi-supervised frameworks to further boost their performance. The code and model weights will be made publicly available after the paper is accepted.

Primary Subject Area: Learning with Noisy Labels and Limited Data

Secondary Subject Area: Learning with Noisy Labels and Limited Data

Registration Requirement: Yes

Visa & Travel: Yes

Read CFP & Author Instructions: Yes

Originality Policy: Yes

Single-blind & Not Under Review Elsewhere: Yes

LLM Policy: Yes

Submission Number: 51