Comparison of multiple feature extractors on Faster RCNN for breast tumor detection

Abstract

The Deep learning algorithm shows is strong capability in pattern recognition tasks such as object detection and speech recognition. Comparing with the typical machine learning methods which require extraction of manual features, deep learning algorithm has a more powerful feature learning ability. In this paper, the deep learning associated object detection method is applied to developed to locate and classify lesions for the detection of medical breast masses. At the same time, transfer learning based on the network of Faster RCNN is also introduced. Furthermore, five feature extractors of the network, which are ResNet101, inception V2, inception V3, Mobilenet, and inception ResNet V2, are investigated for exploring the impact for the model. Digital Database for Screening Mammography(DDSM) dataset is used in the investigation, and the performances of the models associated with the five feature extractors are compared separately in detecting benign and malignant breasts, based on the ROC trade-off curves. The simulation results demonstrate that the classification model with Inception ResNet V2 feature extractor exhibit the best performance, compared with the other four feature extractors.