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

基于ASPP-坐标注意力机制的改进Unet模型用于脑肿瘤图像分割

Improved Unet model for brain tumor image segmentation based on ASPP-coordinate attention mechanism

Carnegie Mellon University(卡内基梅隆大学);New York University(纽约大学);North Carolina State University(北卡罗来纳州立大学)

Zixuan Wang, Yanlin Chen, Feiyang Wang, Qiaozhi Bao

AI总结:

针对脑肿瘤图像分割任务,提出结合坐标注意力机制与ASPP模块的改进Unet模型,实验证明其相比传统Unet在miou指标、分割效果及边缘精度上均有显著提升,为医学图像分析提供更可靠方法。

摘要:

本文提出了一种用于脑肿瘤图像分割的改进Unet模型,该模型结合了坐标注意力机制与ASPP(空洞空间金字塔池化)模块以提升分割效果。数据集划分完成后,我们对图像进行了必要的预处理,并使用改进后的模型开展实验。首先,我们对传统Unet模型进行了训练与验证。通过分析训练集与验证集的损失曲线可以发现,损失值在第1个epoch时持续下降,在第8个epoch时趋于稳定,这一过程表明模型在不断优化自身参数以提升性能。同时,miou(平均交并比)指标的变化显示,miou值在第15个epoch时超过0.6,此后始终保持在0.6以上,并在第46个epoch时达到0.7以上。这些结果表明基础Unet模型在脑肿瘤图像分割任务中具备有效性。接下来,我们引入了基于坐标注意力机制与ASPP模块的改进Unet算法进行实验。通过观察训练集与验证集的损失变化曲线发现,损失值在第6个epoch时达到最低点,随后保持相对稳定。同时,miou指标自第20个epoch起稳定在0.7以上,最高达到0.76。这些结果表明,引入的新机制显著提升了模型的分割能力。最后,我们将训练好的传统Unet模型与基于坐标注意力机制和ASPP模块的改进Unet模型应用于测试集,进行脑肿瘤图像分割预测。与传统Unet相比,增强后的模型具备更优的分割效果与边缘精度,结合坐标注意力机制与ASPP模块为医学图像分析提供了更可靠的方法。

英文摘要:

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.

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

↑