Detection and Classification of Breast Masses in Digital Mammograms

Automatic detection and classification of the masses in mammograms are still a big challenge and play a crucial role to assist radiologists for accurate diagnosis. In this paper, we propose a novel computer-aided diagnosis (CAD) system based on one of the regional deep learning techniques, a ROI-based Convolutional Neural Network (CNN) which is called You Only Look Once (YOLO). Although most previous studies only deal with classification of masses, our proposed YOLO-based CAD system can handle detection and classification simultaneously in one framework. The proposed CAD system contains four main stages: pre-processing of mammograms, feature extraction utilizing deep convolutional networks, mass detection with confidence, and finally mass classification using fully connected neural networks (FC-NNs).