arXiv:2409.13626eess.IVcs.CV2024-09中稿 · CONF-CDS 2024 conf…被引 21

用GSConv和ECA注意力改进U-Net,提升脑肿瘤图像分割精度。

Improved Unet brain tumor image segmentation based on GSConv module and ECA attention mechanism

  • 引入GSConv与ECA注意力,增强多尺度特征提取与通道聚焦能力。
  • mIoU在35轮后稳定达到0.8,训练损失8轮后快速收敛。
  • 尤其改善肿瘤边缘分割,适合临床精准诊断需求。

本文提出一种基于U-Net架构的脑肿瘤医学图像分割改进模型,通过引入GSConv模块与ECA注意力机制,提升了模型在医学图像分割任务中的表现。该改进模型能更高效地提取与利用多尺度特征,并灵活关注重要通道,显著优化分割效果。实验表明,模型在训练集与测试集上的损失值在第8轮后迅速降至最低并趋于稳定,体现良好的学习与泛化能力;同时,平均交并比(mIoU)在第35轮后逐渐接近0.8并保持稳定。相比传统U-Net,该改进模型在脑肿瘤图像边缘处理上表现出更优的分割精度,不仅提升了医学图像分析的准确性,也为临床诊断提供了更可靠的技術支持。

原文摘要 · Abstract (English)

An improved model of medical image segmentation for brain tumor is discussed, which is a deep learning algorithm based on U-Net architecture. Based on the traditional U-Net, we introduce GSConv module and ECA attention mechanism to improve the performance of the model in medical image segmentation tasks. With these improvements, the new U-Net model is able to extract and utilize multi-scale features more efficiently while flexibly focusing on important channels, resulting in significantly improved segmentation results. During the experiment, the improved U-Net model is trained and evaluated systematically. By looking at the loss curves of the training set and the test set, we find that the loss values of both rapidly decline to the lowest point after the eighth epoch, and then gradually converge and stabilize. This shows that our model has good learning ability and generalization ability. In addition, by monitoring the change in the mean intersection ratio (mIoU), we can see that after the 35th epoch, the mIoU gradually approaches 0.8 and remains stable, which further validates the model. Compared with the traditional U-Net, the improved version based on GSConv module and ECA attention mechanism shows obvious advantages in segmentation effect. Especially in the processing of brain tumor image edges, the improved model can provide more accurate segmentation results. This achievement not only improves the accuracy of medical image analysis, but also provides more reliable technical support for clinical diagnosis.

脑肿瘤分割U-Net改进注意力机制医学图像

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