Search papers, labs, and topics across Lattice.
This paper introduces AG-TAL, an anatomically-guided topology-aware loss function, for improved multiclass segmentation of the Circle of Willis (CoW) in neurovascular imaging. AG-TAL combines radius-aware Dice loss, breakage-aware clDice loss using group convolutions, and adjacency-aware co-occurrence loss to address class imbalance, preserve connectivity, and enforce anatomical priors. Experiments on a large multi-center CoW dataset demonstrate that AG-TAL achieves state-of-the-art segmentation performance, particularly for small arteries, and generalizes well across multiple independent datasets.
Segmenting tiny brain arteries just got a whole lot better: a new loss function boosts Dice scores by up to 10% on these critical but challenging structures.
Accurate multiclass segmentation of the Circle of Willis (CoW) is essential for neurovascular disease management but remains challenging due to complex vascular topology and variable morphology. Existing deep learning methods often suffer from vascular discontinuities and inter-class misclassification, while current topological loss functions incur prohibitive computational costs in 3D multiclass settings. To address these limitations, we propose an Anatomically-Guided Topology-Aware Loss (AG-TAL) and introduce a large-scale, multi-center CoW dataset with unified annotations to facilitate robust model training. AG-TAL specifically integrates a radius-aware Dice loss to address class imbalance in small vessels, a breakage-aware clDice loss that utilizes group convolutions to efficiently preserve local connectivity, and an adjacency-aware co-occurrence loss that leverages anatomical priors to enforce distinct boundaries between neighboring arteries. Evaluated using 5-fold cross-validation, AG-TAL achieved an average Dice score of 80.85% for all CoW arteries, with small arteries notably higher by 1.05-3.09% compared to state-of-the-art methods. Across six independent datasets, the performance of AG-TAL achieved Dice scores ranging from 74.46% to 81.17% for all CoW arteries, with improvements of 2.20% to 9.98% for small arteries compared to other methods. This study demonstrates the superiority of AG-TAL in identifying multiclass CoW arteries and its ability to generalize well to multiple independent datasets. Furthermore, reliability analyses and clinical applications in an Alzheimer's disease cohort validate the AG-TAL's robustness and its potential for discovering imaging-based morphological biomarkers.
