Breast mass detection in digital mammography based on anchor-free architecture

Background and Objective: Accurate detection of breast masses in mammography images is critical to diagnose early breast cancer, which can greatly improve the patients survival rate. However, it is still a big challenge due to the heterogeneity of breast masses and the complexity of their surrounding environment.Methods: To address these problems, we propose a one-stage object detection architecture, called Breast Mass Detection Network (BMassDNet), based on anchor-free and feature pyramid which makes the detection of breast masses of different sizes well adapted. We introduce a truncation normalization method and combine it with adaptive histogram equalization to enhance the contrast between the breast mass and the surrounding environment. Meanwhile, to solve the overfitting problem caused by small data size, we propose a natural deformation data augmentation method and mend the train data dynamic updating method based on the data complexity to effectively utilize the limited data. Finally, we use transfer learning to assist the training process and to improve the robustness of the model ulteriorly.Results: On the INbreast dataset, each image has an average of 0.495 false positives whilst the recall rate is 0.930; On the DDSM dataset, when each image has 0.599 false positives, the recall rate reaches 0.943.Conclusions: The experimental results on datasets INbreast and DDSM show that the proposed BMassDNet can obtain competitive detection performance over the current top ranked methods.

[1]  Ge Yu,et al.  Breast tumor detection in digital mammography based on extreme learning machine , 2014, Neurocomputing.

[2]  Hao Chen,et al.  FCOS: Fully Convolutional One-Stage Object Detection , 2019, 2019 IEEE/CVF International Conference on Computer Vision (ICCV).

[3]  V. R. Thool,et al.  Morphological operation and scaled Réyni entropy based approach for masses detection in mammograms , 2018, Multimedia Tools and Applications.

[4]  Xavier Lladó,et al.  Mass detection in mammograms using pre-trained deep learning models , 2018, Other Conferences.

[5]  Ross B. Girshick,et al.  Focal Loss for Dense Object Detection , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[6]  Gustavo Carneiro,et al.  Automated Mass Detection in Mammograms Using Cascaded Deep Learning and Random Forests , 2015, 2015 International Conference on Digital Image Computing: Techniques and Applications (DICTA).

[7]  M. Hanmandlu,et al.  A comparison of two methods for the segmentation of masses in the digital mammograms , 2010, Comput. Medical Imaging Graph..

[8]  A. Jemal,et al.  Cancer Statistics, 2008 , 2008, CA: a cancer journal for clinicians.

[9]  C. D'Orsi,et al.  Diagnostic Performance of Digital Versus Film Mammography for Breast-Cancer Screening , 2005, The New England journal of medicine.

[10]  T. Freer,et al.  Screening mammography with computer-aided detection: prospective study of 12,860 patients in a community breast center. , 2001, Radiology.

[11]  R E Hendrick,et al.  Benefit of mammography screening in women ages 40 to 49 Years. Current evidence from randomized controlled trials , 1995, Cancer.

[12]  Jaime S. Cardoso,et al.  INbreast: toward a full-field digital mammographic database. , 2012, Academic radiology.

[13]  Boaz Ophir,et al.  Automatic Dual-View Mass Detection in Full-Field Digital Mammograms , 2015, MICCAI.

