Yanfeng Li1, Wen Wu1, Houjin Chen1, Lin Cheng2, Shu Wang2. 1. School of Electronic and Information Engineering, Beijing Jiaotong University, Beijing, China. 2. Center for Breast, People's Hospital of Peking University, Beijing, China.
Abstract
PURPOSE: Automated breast ultrasound (ABUS) has drawn attention in breast disease detection and diagnosis applications. Reviewing hundreds of slices produced by ABUS is time-consuming. In this paper, a tumor detection method for ABUS image based on convolutional neural network is proposed. METHODS: First, integrating multitask learning with YOLOv3, an improved YOLOv3 detection network is designed to detect tumor candidate in two-dimensional (2D) slices. Two-dimensional detection separately treats each slice, leading to larger differences of position and score for tumor candidate in adjacent slices. Due to the influence of artifact, noise, and mammary tissues, 2D detection may include many false positive regions. To alleviate these problems, a rescoring processing algorithm is first designed. Then three-dimensional volume forming and FP reduction scheme are built. RESULTS: This method was tested on 340 volumes (124 patients, 181 tumors) with fivefold cross validation. It achieved sensitivities of 90%, 85%, 80%, 75%, and 70% at 7.42, 3.31, 1.62, 1.23, and 0.88 false positives per volume. CONCLUSION: Compared with existing ABUS tumor detection methods, our method gets a promising result.
PURPOSE: Automated breast ultrasound (ABUS) has drawn attention in breast disease detection and diagnosis applications. Reviewing hundreds of slices produced by ABUS is time-consuming. In this paper, a tumor detection method for ABUS image based on convolutional neural network is proposed. METHODS: First, integrating multitask learning with YOLOv3, an improved YOLOv3 detection network is designed to detect tumor candidate in two-dimensional (2D) slices. Two-dimensional detection separately treats each slice, leading to larger differences of position and score for tumor candidate in adjacent slices. Due to the influence of artifact, noise, and mammary tissues, 2D detection may include many false positive regions. To alleviate these problems, a rescoring processing algorithm is first designed. Then three-dimensional volume forming and FP reduction scheme are built. RESULTS: This method was tested on 340 volumes (124 patients, 181 tumors) with fivefold cross validation. It achieved sensitivities of 90%, 85%, 80%, 75%, and 70% at 7.42, 3.31, 1.62, 1.23, and 0.88 false positives per volume. CONCLUSION: Compared with existing ABUS tumor detection methods, our method gets a promising result.