基于深度学习的乳腺超声图像BI-RADS五分类方法

BI-RADS five classification method of breast ultrasound images based on deep learning

  • 摘要: 基于乳腺超声图像研究了一种深度学习自动分类算法TDS-Net,用于实现乳腺影像报告和数据系统(BI-RADS)3、4a、4b、4c、5级的五分类. TDS-Net设计了双支路的结构:首先,第一条支路采用提出的DDModule叠加构成,该模块能够减少超声图像中的伪影并提取丰富的局部细节特征;其次,第二条支路由卷积块构成,它主要用于提取图像的全局特征信息,作为第一支路的信息补充;最后,将两条支路融合得到含有丰富特征信息的融合特征图,并采用深度可分离卷积和SENet进一步提取图中的信息, 其中,深度可分离卷积能够减少参数并增加网络的非线性进而增强其提取特征的能力,SENet注意力机制能增强高阶特征信息的提取. 为验证该算法,采用云南省肿瘤医院提供的数据进行实验,结果显示准确率、精准率、F1值分别为94.67%、94.81%、94.69%,均高于对比算法,体现了该算法的优越性. 同时为验证该算法的鲁棒性和普适性,基于两个公共数据集做了良恶性二分类的实验,实验结果同样高于对比算法. 这些结果表明,所提算法TDS-Net对乳腺超声图像具有较强的识别能力,有望应用于临床医学.

     

    Abstract: Based on breast ultrasound images, a deep learning automatic classification algorithm TDS-Net is studied, which is utilized to realize the five classification levels of 3, 4a, 4b, 4c and 5 in Breast Image Reporting and Data System (BI-RADS). TDS-Net designs a double branch structure. Firstly, the first branch is composed of the superposition of the DD Module proposed in this paper, which can reduce artifacts in ultrasonic images and extract rich local detail features. Secondly, the second branch is composed of convolution blocks, which is mainly used to extract the global feature information of the image as the information supplement of the first branch. Then, the two branches are fused to obtain a fused feature map containing rich feature information, and the information in the map is further extracted by deep separable convolution and SENet. Among them, deep separable convolution can reduce parameters and increase the nonlinearity of the network, to enhance its ability to extract features, and SENet attention mechanism can enhance the extraction of high-order feature information. To verify the algorithm, this paper utilizes the data provided by Yunnan Cancer Hospital for experiments. The results show that the accuracy, precision, and F1 values are 94.67%, 94.81%, and 94.69% respectively, which are higher than the comparison algorithm, reflecting the superiority of the algorithm. At the same time, to verify the robustness and universality of the algorithm, we experiment with benign and malignant binary classification based on two public datasets. The experimental results are also higher than the comparison algorithm. These results show that the proposed algorithm TDS-Net has a strong recognition ability for breast ultrasound images, and is expected to be applied in clinical medicine.

     

/

返回文章
返回