Tumor detection using deep learning method in automated breast ultrasound

Abstract

ObjectiveAccurate and efficient breast tumor detection of automated breast ultrasound (ABUS) plays a significant role for clinical diagnosis. However, reviewing thousands of ABUS slices is extremely a time-consuming task. Besides, speckle, shadowing, non-uniform contrast of certain structures and high variability of the lesion echogenicity, make tumor detection in ABUS images difficult and prone to false positive (FP) regions. To improve the tumor detection rate and reduce FP regions, a Bayesian YOLOv4 network based on Monte Carlo dropout (MC-Drop) is proposed. MethodsFirstly, some tricks in YOLOv4 network are adjusted to make it more suitable for ABUS tumor detection. Secondly, MC-Drop is applied to introduce uncertainty into the detection network and the Bayesian YOLOv4 network is constructed, which effectively reduces accidental FP regions and improves the detection rate for hard tumor region. ResultsThis method was tested on 68 volumes, including 21,624 slices (1683 tumor slices and 19,941 normal slices). It obtains a promising result with sensitivity of 0.88 and FPs/S at 0.19. ConclusionExperimental results demonstrate that our Bayesian YOLOv4 outperforms the original YOLOv4 network and some other mainstream object detectors.