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


Celik G., TALU M. F.

BIOMEDICAL SIGNAL PROCESSING AND CONTROL, cilt.71, 2022 (SCI-Expanded) identifier

  • Yayın Türü: Makale / Tam Makale
  • Cilt numarası: 71
  • Basım Tarihi: 2022
  • Doi Numarası: 10.1016/j.bspc.2021.103155
  • Dergi Adı: BIOMEDICAL SIGNAL PROCESSING AND CONTROL
  • Derginin Tarandığı İndeksler: Science Citation Index Expanded (SCI-EXPANDED), Scopus, Compendex, EMBASE, INSPEC
  • Anahtar Kelimeler: Brain segmentation, 3D MRI scans, GAN, ACFP, PAM
  • İnönü Üniversitesi Adresli: Evet

Özet

Background and aim: Although 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. Methods: The 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. Conclusions: In 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 page1.