Recurrent attention U-Net for segmentation and quantification of breast arterial calcifications on synthesized 2D mammograms

被引:0
作者
AlJabri M. [1 ,2 ]
Alghamdi M. [1 ]
Collado-Mesa F. [3 ]
Abdel-Mottaleb M. [4 ]
机构
[1] Department of Computer Science and Artificial Intelligence, Umm Al-Qura University, Makkah, Makkah
[2] King Abdul Aziz University, Jeddah, Makkah
[3] Department of Radiology, Miller School of Medicine, University of Miami, Miami, Florida
[4] Department of Electrical and Computer Engineering, University of Miami, Miami, Florida
关键词
Cardiovascular; Deep-learning; Mammogram; Quantification; Segmentation; U-Net;
D O I
10.7717/PEERJ-CS.2076
中图分类号
学科分类号
摘要
Breast arterial calcifications (BAC) are a type of calcification commonly observed on mammograms and are generally considered benign and not associated with breast cancer. However, there is accumulating observational evidence of an association between BAC and cardiovascular disease, the leading cause of death in women. We present a deep learning method that could assist radiologists in detecting and quantifying BAC in synthesized 2D mammograms. We present a recurrent attention U-Net model consisting of encoder and decoder modules that include multiple blocks that each use a recurrent mechanism, a recurrent mechanism, and an attention module between them. The model also includes a skip connection between the encoder and the decoder, similar to a U-shaped network. The attention module was used to enhance the capture of long-range dependencies and enable the network to effectively classify BAC from the background, whereas the recurrent blocks ensured better feature representation. The model was evaluated using a dataset containing 2,000 synthesized 2D mammogram images. We obtained 99.8861% overall accuracy, 69.6107% sensitivity, 66.5758% F-1 score, and 59.5498% Jaccard coefficient, respectively. The presented model achieved promising performance compared with related models. © Copyright 2024 AlJabri et al.
引用
收藏
页码:1 / 26
页数:25
相关论文
共 42 条
[31]  
Newallo D, Meinel FG, Schoepf UJ, Baumann S, De Cecco CN, Leddy RJ, Vliegenthart R, Mollmann H, Hamm CW, Morris PB, Renker M., Mammographic detection of breast arterial calcification as an independent predictor of coronary atherosclerotic disease in a single ethnic cohort of African American women, Atherosclerosis, 242, 1, pp. 218-221, (2015)
[32]  
Oktay O, Schlemper J, Folgoc LL, Lee M, Heinrich M, Misawa K, Mori K, McDonagh S, Hammerla NY, Kainz B, Glocker B, Rueckert D., Attention U-Net: learning where to look for the pancreas, (2018)
[33]  
Ouyang X, Huo J, Xia L, Shan F, Liu J, Mo Z, Yan F, Ding Z, Yang Q, Song B, Shi F, Yuan H, Wei Y, Cao X, Gao Y, Wu D, Wang Q, Shen D., Dual-sampling attention network for diagnosis of COVID-19 from community acquired pneumonia, IEEE Transactions on Medical Imaging, 39, 8, pp. 2595-2605, (2020)
[34]  
Ronneberger O, Fischer P, Brox T., U-Net: convolutional networks for biomedical image segmentation, International Conference on Medical Image Computing and Computer-Assisted Intervention, pp. 234-241, (2015)
[35]  
Samala RK, Chan H-P, Hadjiiski L, Helvie MA, Wei J, Cha K., Mass detection in digital breast tomosynthesis: deep convolutional neural network with transfer learning from mammography, Medical Physics, 43, 12, pp. 6654-6666, (2016)
[36]  
Skaane P, Bandos AI, Gullien R, Eben EB, Ekseth U, Haakenaasen U, Izadi M, Jebsen IN, Jahr G, Krager M, Niklason LT, Hofvind S, Gur D., Comparison of digital mammography alone and digital mammography plus tomosynthesis in a population-based screening program, Radiology, 267, 1, pp. 47-56, (2013)
[37]  
Wang J, Ding H, Bidgoli FA, Zhou B, Iribarren C, Molloi S, Baldi P., Detecting cardiovascular disease from mammograms with deep learning, IEEE Transactions on Medical Imaging, 36, 5, pp. 1172-1181, (2017)
[38]  
Wang K, Khan N, Highnam R., Automated segmentation of breast arterial calcifications from digital mammography, 2019 International Conference on Image and Vision Computing New Zealand (IVCNZ), pp. 1-6, (2019)
[39]  
Xiong G, Wu Z, Yi J, Fu L, Yang Z, Hsieh C, Yin M, Zeng X, Wu C, Lu A, Chen X, Hou T, Cao D., ADMETLab 2.0: an integrated online platform for accurate and comprehensive predictions of ADMET properties, Nucleic Acids Research, 49, W1, pp. W5-W14, (2021)
[40]  
Yan Z, Yang X, Cheng K-T., A three-stage deep learning model for accurate retinal vessel segmentation, IEEE Journal of Biomedical and Health Informatics, 23, 4, pp. 1427-1436, (2018)