Fusion Branch Network with Class Learning Difficulty Loss Function for Recongnizition of Haematoma Expansion Signs in Intracerebral Haemorrhage

Ye Yang,Shanxiong Chen,Duo Tan,Rui Yao,Shiyu Zhu,Yuanyuan Jia,Wensong Yang,Yiqing Shen
DOI: https://doi.org/10.1109/bibm52615.2021.9669423
2021-01-01
Abstract:The automatic identification of hematoma expansion signs is very important to the diagnosis and treatment of intracerebral hemorrhage. However, the brain computed tomography samples is uneven distribution, low variation in imaging performance between signs, developing such a solution is challenging.In this article, a novel deep learning network is presented to recognize hematoma expansion signs automatically, which provides a new method to assist in the diagnosis of hematoma growth. First, we propose the fusion branching network (FBN) for extracting and fusing the high-dimensional features of each category dynamically. Second, this paper designs a Class Learning Difficulty (CLD) loss function to dynamically adjust the class weights based on the FBN learning situation to help the model avoid learning bias and improve the performance in signs recognition. In addition, a visualization component is provided to improve the transparency of the model. In the experiment, we retrospectively collected DICOM images from the First Affiliated Hospital of C hongqing Medical University, in order to form a recognition process consistent with clinical work, the data is divided into three categories: Blend Sign (BS), Black Hole Sign (BHS), ICH without BS and BHS (ICHWBB) for sign recognition. The experimental results demonstrate that our method performed the best in terms of comprehensive evaluation indexes compared with other methods, and the recognition sensitivity reached 0.9404, 0.8056, 0.7586 for ICHWBB, BHS, and BS, respectively, which was significantly improved compared with other methods in the same environment.
What problem does this paper attempt to address?