A Two-Stage Generative Model for Unsupervised Brain Tumor Detection Using CycleGAN and Diffusion Models

References

[1] I. Goodfellow et al., “Generative adversarial nets,” Advances in Neural Information Processing Systems (NeurIPS), pp. 2672–2680, 2014.

[2] D. P. Kingma and M. Welling, “Auto-Encoding Variational Bayes,” International Conference on Learning Representations (ICLR), 2014.

[3] J.-Y. Zhu, T. Park, P. Isola, and A. A. Efros, “Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks,” IEEE International Conference on Computer Vision (ICCV), pp. 2223–2232, 2017.

[4] J. Ho, A. Jain, and P. Abbeel, “Denoising Diffusion Probabilistic Models,” Advances in Neural Information Processing Systems (NeurIPS), vol. 33, pp. 6840–6851, 2020.

[5] Y. Song, J. Sohl-Dickstein, D. P. Kingma, A. Kumar, S. Ermon, and B. Poole, “Score-Based Generative Modeling through Stochastic Differential Equations,” International Conference on Learning Representations (ICLR), 2021.

[6] O. Ronneberger, P. Fischer, and T. Brox, “U-Net: Convolutional Networks for Biomedical Image Segmentation,” International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI), pp. 234–241, 2015.

[7] A. Goodfellow, Y. Bengio, and A. Courville, Deep Learning, MIT Press, 2016.

[8] A. Schlegl et al., “Unsupervised Anomaly Detection with Generative Adversarial Networks to Guide Marker Discovery,” International Conference on Information Processing in Medical Imaging (IPMI), pp. 146–157, 2017.

[9] T. Wolleb, F. Bieder, J. Sandkühler, and P. C. Cattin, “Diffusion Models for Medical Anomaly Detection,” IEEE Transactions on Medical Imaging, vol. 41, no. 12, pp. 1–12, 2022.

[10] P. Isola, J.-Y. Zhu, T. Zhou, and A. A. Efros, “Image-to-Image Translation with Conditional Adversarial Networks,” IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 1125–1134, 2017.

[11] X. Wang et al., “Unsupervised Brain Tumor Segmentation Using Generative Models,” IEEE Access, vol. 8, pp. 103593–103606, 2020.

[12] H. Chen et al., “Unsupervised Medical Image Segmentation Using GANs,” Medical Image Analysis, vol. 57, pp. 1–13, 2019.

[13] K. Kamnitsas et al., “Efficient Multi-Scale 3D CNN with Fully Connected CRF for Accurate Brain Lesion Segmentation,” Medical Image Analysis, vol. 36, pp. 61–78, 2017.

[14] A. Madabhushi and G. Lee, “Image Analysis and Machine Learning in Digital Pathology: Challenges and Opportunities,” Medical Image Analysis, vol. 33, pp. 170–175, 2016.

[15] Wenxin Wang et al., “A Two-Stage Generative Model with CycleGAN and Joint Diffusion for MRI-based Brain Tumor Detection,” IEEE Journal of Biomedical and Health Informatics, vol. 28, no. 6, pp. 3534–3543, June 2024.

[16] B. Menze et al., “The Multimodal Brain Tumor Image Segmentation Benchmark (BRATS),” IEEE Transactions on Medical Imaging, vol. 34, no. 10, pp. 1993–2024, 2015.

[17] S. Bakas et al., “Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features,” Scientific Data, vol. 4, pp. 1–13, 2017.

[18] F. Milletari, N. Navab, and S. Ahmadi, “V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation,” International Conference on 3D Vision (3DV), pp. 565–571, 2016.

[19] K. He, X. Zhang, S. Ren, and J. Sun, “Deep Residual Learning for Image Recognition,” IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 770–778, 2016.

[20] G. Litjens et al., “A survey on deep learning in medical image analysis,” Medical Image Analysis, vol. 42, pp. 60–88, 2017.

[21] A. Krizhevsky, I. Sutskever, and G. Hinton, “ImageNet Classification with Deep Convolutional Neural Networks,” Advances in Neural Information Processing Systems (NeurIPS), pp. 1097–1105, 2012.

