Abstract
Introduction: Brain hemorrhage is one of the leading causes of death due to the sudden rupture of a blood vessel in the brain, resulting in bleeding in the brain parenchyma. The early detection and segmentation of brain damage are extremely important for prompt treatment.
Methods: Some previous studies focused on localizing cerebral hemorrhage based on bounding boxes without specifying specific damage regions. However, in practice, doctors need to detect and segment the hemorrhage area more accurately. In this paper, we propose a method for automatic brain hemorrhage detection and segmentation using the proposed network models, which are improved from the U-Net by changing its backbone with typical feature extraction networks, i.e., DenseNet-121, ResNet-50, and MobileNet-V2. The U-Net architecture has many outstanding advantages.
Results: It does not need to do too many preprocessing techniques on the original images and it can be trained with a small dataset providing low error segmentation in medical images. We use the transfer learning approach with the head CT dataset gathered on Kaggle including two classes, bleeding and non-bleeding.
Conclusion: Besides, we give some comparison results between the proposed models and the previous works to provide an overview of the suitable model for cerebral CT images. On the head CT dataset, our proposed models achieve a segmentation accuracy of up to 99%.
[1]
Broderick J, Connolly S, Feldmann E, et al. Guidelines for the management of spontaneous intracerebral hemorrhage in adults. Stroke 2007; 38(6): 2001-23.
[http://dx.doi.org/10.1161/STROKEAHA.107.183689] [PMID: 17478736]
[http://dx.doi.org/10.1161/STROKEAHA.107.183689] [PMID: 17478736]
[2]
Aguilar MI, Freeman WD. Spontaneous intracerebral hemorrhage. Semin Nephrol 2010; 30(5): 555-64.
[http://dx.doi.org/10.1055/s-0030-1268865]
[http://dx.doi.org/10.1055/s-0030-1268865]
[3]
van Asch CJJ, Luitse MJA, Rinkel GJE, van der Tweel I, Algra A, Klijn CJM. Incidence, case fatality, and functional outcome of intracerebral haemorrhage over time, according to age, sex, and ethnic origin: A systematic review and meta-analysis. Lancet Neurol 2010; 9(2): 167-76.
[http://dx.doi.org/10.1016/S1474-4422(09)70340-0] [PMID: 20056489]
[http://dx.doi.org/10.1016/S1474-4422(09)70340-0] [PMID: 20056489]
[4]
Elliott J, Smith M. The acute management of intracerebral hemorrhage: A clinical review. Anesth Analg 2010; 110(5): 1419-27.
[http://dx.doi.org/10.1213/ANE.0b013e3181d568c8] [PMID: 20332192]
[http://dx.doi.org/10.1213/ANE.0b013e3181d568c8] [PMID: 20332192]
[5]
Currie S, Saleem N, Straiton JA, Macmullen-Price J, Warren DJ, Craven IJ. Imaging assessment of traumatic brain injury. Postgrad Med J 2016; 92(1083): 41-50.
[http://dx.doi.org/10.1136/postgradmedj-2014-133211] [PMID: 26621823]
[http://dx.doi.org/10.1136/postgradmedj-2014-133211] [PMID: 26621823]
[6]
Xue Z, Antani S, Long LR, Demner-Fushman D, Thoma GR. Window classification of brain CT images in biomedical articles. AMIA Annu Symp Proc 2012; 2012: 1023-9.
[7]
Wu Y, Supanich MP, Deng J. Ensembled deep neural network for intracranial hemorrhage detection and subtype classification on noncontrast CT images. Artif Intell Med 2021; 2(1-2): 12-20.
[http://dx.doi.org/10.2991/jaims.d.210618.001]
[http://dx.doi.org/10.2991/jaims.d.210618.001]
[8]
Burduja M, Ionescu RT, Verga N. Accurate and efficient intracranial hemorrhage detection and subtype classification in 3D CT scans with convolutional and long short-term memory neural networks. Sensors 2020; 20(19): 5611.
[http://dx.doi.org/10.3390/s20195611] [PMID: 33019508]
[http://dx.doi.org/10.3390/s20195611] [PMID: 33019508]
[9]
Danilov G, Kotik K, Negreeva A, et al. Classification of intracranial hemorrhage subtypes using deep learning on CT scans. Stud Health Technol Inform 2020; 272: 370-3.
[http://dx.doi.org/10.3233/SHTI200572] [PMID: 32604679]
[http://dx.doi.org/10.3233/SHTI200572] [PMID: 32604679]
[10]
Sage A, Badura P. Intracranial hemorrhage detection in head CT using double-branch convolutional neural network, support vector machine, and random forest. Appl Sci 2020; 10(21): 7577.
[http://dx.doi.org/10.3390/app10217577]
[http://dx.doi.org/10.3390/app10217577]
[11]
Castro JS, Chabert S, Saavedra C, Salas R. Convolutional neural networks for detection intracranial hemorrhage in CT images. CRoNe 2019; 2564: 37-43.
[12]
He J. Automated detection of intracranial hemorrhage on head computed tomography with deep learning. ICBET 2020: Proceedings of the 2020 10th International Conference on Biomedical Engineering and Technology. 25-28 March, Tokyo, Japan, 2020, pp. 117-121.
[http://dx.doi.org/10.1145/3397391.3397436]
[http://dx.doi.org/10.1145/3397391.3397436]
[13]
Santhoshkumar S, Varadarajan V, Gavaskar S, Amalraj JJ, Sumathi A. Machine learning model for intracranial hemorrhage diagnosis and classification. Electronics 2021; 10(21): 2574.
[http://dx.doi.org/10.3390/electronics10212574]
[http://dx.doi.org/10.3390/electronics10212574]
[14]
Meng F, Wang J, Zhang H, Li W. Artificial intelligence-enabled medical analysis for intracranial cerebral hemorrhage detection and classification. J Healthc Eng 2022; 2022: 1-13.
[http://dx.doi.org/10.1155/2022/2017223] [PMID: 35356628]
[http://dx.doi.org/10.1155/2022/2017223] [PMID: 35356628]
[15]
Chang PD, Kuoy E, Grinband J, et al. Hybrid 3D/2D convolutional neural network for hemorrhage evaluation on head CT. AJNR Am J Neuroradiol 2018; 39(9): 1609-16.
[http://dx.doi.org/10.3174/ajnr.A5742] [PMID: 30049723]
[http://dx.doi.org/10.3174/ajnr.A5742] [PMID: 30049723]
[16]
Reis EP, Nascimento F, Aranha M, et al. Brain Hemorrhage Extended (BHX): Bounding box extrapolation from thick to thin slice CT images. PhysioNet 2020; 101(23): e215-20.
[17]
Nemcek J, Vicar T, Jakubicek R. Weakly supervised deep learning-based intracranial hemorrhage localization. arXiv:210500781 2021.
[18]
Ertuğrul ÖF, Akıl MF. Detecting hemorrhage types and bounding box of hemorrhage by deep learning. Biomed Signal Process Control 2022; 71: 103085.
[http://dx.doi.org/10.1016/j.bspc.2021.103085]
[http://dx.doi.org/10.1016/j.bspc.2021.103085]
[19]
Phan AC, Tran HD, Phan TC. Efficient brain hemorrhage detection on 3D CT scans with deep neural network. , FDSE 2021, Virtual Event, November 24–26, 2021. Future Data and Security Engineering: 8th International Conference. Virtual Event, November 24–26, 2021, pp. 81-96.
[20]
Chen J, Lu Y, Yu Q, et al. Transunet: Transformers make strong encoders for medical image segmentation arXiv:210204306 2021.
[21]
Wang W, Chen C, Ding M, Yu H, Zha S, Li J. Transbts: Multimodal brain tumor segmentation using transformer. Medical Image Computing and Computer Assisted Intervention–MICCAI 2021: 24th International Conference trasbourg, France, September 27–October 1, 2021, pp. 109-119
[22]
Wu J, Fang H, Zhang Y, Yang Y, Xu Y. Medsegdiff: Medical image segmentation with diffusion probabilistic model. arXiv:221100611 2022.
[23]
Sharma N, Sharma R, Jindal N. Machine learning and deep learning applications-a vision. Glob Transit Proc 2021; 2(1): 24-8.
[http://dx.doi.org/10.1016/j.gltp.2021.01.004]
[http://dx.doi.org/10.1016/j.gltp.2021.01.004]
[24]
Ronneberger O, Fischer P, Brox T. U-net: Convolutional networks for biomedical image segmentation. Medical Image Computing and Computer-Assisted Intervention–MICCAI 2015: 18th International Conference October 5-9, Munich, Germany, 2015, pp. 234-241.
[25]
Çiçek Ö, Abdulkadir A, Lienkamp SS, Brox T, Ronneberger O. 3D U-Net: Learning dense volumetric segmentation from sparse annotation. Medical Image Computing and Computer-Assisted Intervention–MICCAI 2016: 19th International Conference October 17-21, Athens, Greece, 2016, pp. 424-434.
[26]
Hssayeni MD, Croock MS, Salman AD, Al-khafaji HF, Yahya ZA, Ghoraani B. Intracranial hemorrhage segmentation using a deep convolutional model. Data 2020; 5(1): 14.
[http://dx.doi.org/10.3390/data5010014]
[http://dx.doi.org/10.3390/data5010014]
[27]
Falk T, Mai D, Bensch R, et al. U-Net: Deep learning for cell counting, detection, and morphometry. Nat Methods 2019; 16(1): 67-70.
[http://dx.doi.org/10.1038/s41592-018-0261-2] [PMID: 30559429]
[http://dx.doi.org/10.1038/s41592-018-0261-2] [PMID: 30559429]
[28]
Kohl S, Romera-Paredes B, Meyer C, et al. A probabilistic u-net for segmentation of ambiguous images. Adv Neural Inf Process Syst 2018; 31.
[29]
He K, Zhang X, Ren S, Sun J. Deep residual learning for image recognition. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). Las Vegas, NV, USA, 2016, pp. 770-778.
[http://dx.doi.org/10.1109/CVPR.2016.90]
[http://dx.doi.org/10.1109/CVPR.2016.90]
[30]
Rahimzadeh M, Attar A. A modified deep convolutional neural network for detecting COVID-19 and pneumonia from chest X-ray images based on the concatenation of Xception and ResNet50V2. Inform Med Unlocked 2020; 19: 100360.
[31]
Huang G, Liu Z, Van Der Maaten L, Weinberger KQ. Densely connected convolutional networks. 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). Honolulu, HI, USA, 2017, pp. 2261-2269.
[http://dx.doi.org/10.1109/CVPR.2017.243]
[http://dx.doi.org/10.1109/CVPR.2017.243]
[32]
Howard AG, Zhu M, Chen B, et al. Mobilenets: Efficient convolutional neural networks for mobile vision applications. arXiv:170404861 2017.
[33]
Sandler M, Howard A, Zhu M, Zhmoginov A, Chen LC. Mobilenetv2: Inverted residuals and linear bottlenecks. Proceedings of the IEEE conference on computer vision and pattern recognition. Salt Lake City, UT, USA, 2018, pp. 4510-4520.
[34]
Xiang Q, Wang X, Li R, Zhang G, Lai J, Hu Q. Fruit image classification based on MobileNetV2 with transfer learning technique. Proceedings of the 3rd International Conference on Computer Science and Application Engineering. Sanya, China, 2019, pp. 1-7
[http://dx.doi.org/10.1145/3331453.3361658]
[http://dx.doi.org/10.1145/3331453.3361658]
[35]
Dai W, Dai Y, Hirota K, Jia Z. A flower classification approach with mobileNetV2 and transfer learning. Proceedings of the 9th International Symposium on Computational Intelligence and Industrial Applications (ISCIIA2020). Beijing, China, 2020.
[36]
Gomez R. Understanding categorical cross-entropy loss, binary cross-entropy loss, softmax loss, logistic loss, focal loss and all those confusing names. 2018.https://gombru.github.io/2018/ 05/23/cross_entropy_loss/
