Unregistered Multiview Mammogram Analysis with Pre-trained Deep Learning Models

被引:171
作者
Carneiro, Gustavo [1 ]
Nascimento, Jacinto [2 ]
Bradley, Andrew P. [3 ]
机构
[1] Univ Adelaide, ACVT, Adelaide, SA 5005, Australia
[2] Inst Super Tecn, ISR, Lisbon, Portugal
[3] Univ Queensland, Brisbane, Qld 4072, Australia
来源
MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION, PT III | 2015年 / 9351卷
关键词
Deep learning; Mammogram; Multiview classification; CLASSIFICATION; ARCHITECTURES; DIAGNOSIS;
D O I
10.1007/978-3-319-24574-4_78
中图分类号
TP18 [人工智能理论];
学科分类号
140502 [人工智能];
摘要
We show two important findings on the use of deep convolutional neural networks (CNN) in medical image analysis. First, we show that CNN models that are pre-trained using computer vision databases (e.g., Imagenet) are useful in medical image applications, despite the significant differences in image appearance. Second, we show that multiview classification is possible without the pre-registration of the input images. Rather, we use the high-level features produced by the CNNs trained in each view separately. Focusing on the classification of mammograms using craniocaudal (CC) and mediolateral oblique (MLO) views and their respective mass and micro-calcification segmentations of the same breast, we initially train a separate CNN model for each view and each segmentation map using an Imagenet pre-trained model. Then, using the features learned from each segmentation map and unregistered views, we train a final CNN classifier that estimates the patient's risk of developing breast cancer using the Breast Imaging-Reporting and Data System (BI-RADS) score. We test our methodology in two publicly available datasets (InBreast and DDSM), containing hundreds of cases, and show that it produces a volume under ROC surface of over 0.9 and an area under ROC curve (for a 2-class problem - benign and malignant) of over 0.9. In general, our approach shows state-of-the-art classification results and demonstrates a new comprehensive way of addressing this challenging classification problem.
引用
收藏
页码:652 / 660
页数:9
相关论文
共 22 条
[1]
Cruz-Roa AA, 2013, LECT NOTES COMPUT SC, V8150, P403, DOI 10.1007/978-3-642-40763-5_50
[2]
[Anonymous], 2012, NIPS
[3]
[Anonymous], 2014, CORR
[4]
[Anonymous], 2000, P 5 INT WORKSH DIG M
[5]
Learning Deep Architectures for AI [J].
Bengio, Yoshua .
FOUNDATIONS AND TRENDS IN MACHINE LEARNING, 2009, 2 (01) :1-127
[6]
Brosch T, 2014, LECT NOTES COMPUT SC, V8674, P462, DOI 10.1007/978-3-319-10470-6_58
[7]
Combining Multiple Dynamic Models and Deep Learning Architectures for Tracking the Left Ventricle Endocardium in Ultrasound Data [J].
Carneiro, Gustavo ;
Nascimento, Jacinto C. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2013, 35 (11) :2592-2607
[8]
Chatfield K., 2014, ABS14053531 CORR
[9]
Approaches for automated detection and classification of masses in mammograms [J].
Cheng, HD ;
Shi, XJ ;
Min, R ;
Hu, LM ;
Cai, XR ;
Du, HN .
PATTERN RECOGNITION, 2006, 39 (04) :646-668
[10]
Mitosis Detection in Breast Cancer Histology Images with Deep Neural Networks [J].
Ciresan, Dan C. ;
Giusti, Alessandro ;
Gambardella, Luca M. ;
Schmidhuber, Juergen .
MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION - MICCAI 2013, PT II, 2013, 8150 :411-418