用海森矩阵设计轻量网络,小样本下实现脑血管分割新精度
Hessian-Based Lightweight Neural Network HessNet for State-of-the-Art Brain Vessel Segmentation on a Minimal Training Dataset
- 基于海森矩阵构建6000参数轻量网络,支持CPU运行
- 仅用200张标注图像即达顶尖分割效果
- 适合资源有限但需高精度脑血管分割的研究者
准确分割脑磁共振血管成像(MRA)中的血管对动脉瘤修复或搭桥手术至关重要。当前主要依赖人工标注或弗朗吉滤波等传统方法,精度不足。神经网络虽在医学图像分割中表现优异,但依赖大量标注数据,而公开的带精细血管标注的MRA数据集稀缺。为此,我们提出一种基于海森矩阵的半监督轻量级3D神经网络HessNet,专用于复杂管状结构分割。该模型仅含6000个参数,可在CPU上运行,显著降低训练资源需求。在仅200张图像的最小训练集上,其血管分割精度达到当前最优水平。基于HessNet,我们借助三位专家在三位神经血管外科医生监督下,完成了从IXI数据集衍生出的大规模半自动标注脑血管数据集(共200幅图像)。该方法大幅减少专家工作量,使其聚焦于最复杂的病例。数据集已公开:https://git.scinalytics.com/terilat/VesselDatasetPartly。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。