Automated Labeling of Intracranial Arteries with Uncertainty Quantification Using Deep Learning

2509.17726v1 cs.CV, cs.LG, I.4.0 2025-09-24
Авторы:

Javier Bisbal, Patrick Winter, Sebastian Jofre, Aaron Ponce, Sameer A. Ansari, Ramez Abdalla, Michael Markl, Oliver Welin Odeback, Sergio Uribe, Cristian Tejos, Julio Sotelo, Susanne Schnell, David Marlevi

Резюме на русском

#### Контекст Анатомическая меткинга интракраниальных артерий является важной задачей в рамках разработки методов диагностики, мониторинга и терапии кардиоваскулярных заболеваний. Однако данная задача требует значительных временных затрат и подвержена значительной интерпертаторной вариативности. Недостаток универсальных автоматизированных систем, обеспечивающих высокую точность и надежность меткинга, ставит под угрозу качество полученных результатов. Данные проблемы мотивируют развитие новых методов, которые были бы широко применимы в клинической практике. #### Метод Для решения данной задачи предложена автоматизированная система, основанная на нейронных сетях. Методология включает три основных компонента: (1) сверточные нейронные сети (UNet) с резидентными блоками, (2) CS-Net с аугментацией канала и пространственной атенции для повышения чувствительности к кривизненным структурам, и (3) nnUNet, адаптивная система, которая автоматизирует обработку данных и архитектуру сети на основе характеристик данных. Для оценки точности работы системы использовались метрики Dice и surface distance. Для оценки неуверенности в системе внедрена тест-тайм аугментация (TTA) и разработана новая методика, снижающая ошибки интерполяции. #### Результаты Исследование проводилось на 35 сегментированных трехмерных томограммах 3D Time-of-Flight Magnetic Resonance Angiography (3D ToF-MRA). Архитектура nnUNet показала наилучшие результаты среди всех исследуемых моделей, с Dice-score в 0.922 и average surface distance в 0.387 мм. Был продемонстрирован улучшенный показатель точности в анатомически сложной вариативной среде. Оценка неуверенности в результатах проводилась с помощью TTA, которая показала способность системы определять области амбигуности, патологических изменений и несоответствий в ручных метках. Кроме того, система была проверена на валидационных данных 4D Flow MRI, показав близкий соответствие с ручными метками в расчетах потока крови. #### Значимость Предложенная система представляет собой простой и эффективный инструмент для автоматического меткинга интракраниальных артерий, который может существенно снизить время на ручную работу врачей. Данная технология может быть использована в различных областях, включая диагностику заболеваний, планирование хирургических вмешательств и мониторинг состояния здоровья. Основное преимущество системы заключается в ее надежности и возможности оценивать неуверенность в результатах, обеспечивая более точную клиническую интеграцию. #### Выводы Разработанная система демонстрирует высокую точность и надежность в автоматичес

Abstract

Accurate anatomical labeling of intracranial arteries is essential for cerebrovascular diagnosis and hemodynamic analysis but remains time-consuming and subject to interoperator variability. We present a deep learning-based framework for automated artery labeling from 3D Time-of-Flight Magnetic Resonance Angiography (3D ToF-MRA) segmentations (n=35), incorporating uncertainty quantification to enhance interpretability and reliability. We evaluated three convolutional neural network architectures: (1) a UNet with residual encoder blocks, reflecting commonly used baselines in vascular labeling; (2) CS-Net, an attention-augmented UNet incorporating channel and spatial attention mechanisms for enhanced curvilinear structure recognition; and (3) nnUNet, a self-configuring framework that automates preprocessing, training, and architectural adaptation based on dataset characteristics. Among these, nnUNet achieved the highest labeling performance (average Dice score: 0.922; average surface distance: 0.387 mm), with improved robustness in anatomically complex vessels. To assess predictive confidence, we implemented test-time augmentation (TTA) and introduced a novel coordinate-guided strategy to reduce interpolation errors during augmented inference. The resulting uncertainty maps reliably indicated regions of anatomical ambiguity, pathological variation, or manual labeling inconsistency. We further validated clinical utility by comparing flow velocities derived from automated and manual labels in co-registered 4D Flow MRI datasets, observing close agreement with no statistically significant differences. Our framework offers a scalable, accurate, and uncertainty-aware solution for automated cerebrovascular labeling, supporting downstream hemodynamic analysis and facilitating clinical integration.

Ссылки и действия