融合Xception与自注意力的新型网络,提升脑瘤分割精度。
Attention Xception UNet (AXUNet): A Novel Combination of CNN and Self-Attention for Brain Tumor Segmentation
- 用Xception主干加自注意力模块改进UNet结构
- 平均Dice达93.73,优于现有模型
- 适合需要高精度肿瘤分割的研究者
准确分割胶质瘤脑肿瘤对诊断和治疗规划至关重要。深度学习提供了有前景的解决方案,但最优模型架构仍待探索。本研究采用BraTS 2021数据集,选取增强T1(T1CE)、T2及液体抑制反转恢复(FLAIR)序列进行模型开发。提出的注意力Xception UNet(AXUNet)架构将Xception主干与点积自注意力模块结合,灵感源自Google Bard、OpenAI ChatGPT等先进大语言模型,嵌入UNet结构中。与当前最优模型对比,测试集评估显示性能提升:Inception-UNet与Xception-UNet的平均Dice分别为90.88和93.24;Attention ResUNet(AResUNet)达92.80,其中强化肿瘤(ET)最高为84.92;Attention Gate UNet(AGUNet)为90.38。AXUNet表现最佳,平均Dice达93.73,在全肿瘤(WT)与肿瘤核心(TC)区域分别达到92.59与86.81,强化肿瘤为84.89。该设计在捕捉空间与上下文信息方面显著提升,表明其在精准肿瘤勾画中的应用潜力。
原文摘要 · Abstract (English)
Accurate segmentation of glioma brain tumors is crucial for diagnosis and treatment planning. Deep learning techniques offer promising solutions, but optimal model architectures remain under investigation. We used the BraTS 2021 dataset, selecting T1 with contrast enhancement (T1CE), T2, and Fluid-Attenuated Inversion Recovery (FLAIR) sequences for model development. The proposed Attention Xception UNet (AXUNet) architecture integrates an Xception backbone with dot-product self-attention modules, inspired by state-of-the-art (SOTA) large language models such as Google Bard and OpenAI ChatGPT, within a UNet-shaped model. We compared AXUNet with SOTA models. Comparative evaluation on the test set demonstrated improved results over baseline models. Inception-UNet and Xception-UNet achieved mean Dice scores of 90.88 and 93.24, respectively. Attention ResUNet (AResUNet) attained a mean Dice score of 92.80, with the highest score of 84.92 for enhancing tumor (ET) among all models. Attention Gate UNet (AGUNet) yielded a mean Dice score of 90.38. AXUNet outperformed all models with a mean Dice score of 93.73. It demonstrated superior Dice scores across whole tumor (WT) and tumor core (TC) regions, achieving 92.59 for WT, 86.81 for TC, and 84.89 for ET. The integration of the Xception backbone and dot-product self-attention mechanisms in AXUNet showcases enhanced performance in capturing spatial and contextual information. The findings underscore the potential utility of AXUNet in facilitating precise tumor delineation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。