Automatic segmentation of intracranial hemorrhage using coarse and fine deep learning models

Abstract

To save patients’ lives, it is important to go for an early diagnosis of intracranial hemorrhage (ICH). For diagnosing ICH, the widely used method is non-contrast computed tomography (NCCT). It has fast acquisition and availability in medical emergency facilities. To predict hematoma progression and mortality, it is important to estimate the volume of intracranial hemorrhage. Radiologists can manually delineate the ICH region to estimate the hematoma volume. This process takes time and undergoes inter-rater variability. In this research paper, we develop and discuss a fine segmentation model and a coarse model for intracranial hemorrhage segmentations. Basically, two different models are discussed for intracranial hemorrhage segmentation. We trained a 2DDensNet in the first model for coarse segmentation and cascaded the coarse segmentation mask output in the fine segmentation model along with input training samples. A nnUNet model is trained in the second fine stage and will use the segmentation labels of the coarse model with true labels for intracranial hemorrhage segmentation. An optimal performance for intracranial hemorrhage segmentation solution is obtained.