融合坐标注意力与ASPP的改进Unet,提升脑肿瘤分割精度。
Improved Unet model for brain tumor image segmentation based on ASPP-coordinate attention mechanism
- 引入坐标注意力与ASPP模块,增强特征捕捉能力
- miou达0.76,较传统Unet提升显著
- 适合医学图像分割任务,尤其关注边缘细节
本文提出一种改进的Unet模型用于脑肿瘤图像分割,结合坐标注意力机制与ASPP模块以提升分割效果。数据集经预处理后,首先训练并验证传统Unet模型:损失值在第1轮持续下降,第8轮趋于稳定;miou(平均交并比)在第15轮超过0.6,第46轮达到0.7以上,表明基础模型有效。随后引入基于坐标注意力与ASPP模块的改进Unet进行实验:训练与验证集损失值在第6轮达最低并保持稳定,miou自第20轮起稳定高于0.7,最高达0.76,表明新机制显著提升模型性能。最后,在测试集上对比传统与改进模型,改进模型在分割结果与边缘准确性上均更优,证明该组合机制能有效提升脑肿瘤图像分割表现。
原文摘要 · Abstract (English)
In this paper, we propose an improved Unet model for brain tumor image segmentation, which combines coordinate attention mechanism and ASPP module to improve the segmentation effect. After the data set is divided, we do the necessary preprocessing to the image and use the improved model to experiment. First, we trained and validated the traditional Unet model. By analyzing the loss curve of the training set and the validation set, we can see that the loss value continues to decline at the first epoch and becomes stable at the eighth epoch. This process shows that the model constantly optimizes its parameters to improve performance. At the same time, the change in the miou (mean Intersection over Union) index shows that the miou value exceeded 0.6 at the 15th epoch, remained above 0.6 thereafter, and reached above 0.7 at the 46th epoch. These results indicate that the basic Unet model is effective in brain tumor image segmentation. Next, we introduce an improved Unet algorithm based on coordinate attention mechanism and ASPP module for experiments. By observing the loss change curves of the training set and the verification set, it is found that the loss value reaches the lowest point at the sixth epoch and then remains relatively stable. At the same time, the miou indicator has stabilized above 0.7 since the 20th epoch and has reached a maximum of 0.76. These results show that the new mechanism introduced significantly improves the segmentation ability of the model. Finally, we apply the trained traditional Unet model and the improved Unet model based on the coordinate attention mechanism and ASPP module to the test set for brain tumor image segmentation prediction. Compared to the traditional Unet, the enhanced model offers superior segmentation and edge accuracy, providing a more reliable method for medical image analysis with the coordinate attention mechanism and ASPP module.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。