arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2312.11467eess.IVcs.CVcs.LG

基于视觉Transformer集成的胶质母细胞瘤肿瘤分割

Glioblastoma Tumor Segmentation using an Ensemble of Vision Transformers

Rice University(莱斯大学)

Huafeng Liu, Benjamin Dowdell, Todd Engelder, Zarah Pulmano, Nicolas Osa, Arko Barman

AI总结:

针对胶质母细胞瘤MRI精准分割需求,提出BRAINNET流程,通过三方向训练的MaskFormer模型集成预测,在UPenn-GBM数据集上实现三类肿瘤区域分割的SOTA结果。

摘要:

胶质母细胞瘤是最具侵袭性、致死率最高的脑癌类型之一,与其他癌症类型相比存活率更低。磁共振成像(MRI)扫描分析是诊断和治疗胶质母细胞瘤等脑癌的最有效方法之一。MRI图像中的精准肿瘤分割通常是治疗规划和治疗方法风险评估的必要环节。本文提出了一种名为BRAINNET(智能神经网络辅助脑部放射学)的新型流程,该流程借助视觉Transformer模型MaskFormer生成鲁棒的肿瘤分割掩码。我们对3D脑部MRI体数据的三个正交2D切片方向(轴位、矢状位、冠状位)分别训练三个模型,再将九个预测结果进行集成。我们在公开的UPenn-GBM数据集上训练并测试模型,该数据集包含611名受试者的3D多参数MRI(mpMRI)扫描数据。使用Dice系数(DC)和95%豪斯多夫距离(HD)进行评估,我们的模型在分割所有三个不同肿瘤区域时均取得了当前最优结果——肿瘤核心(DC=0.894,HD=2.308)、整个肿瘤(DC=0.891,HD=3.552)以及强化肿瘤(DC=0.812,HD=1.608)。

英文摘要:

Glioblastoma is one of the most aggressive and deadliest types of brain cancer, with low survival rates compared to other types of cancer. Analysis of Magnetic Resonance Imaging (MRI) scans is one of the most effective methods for the diagnosis and treatment of brain cancers such as glioblastoma. Accurate tumor segmentation in MRI images is often required for treatment planning and risk assessment of treatment methods. Here, we propose a novel pipeline, Brain Radiology Aided by Intelligent Neural NETworks (BRAINNET), which leverages MaskFormer, a vision transformer model, and generates robust tumor segmentation maks. We use an ensemble of nine predictions from three models separately trained on each of the three orthogonal 2D slice directions (axial, sagittal, and coronal) of a 3D brain MRI volume. We train and test our models on the publicly available UPenn-GBM dataset, consisting of 3D multi-parametric MRI (mpMRI) scans from 611 subjects. Using Dice coefficient (DC) and 95% Hausdorff distance (HD) for evaluation, our models achieved state-of-the-art results in segmenting all three different tumor regions -- tumor core (DC = 0.894, HD = 2.308), whole tumor (DC = 0.891, HD = 3.552), and enhancing tumor (DC = 0.812, HD = 1.608).

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