Deep Learning Based Mass Detection in Mammograms

Mammogram is the primary imaging technique for breast cancer screening, the leading type of cancer in women worldwide. While the clinical effectiveness of mammogram has been well demonstrated, the mammographic characteristics of breast masses are quite complex. As a result, radiologists certified for reading mammography are lacking, which limits the accessibility of mammography for more population. In this paper, we propose a Computer Aided Detection (CADe) method to automatically detect masses in mammography. Our method combines Faster R-CNN with Feature Pyramid Network, Focal Loss, NonLocal Neural Network to achieve the optimal mass detection performance. We comprehensively evaluated our method and baseline methods on three public datasets combined, namely the Digital Database for Screening Mammography (DDSM), INbreast, and Breast Cancer Digital repository (BCD). Our results demonstrate that our method outperforms the baselines by a large margin, reporting an Average Precision of 0.805, and a recall of 0.977, respectively.

[1]  Kaiming He,et al.  Feature Pyramid Networks for Object Detection , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[2]  Malcolm C Pike,et al.  Does mammographic density reflect ethnic differences in breast cancer incidence rates? , 2004, American journal of epidemiology.

[3]  Abhinav Gupta,et al.  Non-local Neural Networks , 2017, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[4]  Fei-Fei Li,et al.  ImageNet: A large-scale hierarchical image database , 2009, 2009 IEEE Conference on Computer Vision and Pattern Recognition.

[5]  Kaiming He,et al.  Focal Loss for Dense Object Detection , 2017, 2017 IEEE International Conference on Computer Vision (ICCV).

[6]  Kaiming He,et al.  Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[7]  Jacques Ferlay,et al.  Global Burden of Breast Cancer , 2010 .

[8]  Z. Jane Wang,et al.  A CNN Regression Approach for Real-Time 2D/3D Registration , 2016, IEEE Transactions on Medical Imaging.

[9]  Wei Wei,et al.  Thoracic Disease Identification and Localization with Limited Supervision , 2017, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[10]  Daniel L Rubin,et al.  A curated mammography data set for use in computer-aided detection and diagnosis research , 2017, Scientific Data.

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

[12]  Trevor Darrell,et al.  Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation , 2013, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[13]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[14]  Wei Yan,et al.  MobiCardio: A Clinical-grade Mobile Health System for Cardiovascular Disease Management , 2019, 2019 IEEE International Conference on Healthcare Informatics (ICHI).

[15]  Sang Won Yoon,et al.  Breast cancer diagnosis based on feature extraction using a hybrid of K-means and support vector machine algorithms , 2014, Expert Syst. Appl..

[16]  Jean-Michel Morel,et al.  A non-local algorithm for image denoising , 2005, 2005 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR'05).

[17]  Daniel C. Moura,et al.  BCDR : A BREAST CANCER DIGITAL REPOSITORY , 2012 .

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

[19]  Bram van Ginneken,et al.  A survey on deep learning in medical image analysis , 2017, Medical Image Anal..

[20]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.