用注意力机制提升3D U-Net,精准分割脑肿瘤边界。
Advancing Brain Tumor Segmentation via Attention-based 3D U-Net Architecture and Digital Image Processing
- 在3D U-Net中加入注意力模块,聚焦关键区域
- 在BraTS2020上达Dice 0.975、敏感度0.995
- 适合医学影像分析与临床辅助诊断研究者
在医疗诊断领域,人工智能的快速发展显著提升了脑肿瘤分割性能。编码器-解码器架构如U-Net在从磁共振成像(MRI)中提取3D脑肿瘤特征方面发挥了重要作用。然而,标准U-Net在处理不规则形状和模糊边界时仍存在精度不足的问题。此外,在高分辨率MRI数据(如BraTS数据集)上训练鲁棒分割模型需大量计算资源,并面临类别不平衡挑战。本研究将注意力机制融入3D U-Net,使模型能捕捉细微结构并优先关注信息量高的区域。同时,采用基于数字图像处理的肿瘤检测算法缓解训练数据不平衡问题。在BraTS 2020数据集上的评估显示,该模型性能优于已有研究,达到Dice系数0.975、特异性0.988、敏感度0.995,验证了其在提升脑肿瘤分割可靠性方面的有效性,为临床诊断提供有力支持。
原文摘要 · Abstract (English)
In the realm of medical diagnostics, rapid advancements in Artificial Intelligence (AI) have significantly yielded remarkable improvements in brain tumor segmentation. Encoder-Decoder architectures, such as U-Net, have played a transformative role by effectively extracting meaningful representations in 3D brain tumor segmentation from Magnetic resonance imaging (MRI) scans. However, standard U-Net models encounter challenges in accurately delineating tumor regions, especially when dealing with irregular shapes and ambiguous boundaries. Additionally, training robust segmentation models on high-resolution MRI data, such as the BraTS datasets, necessitates high computational resources and often faces challenges associated with class imbalance. This study proposes the integration of the attention mechanism into the 3D U-Net model, enabling the model to capture intricate details and prioritize informative regions during the segmentation process. Additionally, a tumor detection algorithm based on digital image processing techniques is utilized to address the issue of imbalanced training data and mitigate bias. This study aims to enhance the performance of brain tumor segmentation, ultimately improving the reliability of diagnosis. The proposed model is thoroughly evaluated and assessed on the BraTS 2020 dataset using various performance metrics to accomplish this goal. The obtained results indicate that the model outperformed related studies, exhibiting dice of 0.975, specificity of 0.988, and sensitivity of 0.995, indicating the efficacy of the proposed model in improving brain tumor segmentation, offering valuable insights for reliable diagnosis in clinical settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。