Hessian-based lightweight neural network for brain vessel segmentation on a minimal training dataset
2508.15660v1
eess.IV, cs.CV, I.4.6; I.5.4; J.3
2025-08-23
Авторы:
Alexandra Bernadotte, Elfimov Nikita, Mikhail Shutov, Ivan Menshikov
Резюме на русском
## Контекст
Одним из ключевых аспектов успешных вмешательств в нейрохирургии является точная сегментация головных сосудов на магнитных резонансных изображениях. Эта задача критически важна для подготовки к операциям, таким как ремонт аневризмы или трубопрокладка. Несмотря на развитие медицинского искусственного интеллекта, актуальной проблемой остается недостаток качественных данных для обучения нейронных сетей. Аннотация данных часто выполняется вручную или с использованием классических методов, таких как фильтр Франги, которые часто недостаточно точны. Из-за недостатка данных и необходимости в эффективных решениях возникает потребность в новых подходах к автоматизированной сегментации головных сосудов.
## Метод
Мы предлагаем HessNet — новую легковесную нейронную сеть семи слоев, основанную на матрицах Хесса, для трехмерной сегментации комплексных структур, таких как трубы. Наша сеть содержит всего 6000 параметров и может работать на процессоре, что значительно снижает ресурсоемкость обучения. Мы также разработали новую подборку данных на основе IXI MRA-изображений, которая включает в себя 200 семплов с подробной аннотацией головных сосудов, выполненной экспертами под наблюдением нейроvascularных специалистов. Этот подход обеспечивает высокую точность сегментации сосудов, даже при минимальном объеме обучающих данных.
## Результаты
Наши эксперименты показали, что HessNet показывает высокую точность сегментации сосудов даже при ограниченном наборе тренировочных данных. Мы сравнили результаты с другими техниками, такими как Франги, и показали, что HessNet показывает значительно лучший результат. Добавляя только небольшое количество данных, мы достигли состояния технологий, что демонстрирует эффективность нашего подхода в условиях недостатка данных. Также, мы разработали полностью подготовленную подборку данных, которая может быть использована для дальнейших исследований.
## Значимость
Наш подход может быть применен в различных областях, таких как нейрохирургия, для точной сегментации головных сосудов. Он уменьшает необходимость вручную аннотированных данных для обучения, что экономит время и ресурсы. Благодаря своей легковесной архитектуре, HessNet может быть использован на ресурс-ограниченных устройствах, таких как процессоры. Это открывает новые возможности для использования нейронных сетей в медицине, даже в условиях ограниченных вычислительных ресурсов.
## Выводы
Мы представили новую легковесную нейронную сеть HessNet, которая позволяет эффективно сегментировать головные сосуды даже при небольшом количестве т
Abstract
Accurate segmentation of blood vessels in brain magnetic resonance
angiography (MRA) is essential for successful surgical procedures, such as
aneurysm repair or bypass surgery. Currently, annotation is primarily performed
through manual segmentation or classical methods, such as the Frangi filter,
which often lack sufficient accuracy. Neural networks have emerged as powerful
tools for medical image segmentation, but their development depends on
well-annotated training datasets. However, there is a notable lack of publicly
available MRA datasets with detailed brain vessel annotations.
To address this gap, we propose a novel semi-supervised learning lightweight
neural network with Hessian matrices on board for 3D segmentation of complex
structures such as tubular structures, which we named HessNet. The solution is
a Hessian-based neural network with only 6000 parameters. HessNet can run on
the CPU and significantly reduces the resource requirements for training neural
networks. The accuracy of vessel segmentation on a minimal training dataset
reaches state-of-the-art results. It helps us create a large, semi-manually
annotated brain vessel dataset of brain MRA images based on the IXI dataset
(annotated 200 images). Annotation was performed by three experts under the
supervision of three neurovascular surgeons after applying HessNet. It provides
high accuracy of vessel segmentation and allows experts to focus only on the
most complex important cases. The dataset is available at
https://git.scinalytics.com/terilat/VesselDatasetPartly.