1. Interpretable deep learning regression for breast density estimation on MRI
- Author
-
Max A.A. Ragusi, Claudette E. Loo, Markus H. A. Janse, Bas H. M. van der Velden, and Kenneth G. A. Gilhuijs
- Subjects
FOS: Computer and information sciences ,Computer Vision and Pattern Recognition (cs.CV) ,Computer Science - Computer Vision and Pattern Recognition ,shapley additive explanations ,computer.software_genre ,Convolutional neural network ,03 medical and health sciences ,0302 clinical medicine ,Breast cancer ,Voxel ,FOS: Electrical engineering, electronic engineering, information engineering ,medicine ,Breast MRI ,Segmentation ,interpretable deep learning ,skin and connective tissue diseases ,breast density ,mri ,Mathematics ,Ground truth ,cnn regression ,medicine.diagnostic_test ,business.industry ,Deep learning ,Image and Video Processing (eess.IV) ,Pattern recognition ,Electrical Engineering and Systems Science - Image and Video Processing ,medicine.disease ,Regression ,030220 oncology & carcinogenesis ,Artificial intelligence ,business ,computer ,deep shap ,030217 neurology & neurosurgery - Abstract
Breast density, which is the ratio between fibroglandular tissue (FGT) and total breast volume, can be assessed qualitatively by radiologists and quantitatively by computer algorithms. These algorithms often rely on segmentation of breast and FGT volume. In this study, we propose a method to directly assess breast density on MRI, and provide interpretations of these assessments. We assessed breast density in 506 patients with breast cancer using a regression convolutional neural network (CNN). The input for the CNN were slices of breast MRI of 128 x 128 voxels, and the output was a continuous density value between 0 (fatty breast) and 1 (dense breast). We used 350 patients to train the CNN, 75 for validation, and 81 for independent testing. We investigated why the CNN came to its predicted density using Deep SHapley Additive exPlanations (SHAP). The density predicted by the CNN on the testing set was significantly correlated with the ground truth densities (N = 81 patients, Spearman's rho = 0.86, P < 0.001). When inspecting what the CNN based its predictions on, we found that voxels in FGT commonly had positive SHAP-values, voxels in fatty tissue commonly had negative SHAP-values, and voxels in non-breast tissue commonly had SHAP-values near zero. This means that the prediction of density is based on the structures we expect it to be based on, namely FGT and fatty tissue. To conclude, we presented an interpretable deep learning regression method for breast density estimation on MRI with promising results., This paper has been published as: Van der Velden, B.H.M., Ragusi, M.A.A., Janse, M.H.A., Loo, C.E., Gilhuijs, K.G.A. "Interpretable deep learning regression for breast density estimation on MRI." Medical Imaging 2020: Computer-Aided Diagnosis. Vol. 11314. International Society for Optics and Photonics, 2020
- Published
- 2020