An MRI brain tumor segmentation method based on improved U-Net
In order to improve the segmentation effect of brain tumor images and address the issue of feature information loss during convolutional neural network (CNN) training, we present an MRI brain tumor segmentation method that leverages an enhanced U-Net architecture. First, the ResNet50 network was use...
Main Authors: | , , |
---|---|
Format: | Article |
Language: | English |
Published: |
AIMS Press
2024-01-01
|
Series: | Mathematical Biosciences and Engineering |
Subjects: | |
Online Access: | https://www.aimspress.com/article/doi/10.3934/mbe.2024033?viewType=HTML |
_version_ | 1797339231427756032 |
---|---|
author | Jiajun Zhu Rui Zhang Haifei Zhang |
author_facet | Jiajun Zhu Rui Zhang Haifei Zhang |
author_sort | Jiajun Zhu |
collection | DOAJ |
description | In order to improve the segmentation effect of brain tumor images and address the issue of feature information loss during convolutional neural network (CNN) training, we present an MRI brain tumor segmentation method that leverages an enhanced U-Net architecture. First, the ResNet50 network was used as the backbone network of the improved U-Net, the deeper CNN can improve the feature extraction effect. Next, the Residual Module was enhanced by incorporating the Convolutional Block Attention Module (CBAM). To increase characterization capabilities, focus on important features and suppress unnecessary features. Finally, the cross-entropy loss function and the Dice similarity coefficient are mixed to compose the loss function of the network. To solve the class unbalance problem of the data and enhance the tumor area segmentation outcome. The method's segmentation performance was evaluated using the test set. In this test set, the enhanced U-Net achieved an average Intersection over Union (IoU) of 86.64% and a Dice evaluation score of 87.47%. These values were 3.13% and 2.06% higher, respectively, compared to the original U-Net and R-Unet models. Consequently, the proposed enhanced U-Net in this study significantly improves the brain tumor segmentation efficacy, offering valuable technical support for MRI diagnosis and treatment. |
first_indexed | 2024-03-08T09:42:57Z |
format | Article |
id | doaj.art-a2ca2ff8b16c467b8fe7979e56d72383 |
institution | Directory Open Access Journal |
issn | 1551-0018 |
language | English |
last_indexed | 2024-03-08T09:42:57Z |
publishDate | 2024-01-01 |
publisher | AIMS Press |
record_format | Article |
series | Mathematical Biosciences and Engineering |
spelling | doaj.art-a2ca2ff8b16c467b8fe7979e56d723832024-01-30T01:23:13ZengAIMS PressMathematical Biosciences and Engineering1551-00182024-01-0121177879110.3934/mbe.2024033An MRI brain tumor segmentation method based on improved U-NetJiajun Zhu0Rui Zhang 1Haifei Zhang 2School of Computer and Information Engineering, Nantong Institute of Technology, Nantong 226600, ChinaSchool of Computer and Information Engineering, Nantong Institute of Technology, Nantong 226600, ChinaSchool of Computer and Information Engineering, Nantong Institute of Technology, Nantong 226600, ChinaIn order to improve the segmentation effect of brain tumor images and address the issue of feature information loss during convolutional neural network (CNN) training, we present an MRI brain tumor segmentation method that leverages an enhanced U-Net architecture. First, the ResNet50 network was used as the backbone network of the improved U-Net, the deeper CNN can improve the feature extraction effect. Next, the Residual Module was enhanced by incorporating the Convolutional Block Attention Module (CBAM). To increase characterization capabilities, focus on important features and suppress unnecessary features. Finally, the cross-entropy loss function and the Dice similarity coefficient are mixed to compose the loss function of the network. To solve the class unbalance problem of the data and enhance the tumor area segmentation outcome. The method's segmentation performance was evaluated using the test set. In this test set, the enhanced U-Net achieved an average Intersection over Union (IoU) of 86.64% and a Dice evaluation score of 87.47%. These values were 3.13% and 2.06% higher, respectively, compared to the original U-Net and R-Unet models. Consequently, the proposed enhanced U-Net in this study significantly improves the brain tumor segmentation efficacy, offering valuable technical support for MRI diagnosis and treatment.https://www.aimspress.com/article/doi/10.3934/mbe.2024033?viewType=HTMLbrain tumor segmentationmrisemantic segmentationu-netcbam |
spellingShingle | Jiajun Zhu Rui Zhang Haifei Zhang An MRI brain tumor segmentation method based on improved U-Net Mathematical Biosciences and Engineering brain tumor segmentation mri semantic segmentation u-net cbam |
title | An MRI brain tumor segmentation method based on improved U-Net |
title_full | An MRI brain tumor segmentation method based on improved U-Net |
title_fullStr | An MRI brain tumor segmentation method based on improved U-Net |
title_full_unstemmed | An MRI brain tumor segmentation method based on improved U-Net |
title_short | An MRI brain tumor segmentation method based on improved U-Net |
title_sort | mri brain tumor segmentation method based on improved u net |
topic | brain tumor segmentation mri semantic segmentation u-net cbam |
url | https://www.aimspress.com/article/doi/10.3934/mbe.2024033?viewType=HTML |
work_keys_str_mv | AT jiajunzhu anmribraintumorsegmentationmethodbasedonimprovedunet AT ruizhang anmribraintumorsegmentationmethodbasedonimprovedunet AT haifeizhang anmribraintumorsegmentationmethodbasedonimprovedunet AT jiajunzhu mribraintumorsegmentationmethodbasedonimprovedunet AT ruizhang mribraintumorsegmentationmethodbasedonimprovedunet AT haifeizhang mribraintumorsegmentationmethodbasedonimprovedunet |