Deep learning for identifying breast cancer malignancy and false recalls: a robustness study on training strategy

Kadie Clancy; Lei Zhang; Aly Mohamed; Sarah Aboutalib; Wendie Berg; Shandong Wu

doi:10.1117/12.2512942

13 March 2019 Deep learning for identifying breast cancer malignancy and false recalls: a robustness study on training strategy

Kadie Clancy, Lei Zhang, Aly Mohamed, Sarah Aboutalib, Wendie Berg, Shandong Wu

Author Affiliations +

Proceedings Volume 10950, Medical Imaging 2019: Computer-Aided Diagnosis; 1095005 (2019) https://doi.org/10.1117/12.2512942
Event: SPIE Medical Imaging, 2019, San Diego, California, United States

Abstract

Identification of malignancy and false recalls (women who are recalled in screening for additional workup, but later proven benign) in screening mammography has significant clinical value for accurate diagnosis of breast cancer. Deep learning methods have recently shown success in the area of medical imaging classification. However, there are a multitude of different training strategies that can significantly impact the overall model performance for a specific classification task. In this study, we aimed to investigate the impact of training strategy on classification of digital mammograms by performing a robustness analysis of deep learning models to distinguish malignancy and false-recall from normal (benign) findings. Specifically, we employed several pre-training strategies including transfer learning with medical and non-medical datasets, layer freezing, and varied network structure on both binary and three-class classification tasks of digital mammography images. We found that, overall, deep learning models appear to be robust to some modifications of network structure and pre-training strategy that we tested for mammogram-specific classification tasks. However, for specific classification tasks, some training strategies offer performance gains. The most notable performance gains in our experiments involved residual network models.

Citation Download Citation

Kadie Clancy, Lei Zhang, Aly Mohamed, Sarah Aboutalib, Wendie Berg, and Shandong Wu "Deep learning for identifying breast cancer malignancy and false recalls: a robustness study on training strategy", Proc. SPIE 10950, Medical Imaging 2019: Computer-Aided Diagnosis, 1095005 (13 March 2019); https://doi.org/10.1117/12.2512942

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available