Abstract:Brain tumors are a pernicious cancer with one of the lowest five-year survival rates. Neurologists often use magnetic resonance imaging (MRI) to diagnose the type of brain tumor. Automated computer-assisted tools can help them speed up the diagnosis process and reduce the burden on the health care systems. Recent advances in deep learning for medical imaging have shown remarkable results, especially in the automatic and instant diagnosis of various cancers. However, we need a large amount of data (images) to train the deep learning models in order to obtain good results. Large public datasets are rare in medicine. This paper proposes a framework based on unsupervised deep generative neural networks to solve this limitation. We combine two generative models in the proposed framework: variational autoencoders (VAEs) and generative adversarial networks (GANs). We swap the encoder–decoder network after initially training it on the training set of available MR images. The output of this swapped network is a noise vector that has information of the image manifold, and the cascaded generative adversarial network samples the input from this informative noise vector instead of random Gaussian noise. The proposed method helps the GAN to avoid mode collapse and generate realistic-looking brain tumor magnetic resonance images. These artificially generated images could solve the limitation of small medical datasets up to a reasonable extent and help the deep learning models perform acceptably. We used the ResNet50 as a classifier, and the artificially generated brain tumor images are used to augment the real and available images during the classifier training. We compared the classification results with several existing studies and state-of-the-art machine learning models. Our proposed methodology noticeably achieved better results. By using brain tumor images generated artificially by our proposed method, the classification average accuracy improved from 72.63% to 96.25%. For the most severe class of brain tumor, glioma, we achieved 0.769, 0.837, 0.833, and 0.80 values for recall, specificity, precision, and F1-score, respectively. The proposed generative model framework could be used to generate medical images in any domain, including PET (positron emission tomography) and MRI scans of various parts of the body, and the results show that it could be a useful clinical tool for medical experts.

Discriminative Hamiltonian Variational Autoencoder for Accurate Tumor Segmentation in Data-Scarce Regimes

End-to-end autoencoding architecture for the simultaneous generation of medical images and corresponding segmentation masks

3D MRI Synthesis with Slice-Based Latent Diffusion Models: Improving Tumor Segmentation Tasks in Data-Scarce Regimes

Mask-guided Data Augmentation for Multiparametric MRI Generation with a Rare Hepatocellular Carcinoma

A statistical deformation model-based data augmentation method for volumetric medical image segmentation

Advancing Volumetric Medical Image Segmentation via Global-Local Masked Autoencoder

Toward Accurate Cardiac MRI Segmentation With Variational Autoencoder-Based Unsupervised Domain Adaptation

Brain Tumor Classification Using a Combination of Variational Autoencoders and Generative Adversarial Networks

Towards Accurate Cardiac MRI Segmentation with Variational Autoencoder-Based Unsupervised Domain Adaptation

Multi-modal Brain Tumor Segmentation Using Stacked Denoising Autoencoders

A Robust Volumetric Transformer for Accurate 3D Tumor Segmentation

SegTransVAE: Hybrid CNN -- Transformer with Regularization for medical image segmentation

Diminishing Uncertainty Within the Training Pool: Active Learning for Medical Image Segmentation

Hybrid Window Attention Based Transformer Architecture for Brain Tumor Segmentation

3D High-Resolution Cardiac Segmentation Reconstruction from 2D Views using Conditional Variational Autoencoders

Brain Lesion Synthesis via Progressive Adversarial Variational Auto-Encoder

Hetero-Modal Variational Encoder-Decoder for Joint Modality Completion and Segmentation

Modeling the Probabilistic Distribution of Unlabeled Data forOne-shot Medical Image Segmentation

Variational Autoencoders for Feature Exploration and Malignancy Prediction of Lung Lesions

Data Augmentation in High Dimensional Low Sample Size Setting Using a Geometry-Based Variational Autoencoder

Medical variational autoencoder and generative adversarial network for medical imaging