[22] H. Chen, Q. Dou, L. Yu, and P. Heng, “VoxResNet: Deep voxelwise residual networks for brain segmentation,” NeuroImage, vol. 170, pp. 446–455, 2018.

[23] J. Long, E. Shelhamer, and T. Darrell, “Fully Convolutional Networks for Semantic Segmentation,” IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 3431–3440, 2015.

[24] M. Havaei et al., “Brain tumor segmentation with Deep Neural Networks,” Medical Image Analysis, vol. 35, pp. 18–31, 2017.

[25] A. Myronenko, “3D MRI brain tumor segmentation using autoencoder regularization,” International MICCAI Brainlesion Workshop, pp. 311–320, 2018.

[26] L. Pinaya et al., “Brain Imaging Generation with Latent Diffusion Models,” International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI), 2022.

[27] A. Creswell et al., “Generative Adversarial Networks: An Overview,” IEEE Signal Processing Magazine, vol. 35, no. 1, pp. 53–65, 2018.

[28] C. Szegedy et al., “Going deeper with convolutions,” IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 1–9, 2015.

[29] S. Pereira et al., “Brain Tumor Segmentation Using Convolutional Neural Networks in MRI Images,” IEEE Transactions on Medical Imaging, vol. 35, no. 5, pp. 1240–1251, 2016.

[30] N. Bousselham et al., “Brain tumor segmentation using deep neural networks,” Journal of Biomedical Informatics, vol. 87, pp. 1–12, 2018.

[31] Y. LeCun, Y. Bengio, and G. Hinton, “Deep learning,” Nature, vol. 521, pp. 436–444, 2015.

[32] J. Duncan and N. Ayache, “Medical image analysis: Progress over two decades and the challenges ahead,” IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 22, no. 1, pp. 85–106, 2000.

[33] T. Zhou et al., “Unsupervised learning for medical image segmentation using generative models,” IEEE Access, vol. 7, pp. 178118–178130, 2019.

[34] A. Radford, L. Metz, and S. Chintala, “Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks,” International Conference on Learning Representations (ICLR), 2016.

[35] D. Rezende, S. Mohamed, and D. Wierstra, “Stochastic Backpropagation and Approximate Inference in Deep Generative Models,” International Conference on Machine Learning (ICML), pp. 1278–1286, 2014.

@article{1715054,
      author = {B. Manasa, K. D. V. Lakshmi Sanjana, K. Gowtham, G. Tejaswini, V. Akila},
      title = {A Two-Stage Generative Model for Unsupervised Brain Tumor Detection Using CycleGAN and Diffusion Models},
      journal = {Iconic Research And Engineering Journals},
      year = {2026},
      volume = {9},
      number = {9},
      pages = {816-825},
      issn = {2456-8880},
      url = {https://www.irejournals.com/formatedpaper/1715054.pdf},
      abstract = {Brain tumor detection from MRI images is a critical task in medical image analysis, yet it remains challenging due to limited labeled data, complex tumor structures, and high annotation costs. Traditional supervised learning methods depend heavily on large annotated datasets, which are often difficult to obtain in real clinical scenarios. To address these challenges, this research proposes a two-stage unsupervised generative learning framework for brain tumor detection and segmentation. In the first stage, a CycleGAN model is employed to generate synthetic abnormal MRI images [3] from healthy MRI images, creating pseudo-paired healthy–abnormal data without manual annotation. In the second stage, a conditional diffusion-based generative model is applied [4],[5] to reconstruct healthy images from abnormal inputs by learning the joint distribution between healthy and abnormal image pairs. The difference between the reconstructed healthy image and the original abnormal image is used to localize tumor regions. The proposed system is evaluated using confusion matrix–based performance metrics including precision, recall, F1-score, accuracy, and Dice similarity coefficient. Experimental results demonstrate that the proposed framework effectively detects and segments tumor regions, achieving reliable performance without requiring labele training data. This approach provides a robust, scalable, and annotation-free solution for medical image anomaly detection and brain tumor analysis.},
      keywords = {Brain Tumor Detection, MRI, Cyclegan, Diffusion Model, Unsupervised Learning, Anomaly Detection, Medical Image Segmentation.},
      month = {March},
      doi = {https://doi.org/10.64388/IREV9I9-1715054}
  }