Abstract
Breast cancer is the most publicized cancer that hits women around the world. It’s considered as the second cause of death among females. Early detection helps a lot in increasing the survival rate, and the probability of recovery from this disease. The mammogram is the main screening modality that is used regularly for breast cancer diagnosis. The accurate interpretation of the mammogram is very important for mass detection and diagnosis. The rapid evolution of deep learning is contributing to introduce more accurate systems that can act as a second opinion for the radiologists, and accordingly, this can help in providing an accurate diagnosis. In this paper, we propose a model for mass detection and classification based on You Look Only Once (YOLO)v4. We designed the experiment to investigate the performance of different augmentation techniques using YOLOv4 including mosaic that was introduced by YOLOv4. Furthermore, in the preprocessing phase, the images were reconstructed to be in a multichannel format which enhanced the detection accuracy by almost \(\simeq\) 10%. The model was evaluated with the usage of different combinations of augmentation techniques (mosaic, mix-up, and conventional augmentation). The experiments were conducted on the INbreast and MIAS datasets, the results of INbreast showed that mosaic with YOLOv4 achieved the best results with mAP (mean average precision), precession, and recall of almost \(\simeq\) 99.5%, 98%, and 94% respectively for detection. In addition, the proposed model achieved AP of 99.16% and 99.58% for classifying the detected masses as benign and malignant respectively. Furthermore, the best results on MIAS achieved mAP, precession, and recall of 95.28%, 93%, and 90% respectively. Finally, our methodology showed competitive performance compared to other similar studies.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Global Cancer Statistics: GLOBOCAN Estimates of Incidence and Mortality Worldwide for 36 Cancers in 185 Countries. Enhanced Reader (2020). Accessed 28 Feb 2022
Hadjiiski, L., Sahiner, B., Chan, H.P.: Advances in CAD for diagnosis of breast cancer. Curr. Opin. Obstet. Gynecol. 18, 64 (2006). https://doi.org/10.1097/01.GCO.0000192965.29449.DA
Redmon, J., Divvala, S., Girshick, R., Farhadi, A.: You Only Look Once: Unified, Real-Time Object Detection
Redmon, J., Farhadi, A.: YOLO9000: Better, faster, stronger. In: Proceedings of the 30th IEEE Conference on Computer Vision and Pattern Recognition, CVPR 2017, pp. 6517–6525 (2016). https://doi.org/10.1109/CVPR.2017.690
Redmon, J., Farhadi, A.: YOLOv3: An Incremental Improvement (2018)
Bochkovskiy, A., Wang, C.-Y., Liao, H.-Y.M.: YOLOv4: Optimal Speed and Accuracy of Object Detection (2020)
Al-antari, M.A., Al-masni, M.A., Choi, M.T., Han, S.M., Kim, T.S.: A fully integrated computer-aided diagnosis system for digital X-ray mammograms via deep learning detection, segmentation, and classification. Int. J. Med. Inform. 117, 44–54 (2018). https://doi.org/10.1016/J.IJMEDINF.2018.06.003
Al-antari, M.A., Al-masni, M.A., Kim, T.-S.: Deep learning computer-aided diagnosis for breast lesion in digital mammogram. In: Lee, G., Fujita, H. (eds.) Deep Learning in Medical Image Analysis. AEMB, vol. 1213, pp. 59–72. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-33128-3_4
Ribli, D., Horváth, A., Unger, Z., Pollner, P., Csabai, I.: Detecting and classifying lesions in mammograms with deep learning. Sci. Rep. 8(1), 1–7 (2018). https://doi.org/10.1038/s41598-018-22437-z
Agarwal, R., Díaz, O., Yap, M.H., Lladó, X., Martí, R.: Deep learning for mass detection in full field digital mammograms. Comput. Biol. Med. 121, 103774 (2020). https://doi.org/10.1016/J.COMPBIOMED.2020.103774
Peng, J., Bao, C., Hu, C., Wang, X., Jian, W., Liu, W.: Automated mammographic mass detection using deformable convolution and multiscale features. Med. Biol. Eng. Comput. 58, 1405–1417 (2020). https://doi.org/10.1007/S11517-020-02170-4/FIGURES/7
Hassan, N.M., Hamad, S., Mahar, K., Tools, M.: Mammogram breast cancer CAD systems for mass detection and classification: A review. Multimedia Tools Appl. 81, 1–33 (2022). https://doi.org/10.1007/S11042-022-12332-1
Liu, S., Qi, L., Qin, H., Shi, J., Jia, J.: Path aggregation network for instance segmentation. In: Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition, pp. 8759–8768 (2018). https://doi.org/10.1109/CVPR.2018.00913
Woo, S., Park, J., Lee, J.-Y., Kweon, I.S.: CBAM: Convolutional block attention module. In: Ferrari, V., Hebert, M., Sminchisescu, C., Weiss, Y. (eds.) ECCV 2018. LNCS, vol. 11211, pp. 3–19. Springer, Cham (2018). https://doi.org/10.1007/978-3-030-01234-2_1
He, K., Zhang, X., Ren, S., Sun, J.: Spatial pyramid pooling in deep convolutional networks for visual recognition. In: Fleet, D., Pajdla, T., Schiele, B., Tuytelaars, T. (eds.) ECCV 2014. LNCS, vol. 8691, pp. 346–361. Springer, Cham (2014). https://doi.org/10.1007/978-3-319-10578-9_23
Huang, Z., Wang, J.: DC-SPP-YOLO: Dense Connection and Spatial Pyramid Pooling Based YOLO for Object Detection
Zhang, H., Cisse, M., Dauphin, Y.N., Lopez-Paz, D.: Mixup: Beyond Empirical Risk Minimization
Moreira, I.C., Amaral, I., Domingues, I., Cardoso, A., Cardoso, M.J., Cardoso, J.S.: INbreast: Toward a full-field digital mammographic database. Acad. Radiol. 19, 236–248 (2012). https://doi.org/10.1016/J.ACRA.2011.09.014
Mammographic Image Analysis Society (MIAS) database v1.21. https://www.repository.cam.ac.uk/handle/1810/250394. Accessed 26 Mar 2022
Lee, R.S., Gimenez, F., Hoogi, A., Miyake, K.K., Gorovoy, M., Rubin, D.L.: A curated mammography data set for use in computer-aided detection and diagnosis research. Sci. Data 4(1), 1–9 (2017). https://doi.org/10.1038/sdata.2017.177
D’Haeyer, J.P.F.: Gaussian filtering of images: A regularization approach. Signal Process. 18, 169–181 (1989). https://doi.org/10.1016/0165-1684(89)90048-0
Otsu, N.: Threshold selection method from gray-level histograms. IEEE Trans. Syst. Man Cybern. SMC 9, 62–66 (1979). https://doi.org/10.1109/TSMC.1979.4310076
Pizer, S.M., et al.: Adaptive histogram equalization and its variations. Comput. Vis. Graph. Image Process. 39, 355–368 (1987). https://doi.org/10.1016/S0734-189X(87)80186-X
Song, S.Y., Park, B., Hong, S., Kim, M.J., Lee, E.H., Jun, J.K.: Comparison of digital and screen-film mammography for breast-cancer screening: A systematic review and meta analysis. J. Breast Cancer 22, 311–325 (2019). https://doi.org/10.4048/jbc.2019.22.e24
Aly, G.H., Marey, M., El-Sayed, S.A., Tolba, M.F.: YOLO based breast masses detection and classification in full-field digital mammograms. Comput. Methods Programs Biomed. 200, 105823 (2021). https://doi.org/10.1016/J.CMPB.2020.105823
Baccouche, A., Garcia-Zapirain, B., Olea, C.C., Elmaghraby, A.S.: Breast lesions detection and classification via YOLO-based fusion models. Comput. Mater. Continua 69, 1407–1425 (2021). https://doi.org/10.32604/CMC.2021.018461
Platania, R., Shams, S., Yang, S., Zhang, J., Lee, K., Park, S.-J.: Automated breast cancer diagnosis using deep learning and region of interest detection (BC-DROID). In: Proceedings of the 8th ACM International Conference on Bioinformatics, Computational Biology, and Health Informatics (2017). https://doi.org/10.1145/3107411
Al-antari, M.A., Han, S.M., Kim, T.S.: Evaluation of deep learning detection and classification towards computer-aided diagnosis of breast lesions in digital X-ray mammograms. Comput. Methods Programs Biomed. 196, 105584 (2020). https://doi.org/10.1016/J.CMPB.2020.105584
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Hassan, N.M., Hamad, S., Mahar, K. (2022). A Deep Learning Model for Mammography Mass Detection Using Mosaic and Reconstructed Multichannel Images. In: Gervasi, O., Murgante, B., Hendrix, E.M.T., Taniar, D., Apduhan, B.O. (eds) Computational Science and Its Applications – ICCSA 2022. ICCSA 2022. Lecture Notes in Computer Science, vol 13375. Springer, Cham. https://doi.org/10.1007/978-3-031-10522-7_37
Download citation
DOI: https://doi.org/10.1007/978-3-031-10522-7_37
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-10521-0
Online ISBN: 978-3-031-10522-7
eBook Packages: Computer ScienceComputer Science (R0)