A new 3D MRI segmentation method based on Generative Adversarial Network and Atrous Convolution

Abstract

Background and aimAlthough many algorithms have been proposed to segment brain structures in MRI scans, comparison of different algorithms in the same data set is rarely done. Many methods still run on privately held data and include comparisons with previous versions. The purpose of this study is to introduce a new Generative Adversarial Network (GAN) based segmentation architecture. Brain tissues on 3D-MRI scans with T1w modality were segmented into three (GM, WM, CSF) and eight (CGM, BG, WM, WMH, CF, VE, CE ve BS) parts. MethodsThe proposed approach (Vol2SegGAN) consists of two parts: preprocessing and segmentation. The preprocessing includes extraction of the brain region, editing of labels in the datasets, MNI152 registration, clipping/sampling. The segmentation part is carried out by the collaboration of a generator (includes ACFP and PAM modules) and a discriminator (distinguishes real/fake) architectures. ConclusionsIn the three part segmentation process, the proposed method showed the best segmentation success in CSF (Dice = 0.739, VS = 0.967), GM (Dice = 0.878, HD = 2.378) and WM (HD = 2.105) tissues, and the second best segmentation success in WM (Dice = 0.793, VS = 0.972) tissue according to Dice and VS metrics. Similarly, in the segmentation of eight parts, it is seen that it has the best success according to VS and HD metrics and the second best according to Dice metric. Vol2SegGAN, which has fewer parameters (6,883 mil.) than existing architectures, has an average of 11–12 s (CPU) or 0–1 s (GPU) to segment a sample 3D-MRI. Implementation codes of the proposed architecture are available on the github page11(https://github.com/GaffariCelik/Vol2SegGAN).