IHA-Net: An automatic segmentation framework for computer-tomography of tiny intracerebral hemorrhage based on improved attention U-net

Yue Ma,Fuquan Ren,Wenrui Li,Nanan Yu,Dejun Zhang,Yushuang Li,Mingchi Ke
DOI: https://doi.org/10.1016/j.bspc.2022.104320
IF: 5.1
2023-02-01
Biomedical Signal Processing and Control
Abstract:Intracerebral hemorrhage (ICH) is a serious category of head injury, which means bleeding occurs inside the skull, and in many cases leads to a high disability or mortality. Therefore, accurate and rapid hematoma region segmentation is of great significance for ICH diagnosis and treatment. Nevertheless, even with a powerful deep neural network, there are still certain limitations in the case of tiny ICH regions. In this paper, a novel model based on convolutional neural network for the segmentation of hematoma regions in brain Computed Tomography images of patients with ICH is developed. Specifically, in order to prevent the degradation of tiny lesion regions due to repeated down-sampling, we also proposed a Residual Hybrid Atrous Convolution strategy. Furthermore, a multi-object function for joint optimization is introduced to ease the severe pixel imbalanced problem, which is of benefit to training procedure. In addition, to avoid gradient disappearance and slow convergence, intermediate supervision with a lightweight head is utilized during the training. The performance of developed segmentation algorithm for Intracranial hemorrhage is analyzed by using Dice coefficient, Jaccard index, Sensitivity and Accuracy. From both quantitative and visual results, we found that the proposed model outperformed the previous methods and achieved a satisfying segmentation performance for Intracranial hemorrhage test in both internal testing data (Dice of 0.725, Jaccard of 0.605) and external clinical data (Dice of 0.869, Jaccard of 0.774), which indicates that proposed model has better segmentation effect and potential clinical prospect.
engineering, biomedical
What problem does this paper attempt to address?