论文标题

基于变压器的生成对抗网络,用于脑肿瘤分割

A Transformer-based Generative Adversarial Network for Brain Tumor Segmentation

论文作者

Huang, Liqun, Chen, Long, Zhang, Baihai, Chai, Senchun

论文摘要

在医学图像分割任务中,脑肿瘤分割仍然是一个挑战。随着变压器在各种计算机视觉任务中的应用,变压器块显示了在全球空间中学习长距离依赖性的能力,这是与CNN互补的。在本文中,我们提出了一个新型的基于变压器的生成对抗网络,以自动分割具有多模式MRI的脑肿瘤。我们的架构由一个发电机和一个歧视器组成,该发电机和歧视器接受了最低游戏的进度。发电机基于典型的“ U形”编码器架构,其底层由带有Resnet的变压器块组成。此外,发电机还接受了深入监督技术的培训。我们设计的歧视器是一个基于CNN的网络,具有多尺度$ L_ {1} $损失,事实证明,这对于医学语义图像分割是有效的。为了验证我们方法的有效性,我们对BRATS2015数据集进行了实验,比以前的最先进方法实现了可比或更好的性能。

Brain tumor segmentation remains a challenge in medical image segmentation tasks. With the application of transformer in various computer vision tasks, transformer blocks show the capability of learning long-distance dependency in global space, which is complementary with CNNs. In this paper, we proposed a novel transformer-based generative adversarial network to automatically segment brain tumors with multi-modalities MRI. Our architecture consists of a generator and a discriminator, which are trained in min-max game progress. The generator is based on a typical "U-shaped" encoder-decoder architecture, whose bottom layer is composed of transformer blocks with resnet. Besides, the generator is trained with deep supervision technology. The discriminator we designed is a CNN-based network with multi-scale $L_{1}$ loss, which is proved to be effective for medical semantic image segmentation. To validate the effectiveness of our method, we conducted experiments on BRATS2015 dataset, achieving comparable or better performance than previous state-of-the-art methods.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源