Journal of Hebei University(Natural Science Edition) ›› 2022, Vol. 42 ›› Issue (5): 542-551.DOI: 10.3969/j.issn.1000-1565.2022.05.013

Previous Articles     Next Articles

Segmentation algorithm of brain tumor MR images based on receptor field amplification and attentional mechanism

ZHENG Wei1, ZHAO Jinfang1, ZHANG Yijing1, LIU Shuaiqi1, ZHANG Xiaodan2, MA Zepeng2   

  1. 1. College of Electronic and Information Engineering, Hebei University, Baoding 071002, China; 2. Department of Ultrasonography, Affilicated Hospital of Hebei University, Baoding 071000, China
  • Received:2021-06-11 Online:2022-09-25 Published:2022-10-19

Abstract: Although the current full convolutional neural network has been able to accurately identify the location of brain tumors in the brain tumor segmentation task, there are still problems of low segmentation accuracy due to limited receptive field and information loss. To solve these problems, this paper proposes a U-Net brain tumor MR image segmentation algorithm based on receptive field amplification and attentional mechanism. Firstly, the receptive field block(RFB)is introduced into the U-Net network to increase the receptive field of the network, which solves the problem of low segmentation accuracy caused by the limited receptive field. In addition, an efficient channel attention module(ECA)is introduced to increase the networks response to useful features and suppress redundant features in the network. The MR brain tumor image data provided by the BRATS(the brain tumor image segmentation- DOI:10.3969/j.issn.1000-1565.2022.05.013基于感受野扩增和注意力机制的U-Net脑肿瘤MR图像分割郑伟1,赵金芳1,张奕婧1,刘帅奇1,张晓丹2,马泽鹏2(1. 河北大学 电子信息工程学院,河北 保定 071002;2. 河北大学附属医院 超声诊断科,河北 保定 071000)摘 要:针对U-Net网络感受野受限以及信息丢失导致的分割精度低的问题,提出了一种基于感受野扩增和注意力机制的U-Net脑肿瘤MR图像分割算法.首先,在U-Net网络中引入感受野模块(receptive field block,RFB)来增大网络的感受野,解决了网络由于感受野受限带来的分割精度低的问题.此外在网络中引入有效的通道注意模块(efficient channel attention,ECA)来增加网络对有用特征的响应,抑制网络中的冗余特征.使用BraTS(the brain tumor image segmentation challenge)提供的脑肿瘤MR图像数据对本文算法进行测试,用Dice相似性系数等指标进行评价,结果显示在完整肿瘤、核心肿瘤以及增强肿瘤的Dice值分别可达到0.86、0.86、0.79.与U-Net模型以及其他的网络相比得到了提高.实验结果表明,本文提出的算法能够有效提升脑肿瘤分割的精度,具有良好的分割性能.关键词:脑肿瘤分割;U-Net;RFB;注意力机制 中图分类号:TN911 文献标志码:A 文章编号:1000-1565(2022)05-0542-10Segmentation algorithm of brain tumor MR images based on receptor field amplification and attentional mechanismZHENG Wei1, ZHAO Jinfang1, ZHANG Yijing1, LIU Shuaiqi1, ZHANG Xiaodan2, MA Zepeng2(1. College of Electronic and Information Engineering, Hebei University, Baoding 071002, China; 2. Department of Ultrasonography, Affilicated Hospital of Hebei University, Baoding 071000, China )Abstract: Although the current full convolutional neural network has been able to accurately identify the location of brain tumors in the brain tumor segmentation task, there are still problems of low segmentation accuracy due to limited receptive field and information loss. To solve these problems, this paper proposes a U-Net brain tumor MR image segmentation algorithm based on receptive field amplification and attentional mechanism. Firstly, the receptive field block(RFB)is introduced into the U-Net network to increase the receptive field of the network, which solves the problem of low segmentation accuracy caused by the limited receptive field. In addition, an efficient channel attention module(ECA)is introduced to increase the networks response to useful features and suppress redundant features in the network. The MR brain tumor image data provided by the BRATS(the brain tumor image segmentation- 收稿日期:2021-06-11 基金项目:河北省自然科学基金资助项目(F2020201025;H2020201021);河北省高等学校科学技术研究项目(BJ2020030) 第一作者:郑伟(1972—),女,黑龙江兰西人,河北大学教授,博士,主要从事图像处理与分析、图像安全通信、图像加密和隐藏方向研究.E-mail:147685650@qq.com 通信作者:张晓丹(1989—),女,河北承德人,河北大学附属医院主治医师,主要从事常规疾病及肿瘤病变的影像诊断.E-mail:zxd201505@126.com第5期郑伟等:基于感受野扩增和注意力机制的U-Net脑肿瘤MR图像分割challenge)was used to test the method, and indexes such as Dice similarity coefficient were used to evaluate the method. The results showed that the Dice of intact tumor, core tumor and enhanced tumor could reach 0.86, 0.86 and 0.79 respectively. Compared with the U-Net model and other networks, this method is improved. The results show that the proposed method can effectively improve the accuracy of brain tumor segmentation and has good segmentation performance.

Key words: brain tumor segmentation, U-Net, RFB, attentional mechanism

CLC Number: