Multi-View Attention-based Late Fusion (MVALF) CADx system for breast cancer using deep learning

Main Article Content

Iftikhar Hina
Shahid Ahmad Raza
Raza Basit
Khan Hasan

Keywords : breast cancer, mammogram, four-view mammogram, information fusion, late fusion, transfer learning

Breast cancer is a leading cause of death among women. Early detection can significantly reduce the mortality rate among women and improve their prognosis. Mammography is the first line procedure for early diagnosis. In the early era, conventional Computer-Aided Diagnosis (CADx) systems for breast lesion diagnosis were based on just single view information. The last decade evidence the use of two views mammogram: Medio-Lateral Oblique (MLO) and Cranio-Caudal (CC) view for the CADx systems. Most recent studies show the effectiveness of four views of mammogram to train CADx system with feature fusion strategy for classification task. In this paper, we proposed an end-to-end Multi-View Attention-based Late Fusion (MVALF) CADx system that fused the obtained predictions of four view models, which is trained for each view separately. These separate models have different predictive ability for each class. The appropriate fusion of multi-view models can achieve better diagnosis performance. So, it is necessary to assign the proper weights to the multi-view classification models. To resolve this issue, attention-based weighting mechanism is adopted to assign the proper weights to trained models for fusion strategy. The proposed methodology is used for the classification of mammogram into normal, mass, calcification, malignant masses and benign masses. The publicly available datasets CBIS-DDSM and mini-MIAS are used for the experimentation. The results show that our proposed system achieved 0.996 AUC for normal vs. abnormal, 0.922 for mass vs. calcification and 0.896 for malignant vs. benign masses. Superior results are seen for the classification of malignant vs benign masses with our proposed approach, which is higher than the results using single view, two views and four views early fusion-based systems. The overall results of each level show the potential of multi-view late fusion with transfer learning in the diagnosis of breast cancer.

Article Details

How to Cite
Hina, I. ., Raza, S. A. ., Basit , R., & Hasan, K. (2020). Multi-View Attention-based Late Fusion (MVALF) CADx system for breast cancer using deep learning. Machine Graphics and Vision, 29(1/4), 55–78.
Author Biography

Shahid Ahmad Raza, 1) Medical Imaging and Diagnostics Laboratory (MID), National Centre of Artificial Intelligence (NCAI), Islamabad, Pakistan; 2) Department of Computer Science, COMSATS University Islamabad (CUI), Pakistan


Large Scale Visual Recognition Challenge 2015 (ILSVRC2015), 2015. [Aceessed Jun 2020].

R. Agarwal, O. Diaz, X. Lladó, et al. Automatic mass detection in mammograms using deep convolutional neural networks. Journal of Medical Imaging, 6(3):31409, 2019. (Crossref)

A. Akselrod-Ballin, L. Karlinsky, A. Hazan, et al. Deep learning for automatic detection of ab-normal findings in breast mammography. In M. J. Cardoso, T. Arbel, G. Carneiro, et al., editors, Proc. Int. Workshops on Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support DLMIA, ML-CDS, in conjunction with MICCAI 2017, volume 10553 of Lecture Notes in Computer Science, pages 321–329. Springer, Québec City, QC, Canada, 14 Sep 2017. (Crossref)

M. A. Al-masni, M. A. Al-antari, J.-M. Park, et al. Simultaneous detection and classification of breast masses in digital mammograms via a deep learning YOLO-based CAD system. Computer methods programs in biomedicine, 157:85–94, 2018. (Crossref)

J. Arevalo, F. A. González, Ramos-Pollán, et al. Representation learning for mammography mass lesion classification with convolutional neural networks. Computer methods programs in biomedicine,127:248–257, 2016. (Crossref)

L. W. Bassett, I. A. Hirbawi, N. DeBruhl, and M. K. Hayes. Mammographic positioning: evaluation from the view box. Radiology, 188(3):803–806, 1993. (Crossref)

M. Bator and M. Nieniewski. Detection of cancerous masses in mammograms by template matching: Optimization of template brightness distribution by means of evolutionary algorithm. Journal of Digital Imaging, 25(1):162–172, 2012. (Crossref)

G. Carneiro, J. Nascimento, and A. P. Bradley. Unregistered multiview mammogram analysis with pre-trained deep learning models. In N. Navab, J. Hornegger, W. M. Wells, and A. F. Frangi, editors, Proc. Int. Conf. Medical Image Computing and Computer-Assisted Intervention MICCAI 2015, volume 9351 of Lecture Notes in Computer Science, pages 652–660, Munich, Germany, 5-9 Oct 2015. Springer. (Crossref)

G. Carneiro, J. Nascimento, and A. P. Bradley. Automated analysis of unregistered multi-view mammograms with deep learning. IEEE Transactions on Medical Imaging, 36(11):2355–2365, 2017. (Crossref)

J. Chakraborty, A. Midya, and R. Rabidas. Computer-aided detection and diagnosis of mammographic masses using multi-resolution analysis of oriented tissue patterns. Expert Systems with Applications, 99:168–179, 2018. (Crossref)

H. Chen, D. Ni, J. Qin, et al. Standard plane localization in fetal ultrasound via domain transferred deep neural networks. IEEE Journal of Biomedical Health Informatics, 19(5):1627–1636, 2015. (Crossref)

H. Chougrad, H. Zouaki, and O. Alheyane. Deep convolutional neural networks for breast cancer screening. Computer Methods Programs in Biomedicine, 157:19–30, 2018. (Crossref)

F. Ciompi, B. de Hoop, S. J. van Riel, et al. Automatic classification of pulmonary peri-fissural nodules in computed tomography using an ensemble of 2D views and a convolutional neural network out-of-the-box. Medical Image Analysis, 26(1):195–202, 2015. (Crossref)

A. F. Clark. The mini-MIAS database of mammograms, 2012. [Accessed Jun 2020].

K. Clark, B. Vendt, K. Smith, et al. The cancer imaging archive (TCIA): Maintaining and operating a public information repository. Journal of Digital Imaging, 26(6):1045–1057, 2013. (Crossref)

J. Deng, W. Dong, R. Socher, et al. ImageNet: A large-scale hierarchical image database. In Proc. IEEE Conf. Computer Vision and Pattern Recognition CVPR 2009, pages 248–255, Miami, FL,USA, 20-25 Jun 2009. IEEE. (Crossref)

S. Dhahbi, W. Barhoumi, and E. Zagrouba. Multi-view score fusion for content-based mammogram retrieval. In A. Verikas, P. Radeva, and D. Nikolaev, editors, Proc. 8th Int. Conf. Machine Vision ICMV 2015, volume 9875 of Proc. SPIE, page 987515, Barcelona, Spain, 8 Dec 2015. (Crossref)

N. Dhungel, G. Carneiro, and A. P. Bradley. A deep learning approach for the analysis of masses in mammograms with minimal user intervention. Medical Image Analysis, 37:114–128, 2017. (Crossref)

G. W. Eklund. The art of mammographic positioning. In M. Friedrich and E. A. Sickles, editors, Radiological Diagnosis of Breast Diseases, pages 75–88. Springer, 2000. (Crossref)

M. Heath, D. Bowyer, R. Kopans, et al. The digital data base for screening Mammography. In M. J. Yaffe, editor, Proc. 5th Int. Workshop on Digital Mammography, pages 212–218, Toronto, Canada, 11-14 Jun 2000. Medical Physics Publishing, Madison, WI, USA.

M. Heath, K. Bowyer, D. Kopans, et al. Current status of the digital database for screening mammography. In N. Karssemeijer, M. Thijssen, J. Hendriks, and L. van Erning, editors, Digital Mammography, pages 457–460. Springer Netherlands, Dordrecht, 1998. (Crossref)

B. Q. Huynh, H. Li, and M. L. Giger. Digital mammographic tumor classification using transfer learning from deep convolutional neural networks. Journal of Medical Imaging, 3(3):34501, 2016. (Crossref)

Z. Jiao, X. Gao, Y. Wang, and J. Li. A deep feature based framework for breast masses classification. Neurocomputing, 197:221–231, 2016. (Crossref)

A. Jouirou, A. Baâzaoui, and W. Barhoumi. Multi-view information fusion in mammograms: A comprehensive overview. Information Fusion, 52:308–321, 2019. (Crossref)

A. Khan, A. Sohail, U. Zahoora, and A. S. Qureshi. A survey of the recent architectures of deep convolutional neural networks. Artificial Intelligence Review, 53(8):5455–5516, 2020. (Crossref)

H. N. Khan, A. R. Shahid, B. Raza, A. H. Dar, and H. Alquhayz. Multi-view feature fusion based four views model for mammogram classification using convolutional neural network. IEEE Access, 7:165724–165733, 2019. (Crossref)

E. Kozegar, M. Soryani, B. Minaei, and I. Domingues. Assessment of a novel mass detection algorithm in mammograms. Journal of Cancer Research and Therapeutics, 9(4):592, 2013. (Crossref)

R. S. Lee, F. Gimenez, A. Hoogi, et al. A curated mammography data set for use in computer-aided detection and diagnosis research. Scientific Data, 4:170177, 2017. (Crossref)

H. Li, S. Zhuang, D.-A. Li, J. Zhao, and Y. Ma. Benign and malignant classification of mammogram images based on deep learning. Biomedical Signal Processing Control, 51:347–354, 2019. (Crossref)

Y. Li, H. Chen, Y. Yang, et al. A bilateral analysis scheme for false positive reduction in mammogram mass detection. Computers in Biology and Medicine, 57:84–95, 2015. (Crossref)

X. Liu, T. Zhu, L. Zhai, and J. Liu. Improvement of mass detection in mammogram using multi-view information. In C. M. Falco and X. Jiang, editors, Proc. 8th Int. Conf. Digital Image Processing ICDIP 2016, volume 10033 of Proc. SPIE, page 100334M, Chengdu, China, 29 Aug 2016. (Crossref)

W. Lotter, G. Sorensen, and D. Cox. A multi-scale CNN and curriculum learning strategy for mammogram classification. In M. J. Cardoso, T. Arbel, G. Carneiro, et al., editors, Proc. Int. Workshops on Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support DLMIA, ML-CDS, in conjunction with MICCAI 2017, volume 10553 of Lecture Notes in Computer Science, pages 169–177. Springer, Québec City, QC, Canada, 14 Sep 2017. (Crossref)

W. Peng, R. V. Mayorga, and E. M. A. Hussein. An automated confirmatory system for analysis of mammograms. Computer Methods Programs in Biomedicine, 125:134–144, 2016. (Crossref)

D. Ribli, A. Horváth, Z. Unger, et al. Detecting and classifying lesions in mammograms with deep learning. Scientific Reports, 8(1):4165, 2018. (Crossref)

O. Russakovsky, J. Deng, H. Su, et al. ImageNet large scale visual recognition challenge. International Journal of Computer Vision, 115(3):211–252, 2015. (Crossref)

H.-C. Shin, H. R. Roth, M. Gao, et al. Deep convolutional neural networks for computer-aided detection: CNN architectures, dataset characteristics and transfer learning. IEEE Transactions on Medical Imaging, 35(5):1285–1298, 2016. (Crossref)

R. L. Siegel, K. D. Miller, and A. Jemal. Cancer statistics, 2019.CA: A Cancer Journal for Clinicians, 69(1):7–34, 2019. (Crossref)

K. Simonyan and A Zisserman. Very deep convolutional networks for large-scale image recognition. arXiv preprint, 2014. arXiv:1409.1556v6.

K. Smith, J. Kirby, D. Runbin, et al. CBIS-DDSM – Curated Breast Imaging Subset of DDSM. In TCIA Team [44]. [Accessed Jun 2020].

J. Suckling, J. Parker, D. Dance, et al. The Mammographic Images Analysis Society digital mammogram database. In A. G. Gale, S. M. Astley, D. R. Dance, and A. Y. Cairns, editors, Digital Mammography, volume 1069 of Exerpta Medica International Congress Series, pages 375–378. Elsevier, 1994. [Inoperative].

L. Sun, L. Li, W. Xu, et al. A novel classification scheme for breast masses based on multi-view information fusion. In Proc. 4th Int. Conf. Bioinformatics and Biomedical Engineering iCBBE 2010, pages 1–4, Chengdu, China, 18-20 Jun 2010. IEEE. (Crossref)

C. Szegedy, W. Liu, Y. Jia, et al. Going deeper with convolutions. In Proc. IEEE Conf. Computer Vision and Pattern Recognition CVPR 2015, pages 1–9, Boston, MA, USA, 7-12 Jun 2015. (Crossref)

M. Tan, J. Pu, S. Cheng, et al. Assessment of a four-view mammographic image feature based fusion model to predict near-term breast cancer risk. Annals of Biomedical Engineering, 43(10):2416–2428, 2015. (Crossref)

TCIA Team, editors. The Cancer Imaging Archive, 2021. [Accessed Jun 2020].

P. D. Trieu, P. C. Brennan, W. Lee, E. Ryan, et al. The value of the craniocaudal mammographic view in breast cancer detection: a preliminary study. In C. K. Abbey and C. R. Mello-Thoms, editors, Proc. Conf. SPIE Medical Imaging 2013: Image Perception, Observer Performance, and Technology Assessment, volume 8673 of Proc. SPIE, page 86731J, Lake Buena Vista, FL, United States, 28 Mar 2013. (Crossref)

C. J. Vyborny and R. A. Schmidt. Mammography as a radiographic examination: an overview. Radio Graphics, 9(4):723–764, 1989. (Crossref)

WebMD. Breast Calcifications, 2020. [Accessed Jun 2020].

J. Wei, H.-P. Chan, C. Zhou, et al. Computer-aided detection of breast masses: Four-view strategy for screening mammography. Medical Physics, 38(4):1867–1876, 2011. (Crossref)

World Health Organization. Cancer. Facts sheet, 2018. [Aceessed Jun 2020].

N. I. R. Yassin, S. Omran, E. M. F. El Houby, and H. Allam. Machine learning techniques for breast cancer computer aided diagnosis using different image modalities: A systematic review. Computer Methods and Programs in Biomedicine, 156:25–45, 2018. (Crossref)

H. Zhao, W. Xu, L. Li, and J. Zhang. Classification of breast masses based on multi-view information fusion using multi-agent method. In Proc. 5th Int. Conf. Bioinformatics and Biomedical Engineering iCBBE 2011, pages 1–4, Wuhan, China, 10-12 May 2011. IEEE. (Crossref)

Recommend Articles