arXiv:2501.10615cs.CV2025-01被引 5

用分层LoG贝叶斯网络提升主动脉血管分割精度

Hierarchical LoG Bayesian Neural Network for Enhanced Aorta Segmentation

  • 分两路:U-Net初分割+分层LoG自适应检测多尺度血管
  • 在两个数据集上Dice系数提升至少3%,优于当前最优方法
  • 输出置信区间,适合需要可靠医疗决策的临床分析

主动脉及其分支的精准分割对主动脉疾病诊断至关重要。尽管深度学习已显著提升分割效果,但因多尺度结构复杂及周围组织干扰,仍具挑战。本文提出一种基于贝叶斯神经网络的分层拉普拉斯高斯(LoG)模型,由3D U-Net流和分层LoG流组成:前者提供初始分割,后者通过学习合适的LoG核,在不同尺度下增强血管检测能力,实现对主动脉各部分的自适应处理。采用贝叶斯方法参数化LoG流,为分割结果提供置信区间,确保预测的鲁棒性与可靠性。实验表明,该模型可准确分割主干及头臂血管,在两个主动脉数据集的多个体积上,Dice系数较现有最优方法至少提升3%,并能为不同部位提供可靠的置信区间。代码已开源:https://github.com/adlsn/LoGBNet。

原文摘要 · Abstract (English)

Accurate segmentation of the aorta and its associated arch branches is crucial for diagnosing aortic diseases. While deep learning techniques have significantly improved aorta segmentation, they remain challenging due to the intricate multiscale structure and the complexity of the surrounding tissues. This paper presents a novel approach for enhancing aorta segmentation using a Bayesian neural network-based hierarchical Laplacian of Gaussian (LoG) model. Our model consists of a 3D U-Net stream and a hierarchical LoG stream: the former provides an initial aorta segmentation, and the latter enhances blood vessel detection across varying scales by learning suitable LoG kernels, enabling self-adaptive handling of different parts of the aorta vessels with significant scale differences. We employ a Bayesian method to parameterize the LoG stream and provide confidence intervals for the segmentation results, ensuring robustness and reliability of the prediction for vascular medical image analysts. Experimental results show that our model can accurately segment main and supra-aortic vessels, yielding at least a 3% gain in the Dice coefficient over state-of-the-art methods across multiple volumes drawn from two aorta datasets, and can provide reliable confidence intervals for different parts of the aorta. The code is available at https://github.com/adlsn/LoGBNet.

主动脉分割贝叶斯网络多尺度检测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。