A deep learning approach for the analysis of masses in mammograms with minimal user intervention

被引:231
作者
Dhungel, Neeraj [1 ]
Carneiro, Gustavo [2 ]
Bradley, Andrew P. [3 ]
机构
[1] Univ British Columbia, Elect & Comp Engn, Vancouver, BC V5Z 1M9, Canada
[2] Univ British Columbia, Australian Ctr Visual Technol, Vancouver, BC V5Z 1M9, Canada
[3] Univ Queensland, ITEE, Brisbane, Qld 4072, Australia
基金
澳大利亚研究理事会;
关键词
Mammograms; Masses; Detection; Segmentation; Classification; Deep learning; Bayesian optimisation; Transfer learning; Structured output learning; COMPUTER-AIDED DETECTION; DIAGNOSIS CAD; SEGMENTATION; ALGORITHM; FEATURES; SYSTEM; MODEL;
D O I
10.1016/j.media.2017.01.009
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
We present an integrated methodology for detecting, segmenting and classifying breast masses from mammograms with minimal user intervention. This is a long standing problem due to low signal-tonoise ratio in the visualisation of breast masses, combined with their large variability in terms of shape, size, appearance and location. We break the problem down into three stages: mass detection, mass segmentation, and mass classification. For the detection, we propose a cascade of deep learning methods to select hypotheses that are refined based on Bayesian optimisation. For the segmentation, we propose the use of deep structured output learning that is subsequently refined by a level set method. Finally, for the classification, we propose the use of a deep learning classifier, which is pre-trained with a regression to hand-crafted feature values and fine-tuned based on the annotations of the breast mass classification dataset. We test our proposed system on the publicly available INbreast dataset and compare the results with the current state-of-the-art methodologies. This evaluation shows that our system detects 90% of masses at 1 false positive per image, has a segmentation accuracy of around 0.85 (Dice index) on the correctly detected masses, and overall classifies masses as malignant or benign with sensitivity (Se) of 0.98 and specificity (Sp) of 0.7. (C) 2017 Elsevier B.V. All rights reserved.
引用
收藏
页码:114 / 128
页数:15
相关论文
共 62 条
[1]  
[Anonymous], ARXIV161002251
[2]  
[Anonymous], AUSTR I HLTH WELF CA
[3]   Digital mammographic computer aided diagnosis (CAD) using adaptive level set segmentation [J].
Ball, John E. ;
Bruce, Lori Mann .
2007 ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY, VOLS 1-16, 2007, :4973-4978
[4]   A completely automated CAD system for mass detection in a large mammographic database [J].
Bellotti, R. ;
De Carlo, F. ;
Tangaro, S. ;
Gargano, G. ;
Maggipinto, G. ;
Castellano, M. ;
Massafra, R. ;
Cascio, D. ;
Fauci, F. ;
Magro, R. ;
Raso, G. ;
Lauria, A. ;
Forni, G. ;
Bagnasco, S. ;
Cerello, P. ;
Zanon, E. ;
Cheran, S. C. ;
Torres, E. Lopez ;
Bottigli, U. ;
Masala, G. L. ;
Oliva, P. ;
Retico, A. ;
Fantacci, M. E. ;
Cataldo, R. ;
De Mitri, I. ;
De Nunzio, G. .
MEDICAL PHYSICS, 2006, 33 (08) :3066-3075
[5]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[6]   A novel featureless approach to mass detection in digital mammograms based on support vector machines [J].
Campanini, R ;
Dongiovanni, D ;
Iampieri, E ;
Lanconelli, N ;
Masotti, M ;
Palermo, G ;
Riccardi, A ;
Roffilli, M .
PHYSICS IN MEDICINE AND BIOLOGY, 2004, 49 (06) :961-975
[7]   Closed Shortest Path in the Original Coordinates with an Application to Breast Cancer [J].
Cardoso, Jaime S. ;
Domingues, Ines ;
Oliveira, Helder P. .
INTERNATIONAL JOURNAL OF PATTERN RECOGNITION AND ARTIFICIAL INTELLIGENCE, 2015, 29 (01)
[8]   Unregistered Multiview Mammogram Analysis with Pre-trained Deep Learning Models [J].
Carneiro, Gustavo ;
Nascimento, Jacinto ;
Bradley, Andrew P. .
MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION, PT III, 2015, 9351 :652-660
[9]   Active contours without edges [J].
Chan, TF ;
Vese, LA .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2001, 10 (02) :266-277
[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