ReSGAN: Intracranial Hemorrhage Segmentation with Residuals of Synthetic Brain CT Scans

초록

Intracranial hemorrhage (ICH) is a dangerous condition of bleeding within the skull that calls for rapid and precise diagnosis due to potentially fatal consequences. In this paper, we propose Residual Segmentation with Generative Adversarial Networks (ReSGAN) to accurately localize the hemorrhage from computerized tomography (CT) scans with a GAN-based model. Although convolutional neural networks have shown success in the ICH segmentation task, precise localization remains challenging due to in-balance and scarcity of labeled training data. Synthetic samples from generative models, and aligned templates as reference from brain atlas have been demonstrated to alleviate the issues. We consider synthetic templates as another candidate and solve the problem by directly applying a generative model to segmentation. Our ReSGAN learns a distribution of pseudo-normal brain CT scans, that through residuals, reliably delineates the hemorrhaging areas. We perform experiments on two datasets and compare our model against a well established baseline, that consistently shows significant improvements, therefore demonstrating the validity of our novel method.

키워드