PESA R-CNN: Perihematomal Edema Guided Scale Adaptive R-CNN for Hemorrhage Segmentation

초록

Intracranial hemorrhage (ICH) is a type of stroke with a high mortality rate and failing to localize even minor ICH can put a patient's life at risk. However, its patterns are diverse in shapes and sizes and, sometimes, even hard to recognize its existence. Therefore, it is challenging to accurately detect and localize diverse ICH patterns. In this article, we propose a novel Perihematomal Edema Guided Scale Adaptive R-CNN (PESA R-CNN) for accurate segmentation of various size hemorrhages with the goal of minimizing missed hemorrhage regions. In our approach, we design a Center Surround Difference U-Net (CSD U-Net) to incorporate Perihematomal Edema (PHE) for more accurate Region of Interest (RoI) generation. We trained CSD U-Net to predict PHE and hemorrhage regions as targets in a weakly supervised manner and utilized its prediction results to generate RoI. By including more informative features of PHE around hemorrhage, this enhanced RoI generation allows a model to reduce the false-negative rate. Furthermore, these expanded RoIs are aligned with the Scale Adaptive RoI Align (SARA) module based on their size to prevent the loss of fine-scale information and small hemorrhage patterns. Each scale adaptively aligned RoI is processed with the corresponding separate segmentation network of Multi-Scale Segmentation Network (MSSN), which integrates the results from each scale's segmentation network. In experiments, our model shows significant improvement on dice coefficient (0.697) and Hausdorff distance (12.918), compared to all other segmentation models. It also minimizes the number of missing small hemorrhage regions and enhances overall segmentation performance on diverse ICH patterns.

키워드