用血管先验信息提升脑动脉瘤弱监督检测与分割精度
Weakly Supervised Intracranial Aneurysm Detection and Segmentation in MR angiography via Multi-task UNet with Vesselness Prior
- 设计多任务UNet融合弗朗吉血管性滤波器作为先验
- 在Lausanne数据集上达Dice=0.614,敏感度92.9%
- 仅需粗略标注即可训练,适合标注稀缺场景
脑动脉瘤(IA)是颅内血管的异常扩张,破裂后可导致致命后果。然而,其尺寸小、影像对比度低,使得准确高效地检测与形态分析困难,而缺乏带体素级专家标注的大规模公开数据集进一步制约了深度学习算法的发展。为此,我们提出一种新型弱监督3D多任务UNet,结合血管性先验,在时间飞跃磁共振血管成像(TOF-MRA)中联合完成动脉瘤检测与分割。具体而言,采用流行的Frangi血管性滤波器生成软脑血管先验,用于网络输入及解码器中的注意力模块,分别支持分割与辅助分支的检测。模型在包含粗标注的洛桑数据集上训练,并在同库经细化标注的测试集上评估;为验证泛化能力,还外推至ADAM数据集。结果表明,该方法在动脉瘤分割(Dice=0.614,95%HD=1.38mm)和检测(假阳性率=1.47,敏感度=92.9%)方面均优于现有最先进水平。
原文摘要 · Abstract (English)
Intracranial aneurysms (IAs) are abnormal dilations of cerebral blood vessels that, if ruptured, can lead to life-threatening consequences. However, their small size and soft contrast in radiological scans often make it difficult to perform accurate and efficient detection and morphological analyses, which are critical in the clinical care of the disorder. Furthermore, the lack of large public datasets with voxel-wise expert annotations pose challenges for developing deep learning algorithms to address the issues. Therefore, we proposed a novel weakly supervised 3D multi-task UNet that integrates vesselness priors to jointly perform aneurysm detection and segmentation in time-of-flight MR angiography (TOF-MRA). Specifically, to robustly guide IA detection and segmentation, we employ the popular Frangi's vesselness filter to derive soft cerebrovascular priors for both network input and an attention block to conduct segmentation from the decoder and detection from an auxiliary branch. We train our model on the Lausanne dataset with coarse ground truth segmentation, and evaluate it on the test set with refined labels from the same database. To further assess our model's generalizability, we also validate it externally on the ADAM dataset. Our results demonstrate the superior performance of the proposed technique over the SOTA techniques for aneurysm segmentation (Dice = 0.614, 95%HD =1.38mm) and detection (false positive rate = 1.47, sensitivity = 92.9%).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。