Abstract:Background: Breast ultrasound (BUS) imaging is one of the most prevalent approaches for the detection of breast cancers. Tumor segmentation of BUS images can facilitate doctors in localizing tumors and is a necessary step for computer-aided diagnosis systems. While the majority of clinical BUS scans are normal ones without tumors, segmentation approaches such as U-Net often predict mass regions for these images. Such false-positive problem becomes serious if a fully automatic artificial intelligence system is used for routine screening. Methods: In this study, we proposed a novel model which is more suitable for routine BUS screening. The model contains a classification branch that determines whether the image is normal or with tumors, and a segmentation branch that outlines tumors. Two branches share the same encoder network. We also built a new dataset that contains 1600 BUS images from 625 patients for training and a testing dataset with 130 images from 120 patients for testing. The dataset is the largest one with pixel-wise masks manually segmented by experienced radiologists. Our code is available at https://github.com/szhangNJU/BUS_segmentation. Results: The area under the receiver operating characteristic curve (AUC) for classifying images into normal/abnormal categories was 0.991. The dice similarity coefficient (DSC) for segmentation of mass regions was 0.898, better than the state-of-the-art models. Testing on an external dataset gave a similar performance, demonstrating a good transferability of our model. Moreover, we simulated the use of the model in actual clinic practice by processing videos recorded during BUS scans; the model gave very low false-positive predictions on normal images without sacrificing sensitivities for images with tumors. Conclusions: Our model achieved better segmentation performance than the state-of-the-art models and showed a good transferability on an external test set. The proposed deep learning architecture holds potential for use in fully automatic BUS health screening.

Mass segmentation in automated breast ultrasound using an enhanced attentive UNet

Tumor segmentation in automated whole breast ultrasound using bidirectional LSTM neural network and attention mechanism

ACL-DUNet: A tumor segmentation method based on multiple attention and densely connected breast ultrasound images

Attention based UNet model for breast cancer segmentation using BUSI dataset

Dilated Densely Connected U-Net with Uncertainty Focus Loss for 3D ABUS Mass Segmentation.

Mass detection in automated three dimensional breast ultrasound using cascaded convolutional neural networks

Cross-Model Attention-Guided Tumor Segmentation for 3D Automated Breast Ultrasound (ABUS) Images

Automatic Tumor Segmentation in 3D Automated Breast Ultrasound Using Convolutional Neural Network

Residual attention based uncertainty-guided mean teacher model for semi-supervised breast masses segmentation in 2D ultrasonography

CBAM-RIUnet: Breast Tumor Segmentation With Enhanced Breast Ultrasound and Test-Time Augmentation

Breast cancer: One-stage automated detection, segmentation, and classification of digital mammograms using UNet model based-semantic segmentation

WATUNet: A Deep Neural Network for Segmentation of Volumetric Sweep Imaging Ultrasound

Uncertainty Aware Temporal-Ensembling Model for Semi-Supervised ABUS Mass Segmentation

DAU-Net: Dual attention-aided U-Net for segmenting tumor in breast ultrasound images

Auto-DenseUNet: Searchable neural network architecture for mass segmentation in 3D automated breast ultrasound

Medical Breast Ultrasound Image Segmentation by Machine Learning

A Convolutional Neural Network-Based Auto-Segmentation Pipeline for Breast Cancer Imaging

Fully automatic tumor segmentation of breast ultrasound images with deep learning

DMAeEDNet: Dense Multiplicative Attention Enhanced Encoder Decoder Network for Ultrasound-Based Automated Breast Lesion Segmentation

Multi-task learning for segmentation and classification of breast tumors from ultrasound images