Application of Generative Adversarial Networks Based on Global and Local Feature Information in Hippocampus Segmentation

Abstract

Due to the complex structure and small size of the hippocampus, precise segmentation of the hippocampus remains challenging. To address this issue, this study proposes a generative adversarial network (GAN) based on global and local feature information (GLGAN) for hippocampus segmentation. First, to improve network stability and segmentation accuracy while reducing the likelihood of problems such as information loss and gradient explosion, we proposed the global GAN (GGAN) by optimizing the generator and loss function of GAN. Second, since the discriminator is essentially a binary classifier and is not sensitive to small local changes, we introduced a GAN method of dual discriminator network structure that integrates both global and local feature information. Finally, a total loss function was designed to balance GAN adversarial loss and 3D u-net segmentation loss. The experimental results show that proposed method based on GLGAN facilitates intensive evaluation of the hippocampus, and drives the discriminator to push the mask value provided by the generator to a more realistic distribution, thereby enhancing hippocampus segmentation accuracy. The Dice coefficient and IOU for hippocampus segmentation using GLGAN are 0.804 and 0.672 respectively.

Keywords

• generative adversarial network (gan)
• 3d cnn
• segmentation
• hippocampus
• 3d u-net