Embedding Weighted Feature Aggregation Network with Domain Knowledge Integration for Breast Ultrasound Image Segmentation

Yuxi Liu,Xing An,Longfei Cong,Guohao Dong,Lei Zhu
DOI: https://doi.org/10.1007/978-3-030-60334-2_7
2020-01-01
Abstract:Breast cancer is the most common cancer in women, and ultrasound imaging is one of the most widely used approaches for diagnosis due to its non-radioactive process, ease of operation and low cost. Moreover, image segmentation plays a great role in medical image analysis, since it affects the accuracy of computer aided diagnosis (CAD) results. However, the malignant mass of breast in ultrasound images often appears irregular boundary and indistinct margin which is difficult to distinguish from other surrounding tissues. Therefore, breast ultrasound images segmentation is significant for diagnosis, and it has attracted the attention of researchers for many years. In this paper, we propose a weighted feature aggregation network with fusing domain knowledge for two-dimensional breast ultrasound images segmentation. (I) We modify the U-Net by adding a classification branch, in which BI-RADS category information is applied as the classification label. (II) In order to deal with the artifacts in ultrasound, such as posterior shadowing, we conduct Squeeze-and-Excitation (SE) block and aggregation mechanism to compose the up-sampling part in U-Net. (III) We employ the conditional random field (CRF) to optimize segmentation to make the boundaries more continuous and integral after getting the output of U-Net. The experiment conducted on a challenging and representative dataset includes more than three thousand two-dimensional breast ultrasound images. Our method achieves Jaccard Index of 84.9%, Matthew correlation coefficient of 90.9%, and Dice Coefficient of 90.8% in testing which demonstrates the potential clinical value of our work.
What problem does this paper attempt to address?