DOI QR코드

DOI QR Code

Diagnosis and Visualization of Intracranial Hemorrhage on Computed Tomography Images Using EfficientNet-based Model

전산화 단층 촬영(Computed tomography, CT) 이미지에 대한 EfficientNet 기반 두개내출혈 진단 및 가시화 모델 개발

  • Youn, Yebin (Korea Brain Research Institute, Cognitive Science Group, Deep Memory Lab) ;
  • Kim, Mingeon (Siemens Healthineers Ltd. Diagnostic Imaging) ;
  • Kim, Jiho (Korea Brain Research Institute, Cognitive Science Group, Deep Memory Lab) ;
  • Kang, Bongkeun (Daegu-Gyeongbuk Medical Innovation Foundation, Medical Device Development Center) ;
  • Kim, Ghootae (Korea Brain Research Institute, Cognitive Science Group, Deep Memory Lab)
  • Received : 2021.03.05
  • Accepted : 2021.07.13
  • Published : 2021.08.31

Abstract

Intracranial hemorrhage (ICH) refers to acute bleeding inside the intracranial vault. Not only does this devastating disease record a very high mortality rate, but it can also cause serious chronic impairment of sensory, motor, and cognitive functions. Therefore, a prompt and professional diagnosis of the disease is highly critical. Noninvasive brain imaging data are essential for clinicians to efficiently diagnose the locus of brain lesion, volume of bleeding, and subsequent cortical damage, and to take clinical interventions. In particular, computed tomography (CT) images are used most often for the diagnosis of ICH. In order to diagnose ICH through CT images, not only medical specialists with a sufficient number of diagnosis experiences are required, but even when this condition is met, there are many cases where bleeding cannot be successfully detected due to factors such as low signal ratio and artifacts of the image itself. In addition, discrepancies between interpretations or even misinterpretations might exist causing critical clinical consequences. To resolve these clinical problems, we developed a diagnostic model predicting intracranial bleeding and its subtypes (intraparenchymal, intraventricular, subarachnoid, subdural, and epidural) by applying deep learning algorithms to CT images. We also constructed a visualization tool highlighting important regions in a CT image for predicting ICH. Specifically, 1) 27,758 CT brain images from RSNA were pre-processed to minimize the computational load. 2) Three different CNN-based models (ResNet, EfficientNet-B2, and EfficientNet-B7) were trained based on a training image data set. 3) Diagnosis performance of each of the three models was evaluated based on an independent test image data set: As a result of the model comparison, EfficientNet-B7's performance (classification accuracy = 91%) was a way greater than the other models. 4) Finally, based on the result of EfficientNet-B7, we visualized the lesions of internal bleeding using the Grad-CAM. Our research suggests that artificial intelligence-based diagnostic systems can help diagnose and treat brain diseases resolving various problems in clinical situations.

Keywords

Acknowledgement

본 연구는 과학기술정보통신부의 재원으로 한국뇌연구원의 KBRI 기초 연구 프로그램의 지원을 받아 수행된 연구임(21-BR-03-04).

References

  1. Davenport T, Kalakota R. The potential for artificial intelligence in healthcare. Future Healthcare Journal, 2019;6(2):94-98. https://doi.org/10.7861/futurehosp.6-2-94
  2. Liu GD, Li YC, Zhang W, Zhang L. A Brief Review of Artificial Intelligence Applications and Algorithms for Psychiatric Disorders. Engineering, 2020;6(4):462-467. https://doi.org/10.1016/j.eng.2019.06.008
  3. Marietta M, Pedrazzi P, Girardis M, Torelli G. Intracerebral haemorrhage: An often neglected medical emergency. Internal and Emergency Medicine, 2007;2(1):38-45. https://doi.org/10.1007/s11739-007-0009-y
  4. Caceres JA, Goldstein JN. Intracranial Hemorrhage. Emergency Medicine Clinics of North America, 2012;30(3):771-794. https://doi.org/10.1016/j.emc.2012.06.003
  5. Ye H, Gao F, Yin Y, Guo D, Zhao P, Lu Y, Wang X, Bai J, Cao K, Song Q, Zhang H, Chen W, Guo X, Xia J. Precise diagnosis of intracranial hemorrhage and subtypes using a three-dimensional joint convolutional and recurrent neural network. European Radiology, 2019;29(11):6191-6201. https://doi.org/10.1007/s00330-019-06163-2
  6. Alfaro D, Levitt MA, English DK, Williams V, Eisenberg R. Accuracy of interpretation of cranial computed tomography scans in an emergency medicine residency program. Annals of Emergency Medicine, 1995;25(2):169-174. https://doi.org/10.1016/S0196-0644(95)70319-5
  7. Lal NR, Murray UM, Eldevik OP, Desmond JS. Clinical consequences of misinterpretations of neuroradiologic CT scans by on-call radiology residents. AJNR. American Journal of Neuroradiology, 2000;21(1):124-129.
  8. Strub WM, Leach JL, Tomsick T, Vagal A. Overnight Preliminary Head CT Interpretations Provided by Residents: Locations of Misidentified Intracranial Hemorrhage. American Journal of Neuroradiology, 2007;28(9):1679-1682. https://doi.org/10.3174/ajnr.a0653
  9. Arbabshirani MR, Plis S, Sui J, Calhoun VD. Single subject prediction of brain disorders in neuroimaging: Promises and pitfalls. NeuroImage, 2017;145:137-165. https://doi.org/10.1016/j.neuroimage.2016.02.079
  10. He K, Zhang X, Ren S, Sun J. Deep Residual Learning for Image Recognition. 2015;ArXiv:1512.03385 [Cs].
  11. Tan M, Le QV. EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks. 2020;ArXiv:1905.11946 [Cs, Stat].
  12. Chilamkurthy S, Ghosh R, Tanamala S, Biviji M, Campeau NG, Venugopal VK, Mahajan V, Rao P, Warier P. Deep learning algorithms for detection of critical findings in head CT scans: a retrospective study. Lancet, 2018;392:2388-96. https://doi.org/10.1016/S0140-6736(18)31645-3
  13. Alfidi RJ, Haaga J, Meaney TF, MacIntyre WJ, Gonzalez L, Tarar R, Zelch MG, Boller M, Cook SA, Jelden G. Computed tomography of the thorax and abdomen: a preliminary report. Radiology, 1975;117:257-264. https://doi.org/10.1148/117.2.257
  14. Kim CH, Jung JI. Study for hounsfield units in computed tomogram with jaw lesion. J Korean Assoc Oral Maxillofac Surg, 2006;32:391-396.
  15. Lin T, Goyal P, Girshick R, He K, Dollar P. Focal Loss for Dense Object Detection. 2017;arXiv:1708.02002.
  16. Selvaraju RR, Cogswell M, Das A, Vedantam R, Parikh D, Batra D. Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localization. International Journal of Computer Vision, 2020;128(2):336-359. https://doi.org/10.1007/s11263-019-01228-7
  17. Pope PE, Kolouri S, Rostami M, Martin CE, Hoffmann H. Explainability Methods for Graph Convolutional Neural Networks. 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2019;10764-10773.
  18. Xie Q, Luong M, Hovy E, Le QV. Self-Training with Noisy Student Improves ImageNet Classification. 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2020;10687-10698.