[14]  Tsuyoshi Murata,et al.  {m , 1934, ACML.

[15]  Heng-Da Cheng,et al.  Approaches for automated detection and classification of masses in mammograms , 2006, Pattern Recognit..

[16]  Edwin Valarezo,et al.  Simultaneous Detection and Classification of Breast Masses in Digital Mammograms via a Deep Learning YOLO-based CAD System , 2018, Comput. Methods Programs Biomed..

[17]  J. Anitha,et al.  A dual stage adaptive thresholding (DuSAT) for automatic mass detection in mammograms , 2017, Comput. Methods Programs Biomed..

[18]  A. Jemal,et al.  Global cancer statistics 2018: GLOBOCAN estimates of incidence and mortality worldwide for 36 cancers in 185 countries , 2018, CA: a cancer journal for clinicians.

[19]  Yasser M. Kadah,et al.  An Automatic Computer-Aided Diagnosis System for Breast Cancer in Digital Mammograms via Deep Belief Network , 2018 .

[20]  Yuya Kajikawa,et al.  Computer-aided diagnosis: A survey with bibliometric analysis , 2017, Int. J. Medical Informatics.

[21]  Huilong Duan,et al.  A Model-based Algorithm for Mass Segmentation in Mammograms , 2005, 2005 IEEE Engineering in Medicine and Biology 27th Annual Conference.

[22]  István Csabai,et al.  Detecting and classifying lesions in mammograms with Deep Learning , 2017, Scientific Reports.

[23]  Ali Farhadi,et al.  YOLOv3: An Incremental Improvement , 2018, ArXiv.

[24]  Rangaraj M. Rangayyan,et al.  A review of computer-aided diagnosis of breast cancer: Toward the detection of subtle signs , 2007, J. Frankl. Inst..

[25]  N. Otsu.  A threshold selection method from gray level histograms , 1979 .

[26]  Anselmo Cardoso de Paiva,et al.  Automatic Detection of Masses in Mammograms Using Quality Threshold Clustering, Correlogram Function, and SVM , 2015, Journal of Digital Imaging.

[27]  Karel J. Zuiderveld,et al.  Contrast Limited Adaptive Histogram Equalization , 1994, Graphics Gems.

[28]  N. Dubrawsky.  Cancer statistics , 1989, CA: a cancer journal for clinicians.

[29]  Martin P DeSimio,et al.  A computer-aided detection system for the evaluation of breast cancer by mammographic appearance and lesion size. , 2005, AJR. American journal of roentgenology.

[30]  Arnau Oliver,et al.  A review of automatic mass detection and segmentation in mammographic images , 2010, Medical Image Anal..

[31]  M. A. Al-masni,et al.  Detection and classification of the breast abnormalities in digital mammograms via regional Convolutional Neural Network , 2017, 2017 39th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[32]  Shen-Chuan Tai,et al.  An Automatic Mass Detection System in Mammograms Based on Complex Texture Features , 2014, IEEE Journal of Biomedical and Health Informatics.

[33]  Yo-Sung Ho,et al.  Automated Detection of Tumors in Mammograms Using Two Segments for Classification , 2005, PCM.

[34]  Yifan Wu,et al.  Automatic Mass Detection from Mammograms with Region-Based Convolutional Neural Network , 2018, IGTA.

[35]  Chen-Chung Liu,et al.  A breast region extraction scheme for digital mammograms using gradient vector flow Snake , 2010, 4th International Conference on New Trends in Information Science and Service Science.

[36]  Jaewoo Kang,et al.  Detection of masses in mammograms using a one-stage object detector based on a deep convolutional neural network , 2018, PloS one.

[37]  Patrice Y. Simard,et al.  Best practices for convolutional neural networks applied to visual document analysis , 2003, Seventh International Conference on Document Analysis and Recognition, 2003. Proceedings..

[38]  Alexandru Telea,et al.  An Image Inpainting Technique Based on the Fast Marching Method , 2004, J. Graphics, GPU, & Game Tools.

[39]  Leonid Karlinsky,et al.  Deep Learning for Automatic Detection of Abnormal Findings in Breast Mammography , 2017, DLMIA/ML-CDS@MICCAI.

[40]  Anselmo Cardoso de Paiva,et al.  Detection of mass regions in mammograms by bilateral analysis adapted to breast density using similarity indexes and convolutional neural networks , 2018, Comput. Methods Programs Biomed..

[41]  Xavier Lladó,et al.  Automatic mass detection in mammograms using deep convolutional neural networks , 2019, Journal of medical imaging.

[42]  Marios Savvides,et al.  Feature Selective Anchor-Free Module for Single-Shot Object Detection , 2019, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[43]  Anselmo Cardoso de Paiva,et al.  Detection of masses in mammograms with adaption to breast density using genetic algorithm, phylogenetic trees, LBP and SVM , 2015, Expert Syst. Appl..

[44]  Xinbo Gao,et al.  A parasitic metric learning net for breast mass classification based on mammography , 2018, Pattern Recognit..

[45]  I. Jatoi,et al.  The global breast cancer burden. , 2012, Future oncology.