MLP-GAN for Brain Vessel Image Segmentation
- URL: http://arxiv.org/abs/2207.08265v1
- Date: Sun, 17 Jul 2022 19:24:38 GMT
- Title: MLP-GAN for Brain Vessel Image Segmentation
- Authors: Bin Xie, Hao Tang, Bin Duan, Dawen Cai, Yan Yan
- Abstract summary: Brain vessel image segmentation can be used as a promising biomarker for better prevention and treatment of different diseases.
One successful approach is to consider the segmentation as an image-to-image translation task and to learn a conditional Generative Adversarial Network (cGAN) to learn a transformation between two distributions.
We present a novel multi-view approach, which perform a 3D volumetric brain vessel image into three different 2D images (i.e., sagittal, coronal, axial) and then feed them into three different 2D cGANs.
Our model obtains the ability to capture cross-patch information
- Score: 19.807219907693145
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Brain vessel image segmentation can be used as a promising biomarker for
better prevention and treatment of different diseases. One successful approach
is to consider the segmentation as an image-to-image translation task and
perform a conditional Generative Adversarial Network (cGAN) to learn a
transformation between two distributions. In this paper, we present a novel
multi-view approach, MLP-GAN, which splits a 3D volumetric brain vessel image
into three different dimensional 2D images (i.e., sagittal, coronal, axial) and
then feed them into three different 2D cGANs. The proposed MLP-GAN not only
alleviates the memory issue which exists in the original 3D neural networks but
also retains 3D spatial information. Specifically, we utilize U-Net as the
backbone for our generator and redesign the pattern of skip connection
integrated with the MLP-Mixer which has attracted lots of attention recently.
Our model obtains the ability to capture cross-patch information to learn
global information with the MLP-Mixer. Extensive experiments are performed on
the public brain vessel dataset that show our MLP-GAN outperforms other
state-of-the-art methods. We release our code at
https://github.com/bxie9/MLP-GAN
Related papers
- 3DPX: Progressive 2D-to-3D Oral Image Reconstruction with Hybrid MLP-CNN Networks [16.931777224277347]
Panoramic X-ray (PX) is a prevalent modality in dental practice for its wide availability and low cost.
As a 2D projection image, PX does not contain 3D anatomical information.
We propose a progressive hybrid Multilayer Perceptron (MLP)-CNN pyra-mid network (3DPX) for 2D-to-3D oral PX reconstruction.
arXiv Detail & Related papers (2024-08-02T14:28:10Z) - ToNNO: Tomographic Reconstruction of a Neural Network's Output for Weakly Supervised Segmentation of 3D Medical Images [6.035125735474387]
ToNNO is based on the Tomographic reconstruction of a Neural Network's Output.
It extracts stacks of slices with different angles from the input 3D volume, feeds these slices to a 2D encoder, and applies the inverse Radon transform in order to reconstruct a 3D heatmap of the encoder's predictions.
We apply it to weakly supervised medical image segmentation by training the 2D encoder to output high values for slices containing the regions of interest.
arXiv Detail & Related papers (2024-04-19T11:27:56Z) - Multi-View Vertebra Localization and Identification from CT Images [57.56509107412658]
We propose a multi-view vertebra localization and identification from CT images.
We convert the 3D problem into a 2D localization and identification task on different views.
Our method can learn the multi-view global information naturally.
arXiv Detail & Related papers (2023-07-24T14:43:07Z) - MLP-3D: A MLP-like 3D Architecture with Grouped Time Mixing [123.43419144051703]
We present a novel-like 3D architecture for video recognition.
The results are comparable to state-of-the-art widely-used 3D CNNs and video.
arXiv Detail & Related papers (2022-06-13T16:21:33Z) - UNeXt: MLP-based Rapid Medical Image Segmentation Network [80.16644725886968]
UNet and its latest extensions like TransUNet have been the leading medical image segmentation methods in recent years.
We propose UNeXt which is a Convolutional multilayer perceptron based network for image segmentation.
We show that we reduce the number of parameters by 72x, decrease the computational complexity by 68x, and improve the inference speed by 10x while also obtaining better segmentation performance.
arXiv Detail & Related papers (2022-03-09T18:58:22Z) - Cascaded Cross MLP-Mixer GANs for Cross-View Image Translation [70.00392682183515]
It is hard to generate an image at target view well for previous cross-view image translation methods.
We propose a novel two-stage framework with a new Cascaded Cross-Mixer (CrossMLP) sub-network.
In the first stage, the CrossMLP sub-network learns the latent transformation cues between image code and semantic map code.
In the second stage, we design a refined pixel-level loss that eases the noisy semantic label problem.
arXiv Detail & Related papers (2021-10-19T18:03:30Z) - RepMLP: Re-parameterizing Convolutions into Fully-connected Layers for
Image Recognition [123.59890802196797]
We propose RepMLP, a multi-layer-perceptron-style neural network building block for image recognition.
We construct convolutional layers inside a RepMLP during training and merge them into the FC for inference.
By inserting RepMLP in traditional CNN, we improve ResNets by 1.8% accuracy on ImageNet, 2.9% for face recognition, and 2.3% mIoU on Cityscapes with lower FLOPs.
arXiv Detail & Related papers (2021-05-05T06:17:40Z) - MLP-Mixer: An all-MLP Architecture for Vision [93.16118698071993]
We present-Mixer, an architecture based exclusively on multi-layer perceptrons (MLPs).
Mixer attains competitive scores on image classification benchmarks, with pre-training and inference comparable to state-of-the-art models.
arXiv Detail & Related papers (2021-05-04T16:17:21Z) - Spatial Context-Aware Self-Attention Model For Multi-Organ Segmentation [18.76436457395804]
Multi-organ segmentation is one of most successful applications of deep learning in medical image analysis.
Deep convolutional neural nets (CNNs) have shown great promise in achieving clinically applicable image segmentation performance on CT or MRI images.
We propose a new framework for combining 3D and 2D models, in which the segmentation is realized through high-resolution 2D convolutions.
arXiv Detail & Related papers (2020-12-16T21:39:53Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.