Breast mass detection in mammography and tomosynthesis via fully convolutional network-based heatmap regression

Jun Zhang; Elizabeth Hope Cain; Ashirbani Saha; Zhe Zhu; Maciej A. Mazurowski

doi:10.1117/12.2295443

27 February 2018 Breast mass detection in mammography and tomosynthesis via fully convolutional network-based heatmap regression

Jun Zhang, Elizabeth Hope Cain, Ashirbani Saha, Zhe Zhu, Maciej A. Mazurowski

Author Affiliations +

Proceedings Volume 10575, Medical Imaging 2018: Computer-Aided Diagnosis; 1057525 (2018) https://doi.org/10.1117/12.2295443
Event: SPIE Medical Imaging, 2018, Houston, Texas, United States

Abstract

Breast mass detection in mammography and digital breast tomosynthesis (DBT) is an essential step in computerized breast cancer analysis. Deep learning-based methods incorporate feature extraction and model learning into a unified framework and have achieved impressive performance in various medical applications (e.g., disease diagnosis, tumor detection, and landmark detection). However, these methods require large-scale accurately annotated data. Unfortunately, it is challenging to get precise annotations of breast masses. To address this issue, we propose a fully convolutional network (FCN) based heatmap regression method for breast mass detection, using only weakly annotated mass regions in mammography images. Specifically, we first generate heat maps of masses based on human-annotated rough regions for breast masses. We then develop an FCN model for end-to-end heatmap regression with an F-score loss function, where the mammography images are regarded as the input and heatmaps for breast masses are used as the output. Finally, the probability map of mass locations can be estimated with the trained model. Experimental results on a mammography dataset with 439 subjects demonstrate the effectiveness of our method. Furthermore, we evaluate whether we can use mammography data to improve detection models for DBT, since mammography shares similar structure with tomosynthesis. We propose a transfer learning strategy by fine-tuning the learned FCN model from mammography images. We test this approach on a small tomosynthesis dataset with only 40 subjects, and we show an improvement in the detection performance as compared to training the model from scratch.

Conference Presentation

Citation Download Citation

Jun Zhang, Elizabeth Hope Cain, Ashirbani Saha, Zhe Zhu, and Maciej A. Mazurowski "Breast mass detection in mammography and tomosynthesis via fully convolutional network-based heatmap regression", Proc. SPIE 10575, Medical Imaging 2018: Computer-Aided Diagnosis, 1057525 (27 February 2018); https://doi.org/10.1117/12.2295443

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