Anatomy-Informed Deep Learning for Abdominal Aortic Aneurysm Segmentation
Authors: Osamah Sufyan, Martin Brückmann, Ralph Wickenhöfer, Babette Dellen, Uwe Jaekel
Organizations: Faculty of Mathematics, Informatics and Technology, University of Applied Sciences Koblenz, 53424 Remagen, Germany · Department of Radiology, Herz-Jesu-Krankenhaus, 56428 Dernbach, Germany
In CT angiography, the accurate segmentation of abdominal aortic aneurysms (AAAs) is difficult due to large anatomical variability, low-contrast vessel boundaries, and the close proximity of organs whose intensities resemble vascular structures, often leading to false positives. To address these challenges, we propose an anatomy-aware segmentation framework that integrates organ exclusion masks derived from TotalSegmentator into the training process. These masks encode explicit anatomical priors by identifying non-vascular organsand penalizing aneurysm predictions within these regions, thereby guiding the U-Net to focus on the aorta and its pathological dilation while suppressing anatomically implausible predictions. Despite being trained on a relatively small dataset, the anatomy-aware model achieves high accuracy, substantially reduces false positives, and improves boundary consistency compared to a standard U-Net baseline. The results demonstrate that incorporating anatomical knowledge through exclusion masks provides an efficient mechanism to enhance robustness and generalization, enabling reliable AAA segmentation even with limited training data.
Figures & tables
Figure 1: (a) Comparison between a normal aorta and an aorta with a large aneurysm. (b) Example axial CT slice with lumen, thrombus, and corresponding ground-truth segmentation.
Figure 2: Pipeline of the proposed method from preprocessing to training.
Figure 3: (a) Qualitative comparison showing two representative slices from different patients. Each row corresponds to a different patient case. The model correctly segments the aneurysm with thrombus. (b) Loss curve and validation DICE vs epoch with the epoch with the best validation result marked with vertical red line
Model
Train Patients
Mean Dice
Precision
Recall
Baseline U-Net
10
0.637±0.442
0.796
0.762
Anatomy-Aware U-Net
10
0.914±0.033
0.965
0.882
Anatomy-Aware U-Net
16
0.906±0.179
0.961
0.942
Table 1: Model comparison on the test set.
Measurement
Test Patient 1
Test Patient 2
Maximal Diameter (pred)
5.3 cm
4.7 cm
Maximal Diameter (true)
5.8 cm
5.0 cm
Surface Area (pred)
616 cm 2
253 cm 2
Surface Area (true)
647 cm 2
191 cm 2
Volume (pred)
492 cm 3
203 cm 3
Volume (true)
538 cm 3
163 cm 3
Table 2: Predicted and ground-truth geometric measurements for each patient.
Figure 4: 3D reconstructed aneurysm from predicted segmentation (right) and corresponding ground truth (left).
Medical image segmentation delineates organs, lesions, and other structures in scans to support measurement, diagnosis, treatment planning, and research.
Robust segmentation of intraluminal thrombus is critical for risk assessment in Abdominal Aortic Aneurysm, yet it remains challenging due to heterogeneous thrombus features and low contrast with surrounding non-enhanced tissues. Domain shifts induced by different Computed Tomography Angiography (CTA) protocols further inhibit multi-center generalization of deep learning models. To address these challenges, we propose a patient-specific framework that integrates discriminative learning with anatomically informed priors. Our approach introduces two key components: (1) a patient-specific intensity normalization based on a Gaussian Mixture Model of local anatomy, and (2) an Uncertainty-Gated Anatomical Attention module that incorporates spatial priors while adaptively modulating their influence according to voxel-wise confidence. This design allows for anatomical guidance in ambiguous regions while suppressing unreliable priors. The proposed method achieves state-of-the-art performance on in-distribution test data and substantially outperforms existing alternatives in generalization to external multi-center CTA data, while remaining interpretable through an explicit separation of visual and anatomical evidence.
Erich Robbi, Daniele Ravanelli, Andrea Passerini
Department of Computer Science University of Trento Trento, Italy · Medical Physics Unit Santa Chiara Hospital Trento, Italy
Computed tomography image segmentation of complex abdominal aortic aneurysms (AAA) often fails because the models assign internal focus to irrelevant structures or do not focus on thin, low-contrast targets. Where the model looks is the primary training signal, and thus we propose an Explainable AI (XAI) guided encoder shaping framework. Our method computes a dense, attribution-based encoder focus map ("XAI field") from the final encoder block and uses it in two complementary ways: (i) we align the predicted probability mass to the XAI field to promote agreement between focus and output; and (ii) we route the field into a lightweight refinement pathway and a confidence prior that modulates logits at inference, suppressing distractors while preserving subtle structures. The objective terms serve only as control signals; the contribution is the integration of attribution guidance into representation and decoding. We evaluate clinically validated challenging cases curated for failure-prone scenarios. Compared to a base SAM setup, our implementation yields substantial improvements. The observed gains suggest that explicitly optimizing encoder focus via XAI guidance is a practical and effective principle for reliable segmentation in complex scenarios.
Abu Noman Md Sakib, Merjulah Roby, Zijie Zhang +3
University of Texas at San Antonio · Drexel University · Northwestern University
The TopAneu 2026 challenge requires intracranial aneurysms to be segmented at the voxel level and assigned to one of more than fifty vessel-specific classes, on CT and MR angiography alike. We address it with a six-stage anatomy-aware pipeline rather than a single segmentation network. A logistic-regression classifier routes each scan to a modality-specific branch; networks segment the vasculature, first as a binary mask and then over the full 36-class arterial taxonomy; a candidate localiser turns raw angiography and vessel priors into suspicious regions; a candidate-centred network segments the lesion in each crop; connected components become instances; and a per-instance classifier assigns each its anatomical location. Two principles organise the design. Whenever an upstream model output becomes a downstream training input it is generated out-of-fold, so no stage trains on unrealistically clean predictions of itself. And the anatomical decision is made once per lesion rather than once per voxel, which fits the data: of 406 connected aneurysm components in the full release, 405 carry exactly one location label. Task 1 is read off the Task 2 mask, making the two consistent by construction. We report stage-wise performance and a five-part ablation isolating what vessel priors and class-aware sampling contribute at each stage.
Gniewosz Drwiega, Wojciech Szymanski, Marek Wodzinski
Sano – Centre for Computational Personalised Medicine, International Research Foundation, Kraków, Poland · AGH University of Krakow, Kraków, Poland