Hessian-Based Lightweight Neural Network HessNet for State-of-the-Art Brain Vessel Segmentation on a Minimal Training Dataset
2508.15660v2
eess.IV, cs.CV, I.4.6; I.5.4; J.3
2025-08-26
Авторы:
Alexandra Bernadotte, Elfimov Nikita, Mikhail Shutov, Ivan Menshikov
Резюме на русском
## Контекст
Успешная и точная сегментация кровеносных сосудов в головном мозгу человека, полученных с помощью МРА (магнитно-резонансной ангиографии), является критически важной задачей в медицинских процедурах, таких как ремонт аневризм, операции зеленого шоссе и другие важные хирургические вмешательства. Однако текущие методы сегментации, такие как метод Франги или ручная сегментация, либо требуют большого количества ресурсов для ручной работы, либо недостаточно точны. Нейронные сети предлагаются как мощные инструменты для автоматизации этой задачи, но их развитие ограничено отсутствием доступных для обучения больших аннотированных данных. Настоящая работа адресует эту проблему, предлагая метод, который обеспечивает высокоточную сегментацию кровеносных сосудов с минимальными требованиями к обучающим данным.
## Метод
Мы предлагаем HessNet — новую легковесную нейронную сеть, основанную на методах Гессиана. Она использует 6000 параметров и может работать даже на процессоре (без видеокарты). Метод включает в себя новый подход к сегментации, основанный на Гессианских матрицах, который позволяет эффективно обнаруживать и сегментировать комплексные трубчатые структуры. Такой подход позволяет использовать минимальное количество тренировочных данных, не снижая качество результата. Также в рамках проекта проведена аннотация 200 изображений МРА с помощью HessNet, после чего эти данные были расширены с помощью трех экспертов под наблюдением трех васкулярных хирургов.
## Результаты
Выполненные эксперименты показали, что HessNet достигает современных результатов по точности сегментации кровеносных сосудов, даже при использовании минимального количества тренировочных данных. На основе 200 аннотированных изображений была построена большая выборка данных, полученная через семиуровное расширение под эгидой экспертов. Эта выборка позволяет значительно уменьшить затраты на ручную аннотацию, снизить требования к вычислительным ресурсам и увеличить точность сегментации.
## Значимость
Метод 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.