Multi-task learning to incorporate clinical knowledge into deep learning for breast cancer diagnosis

Giacomo Nebbia; Dooman Arefan; Margarita Zuley; Jules Sumkin; Shandong Wu

doi:10.1117/12.2582285

15 February 2021 Multi-task learning to incorporate clinical knowledge into deep learning for breast cancer diagnosis

Giacomo Nebbia, Dooman Arefan, Margarita Zuley, Jules Sumkin, Shandong Wu

Author Affiliations +

Proceedings Volume 11597, Medical Imaging 2021: Computer-Aided Diagnosis; 1159710 (2021) https://doi.org/10.1117/12.2582285
Event: SPIE Medical Imaging, 2021, Online Only

Abstract

Deep learning models are traditionally trained purely in a data-driven approach; the information for the model training usually only comes from a single source of the training data. In this work, we investigate how to supply additional clinical knowledge that is associated with the training data. Our goal is to train deep learning models for breast cancer diagnosis using mammogram images. Along with the main classification task between clinically proven cancer vs negative/benign cases, we design two auxiliary tasks each capturing a form of additional knowledge to facilitate the main task. Specifically, one auxiliary task is to classify images according to the radiologist-made BI-RADS diagnosis scores and the other auxiliary task is to classify images in terms of the BI-RADS breast density categories. We customize a Multi-Task Learning model to jointly perform the three tasks (main task and two auxiliary tasks). We test four deep learning architectures: CBR–Tiny, ResNet18, GoogleNet, and DenseNet and we investigate the benefit of incorporating such knowledge over ImageNet pre-trained models and in the case of randomly initialized models. We run experiments on an internal dataset consisting of screening full field digital mammography images for a total of 1,380 images (341 cancer and 1,039 negative or benign). Our results show that, by adding clinical knowledge conveyed through the two auxiliary tasks to the training process, we can improve the performance of the target task of breast cancer diagnosis, thus highlighting the benefit of incorporating clinical knowledge into data-driven learning to enhance deep learning model training.

Conference Presentation

Citation Download Citation

Giacomo Nebbia, Dooman Arefan, Margarita Zuley, Jules Sumkin, and Shandong Wu "Multi-task learning to incorporate clinical knowledge into deep learning for breast cancer diagnosis", Proc. SPIE 11597, Medical Imaging 2021: Computer-Aided Diagnosis, 1159710 (15 February 2021); https://doi.org/10.1117/12.2582285

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