A hybrid attentional guidance network for tumors segmentation of breast ultrasound images

Int J Comput Assist Radiol Surg. 2023 Aug;18(8):1489-1500. doi: 10.1007/s11548-023-02849-7. Epub 2023 Feb 28.

Abstract

Purpose: In recent years, breast cancer has become the greatest threat to women. There are many studies dedicated to the precise segmentation of breast tumors, which is indispensable in computer-aided diagnosis. Deep neural networks have achieved accurate segmentation of images. However, convolutional layers are biased to extract local features and tend to lose global and location information as the network deepens, which leads to a decrease in breast tumors segmentation accuracy. For this reason, we propose a hybrid attention-guided network (HAG-Net). We believe that this method will improve the detection rate and segmentation of tumors in breast ultrasound images.

Methods: The method is equipped with multi-scale guidance block (MSG) for guiding the extraction of low-resolution location information. Short multi-head self-attention (S-MHSA) and convolutional block attention module are used to capture global features and long-range dependencies. Finally, the segmentation results are obtained by fusing multi-scale contextual information.

Results: We compare with 7 state-of-the-art methods on two publicly available datasets through five random fivefold cross-validations. The highest dice coefficient, Jaccard Index and detect rate ([Formula: see text]%, [Formula: see text]%, [Formula: see text]% and [Formula: see text]%, [Formula: see text]%, [Formula: see text]%, separately) obtained on two publicly available datasets(BUSI and OASUBD), prove the superiority of our method.

Conclusion: HAG-Net can better utilize multi-resolution features to localize the breast tumors. Demonstrating excellent generalizability and applicability for breast tumors segmentation compare to other state-of-the-art methods.

Keywords: Attentional mechanisms; Breast tumors; Long-range dependences; Ultrasound images.

MeSH terms

  • Breast Neoplasms* / diagnostic imaging
  • Diagnosis, Computer-Assisted
  • Female
  • Humans
  • Image Processing, Computer-Assisted* / methods
  • Neural Networks, Computer
  • Ultrasonography, Mammary