Improving the Ability of Deep Neural Networks to Use Information from Multiple Views in Breast Cancer Screening

In breast cancer screening, radiologists make the diagnosis based on images that are taken from two angles. Inspired by this, we seek to improve the performance of deep neural networks applied to this task by encouraging the model to use information from both views of the breast. First, we took a closer look at the training process and observed an imbalance between learning from the two views. In particular, we observed that layers processing one of the views have parameters with larger gradients in magnitude, and contribute more to the overall loss reduction. Next, we tested several methods targeted at utilizing both views more equally in training. We found that using the same weights to process both views, or using modality dropout, leads to a boost in performance. Looking forward, our results indicate improving learning dynamics as a promising avenue for improving utilization of multiple views in deep neural networks for medical diagnosis.

[1]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[2]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[3]  Marie Synnestvedt,et al.  Breast Cancer Conspicuity on Simultaneously Acquired Digital Mammographic Images versus Digital Breast Tomosynthesis Images. , 2019, Radiology.

[4]  Nan Wu,et al.  Deep Neural Networks Improve Radiologists’ Performance in Breast Cancer Screening , 2019, IEEE Transactions on Medical Imaging.

[5]  Jeff A. Bilmes,et al.  On Deep Multi-View Representation Learning , 2015, ICML.

[6]  Gustavo Carneiro,et al.  Unregistered Multiview Mammogram Analysis with Pre-trained Deep Learning Models , 2015, MICCAI.

[7]  Andrew P. Bradley,et al.  Automated Analysis of Unregistered Multi-View Mammograms With Deep Learning , 2017, IEEE Transactions on Medical Imaging.

[8]  Kyunghyun Cho,et al.  High-Resolution Breast Cancer Screening with Multi-View Deep Convolutional Neural Networks , 2017, ArXiv.

[9]  Du Tran,et al.  What Makes Training Multi-Modal Networks Hard? , 2019, ArXiv.

[10]  Krzysztof J. Geras,et al.  Artificial Intelligence for Mammography and Digital Breast Tomosynthesis: Current Concepts and Future Perspectives. , 2019, Radiology.

[11]  Yoshua Bengio,et al.  On the Relation Between the Sharpest Directions of DNN Loss and the SGD Step Length , 2018, ICLR.

[12]  Zhao Chen,et al.  GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks , 2017, ICML.

[13]  Juhan Nam,et al.  Multimodal Deep Learning , 2011, ICML.

[14]  Hayit Greenspan,et al.  A Mixture of Views Network With Applications to the Classification of Breast Microcalcifications , 2019, 2019 IEEE 16th International Symposium on Biomedical Imaging (ISBI 2019).

[15]  Subhransu Maji,et al.  Multi-view Convolutional Neural Networks for 3D Shape Recognition , 2015, 2015 IEEE International Conference on Computer Vision (ICCV).

[16]  Louis-Philippe Morency,et al.  Multimodal Machine Learning: A Survey and Taxonomy , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[17]  Nan Wu,et al.  Breast Density Classification with Deep Convolutional Neural Networks , 2017, 2018 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[18]  Mingkui Tan,et al.  Deep Multi-View Learning Using Neuron-Wise Correlation-Maximizing Regularizers , 2019, IEEE Transactions on Image Processing.

[19]  Nan Wu,et al.  An interpretable classifier for high-resolution breast cancer screening images utilizing weakly supervised localization , 2020, Medical Image Anal..

[20]  David Gur,et al.  Digital breast tomosynthesis: observer performance study. , 2009, AJR. American journal of roentgenology.

[21]  David S. Melnick,et al.  International evaluation of an AI system for breast cancer screening , 2020, Nature.

[22]  M Smans,et al.  Mammography screening and breast cancer mortality in Sweden. , 2012, Journal of the National Cancer Institute.

[23]  Jason Yosinski,et al.  LCA: Loss Change Allocation for Neural Network Training , 2019, NeurIPS.

[24]  Christian Wolf,et al.  ModDrop: Adaptive Multi-Modal Gesture Recognition , 2014, IEEE Trans. Pattern Anal. Mach. Intell..

[25]  Mihaela van der Schaar,et al.  Multi-view Multi-task Learning for Improving Autonomous Mammogram Diagnosis , 2019, MLHC.

[26]  Berkman Sahiner,et al.  Evaluation of Combined Artificial Intelligence and Radiologist Assessment to Interpret Screening Mammograms , 2020, JAMA network open.

[27]  Joseph Paul Cohen,et al.  Do Lateral Views Help Automated Chest X-ray Predictions? , 2019, ArXiv.