Abstract
Brain tumour segmentation plays a critical role in the diagnosis, treatment planning, and monitoring of brain tumour patients. However, accurate and efficient segmentation remains challenging due to the complex and heterogeneous structure of brain tumour regions. The current CNN models have shown good performance in brain tumour segmentation and identification, but several research challenges, like limited generalizability, Adaptive Model Complexity, etc., still need to be addressed. In this research, we propose a novel approach that combines the progressively growing and One-Shot learning approaches with a semantic segmentation network to enhance the accuracy and generalization of brain tumour segmentation in MRI images. Our method joins the strengths of progressively growing and One-Shot learning techniques with a semantic segmentation network, enabling improved generalization, effective feature selection, and continuous integration of contextual information at the pixel level. Experimental results on benchmark Br35H MRI image datasets demonstrate the dominance of our approach over existing methods in terms of segmentation accuracy and adaptability to diverse brain tumour instances. A total of 3000 images (1500 tumorous and 1500 non-tumorous images) were used during the training and testing of the model. The evaluation metrics reveal the high performance of our proposed model for brain tumour segmentation. Achieving high Dice Similarity Coefficients (0.9849), Intersection over Union (0.9319), accuracy (0.9520), precision (0.9235), and recall (0.9572) across average training, validation, and test sets. These results demonstrate the model's efficiency in accurately segmenting both tumorous and non-tumorous regions in MRI images.
Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.Data availability
The data used in this paper can be accessed from the KAGGLE online database.
https://www.kaggle.com/datasets/ahmedhamada0/brain-tumor-detection
References
Kang J, Ullah Z, Gwak J (2021) Mri-based brain tumor classification using ensemble of deep features and machine learning classifiers. Sensors 21(6):1–21. https://doi.org/10.3390/s21062222
Gómez-Guzmán MA et al (2023) Classifying Brain Tumors on Magnetic Resonance Imaging by Using Convolutional Neural Networks. Electron 12(4):1–22. https://doi.org/10.3390/electronics12040955
Gu X, Shen Z, Xue J, Fan Y, Ni T (2021) Brain Tumor MR Image Classification Using Convolutional Dictionary Learning With Local Constraint. Front Neurosci 15(May):1–12. https://doi.org/10.3389/fnins.2021.679847
Nazir M, Shakil S, Khurshid K (2021) Role of deep learning in brain tumor detection and classification (2015 to 2020): A review. Comput Med Imaging Graph 91:101940. https://doi.org/10.1016/J.COMPMEDIMAG.2021.101940
Ranjbarzadeh R, Caputo A, Tirkolaee EB, Jafarzadeh Ghoushchi S, Bendechache M (2023) Brain tumor segmentation of MRI images: A comprehensive review on the application of artificial intelligence tools. Comput Biol Med 152:106405. https://doi.org/10.1016/j.compbiomed.2022.106405
American Society of Clinical Oncology 2020 Brain Tumor: Diagnosis | Cancer.Net. https://www.cancer.net/cancer-types/brain-tumor/diagnosis (accessed Jun. 06, 2023).
Philip AK, Samuel BA, Bhatia S, Khalifa SAM, El-Seedi HR (2023) Artificial Intelligence and Precision Medicine: A New Frontier for the Treatment of Brain Tumors. Life 13(1):1–16. https://doi.org/10.3390/life13010024
Saeedi S, Rezayi S, Keshavarz H, Niakan Kalhori SR (2023) MRI-based brain tumor detection using convolutional deep learning methods and chosen machine learning techniques. BMC Med Inform Decis Mak 23(1):1–17. https://doi.org/10.1186/s12911-023-02114-6
Badža MM, Barjaktarović MC (2020) Classification of brain tumors from mri images using a convolutional neural network. Appl Sci 10(6):1999. https://doi.org/10.3390/app10061999
Arora A, Jayal A, Gupta M, Mittal P, Satapathy SC (2021) Brain tumor segmentation of MRI images using processed image driven u-net architecture. Computers 10(11):139. https://doi.org/10.3390/computers10110139
Liu Z et al (2023) Deep learning based brain tumor segmentation: a survey. Complex Intell Syst 9(1):1001–1026. https://doi.org/10.1007/s40747-022-00815-5
Vankdothu R et al (2022) Enhanced performance of brain tumor classification via tumor region augmentation and partition. PLoS ONE 11(6):1–20. https://doi.org/10.1371/journal.pone.0157112
Kazuhiro K et al (2018) Generative Adversarial Networks for the Creation of Realistic Artificial Brain Magnetic Resonance Images. Tomography 4(4):159–163. https://doi.org/10.18383/j.tom.2018.00042
Krishnapriya S, Karuna Y (2023) Pre-trained deep learning models for brain MRI image classification. Front Hum Neurosci 17:1150120. https://doi.org/10.3389/fnhum.2023.1150120
Irmak E (2021) Multi-Classification of Brain Tumor MRI Images Using Deep Convolutional Neural Network with Fully Optimized Framework. Iran J Sci Technol - Trans Electr Eng 45(3):1015–1036. https://doi.org/10.1007/s40998-021-00426-9
Chattopadhyay A, Maitra M (2022) MRI-based brain tumour image detection using CNN based deep learning method. Neurosci Informatics 2(4):100060. https://doi.org/10.1016/j.neuri.2022.100060
Howard AG et al (2017) MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications. Available: http://arxiv.org/abs/1704.04861. Accessed 11 Jun 2023
Zhou Z, Siddiquee MMR, Tajbakhsh N, Liang J (2020) UNet++: Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentation. IEEE Trans Med Imaging 39(6):1856–1867. https://doi.org/10.1109/TMI.2019.2959609
Yang Q, Li N, Zhao Z, Fan X, Chang EI-C, Xu Y (2018) MRI cross-modality neuroimage-to-neuroimage translation, no. Nannan Li, 2018, [Online]. Available: http://arxiv.org/abs/1801.06940. Accessed 9 Jun 2023
Ren X et al (2019) Task decomposition and synchronization for semantic biomedical image segmentation. arXiv 29:7497–7510
Motiian S, Jones Q, Iranmanesh SM, Doretto G (2017) Few-shot adversarial domain adaptation. Adv Neural Inf Process Syst 2017:6671–6681
Achmamad A, Ghazouani F, Ruan S (2022) Few-shot learning for brain tumor segmentation from MRI images. Int Conf Signal Process Proceedings, ICSP 2022:489–494. https://doi.org/10.1109/ICSP56322.2022.9965315
Khadka R et al (2022) Meta-learning with implicit gradients in a few-shot setting for medical image segmentation. Comput Biol Med 143:105227. https://doi.org/10.1016/j.compbiomed.2022.105227
Pambala AK, Dutta T, Biswas S (2021) SML: Semantic meta-learning for few-shot semantic segmentation☆. Pattern Recognit Lett 147:93–99. https://doi.org/10.1016/j.patrec.2021.03.036
Balasundaram A, Kavitha MS, Pratheepan Y, Akshat D, Kaushik MV (2023) A Foreground Prototype-Based One-Shot Segmentation of Brain Tumors. Diagnostics 13(7):1282. https://doi.org/10.3390/diagnostics13071282
Bakas S et al (2017) Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features. Sci Data 4(1):1–13. https://doi.org/10.1038/sdata.2017.117
Tian P, Wu Z, Qi L, Wang L, Shi Y, Gao Y (2020) Differentiable Meta-Learning Model for Few-Shot Semantic Segmentation. Proc AAAI Conf Artif Intell 34(07):12087–12094. https://doi.org/10.1609/AAAI.V34I07.6887
Alrashedy HHN, Almansour AF, Ibrahim DM, Hammoudeh MAA (2022) BrainGAN: Brain MRI Image Generation and Classification Framework Using GAN Architectures and CNN Models. Sensors 22(11):4297. https://doi.org/10.3390/s22114297
Ge C, Gu IYH, Jakola AS, Yang J (2020) Enlarged Training Dataset by Pairwise GANs for Molecular-Based Brain Tumor Classification. IEEE Access 8:22560–22570. https://doi.org/10.1109/ACCESS.2020.2969805
Han C et al., (2019) Learning more with less: Conditional PGGAN-based data augmentation for brain metastases detection using highly-rough annotation on MR images. Int Conf Inf Knowl Manag Proc pp. 119–127, https://doi.org/10.1145/3357384.3357890
Han C et al (2020) Infinite Brain MR Images: PGGAN-Based Data Augmentation for Tumor Detection. Smart Innov Syst Technol 151:291–303. https://doi.org/10.1007/978-981-13-8950-4_27/COVER
Han C et al (2019) Combining noise-to-image and image-to-image GANs: Brain MR image augmentation for tumor detection. IEEE Access 7:156966–156977. https://doi.org/10.1109/ACCESS.2019.2947606
Han C et al (2021) MADGAN: unsupervised medical anomaly detection GAN using multiple adjacent brain MRI slice reconstruction. BMC Bioinformatics 22(2):1–20. https://doi.org/10.1186/S12859-020-03936-1/TABLES/2
Ghassemi N, Shoeibi A, Rouhani M (2020) Deep neural network with generative adversarial networks pre-training for brain tumor classification based on MR images. Biomed Signal Process Control 57:101678. https://doi.org/10.1016/J.BSPC.2019.101678
Allah AMG, Sarhan AM, Elshennawy NM (2021) Classification of brain MRI tumor images based on deep learning PGGAN augmentation. Diagnostics 11(12):1–20. https://doi.org/10.3390/diagnostics11122343
Vankdothu R, Hameed MA (2022) Brain tumor MRI images identification and classification based on the recurrent convolutional neural network. Meas Sensors 24:100412. https://doi.org/10.1016/j.measen.2022.100412
Zulpe N, Pawar V (2012) GLCM textural features for Brain Tumor Classification. Int J Comput Sci 9(3):354–359 (http://www.doaj.org/doaj?func=abstract&id=1158398)
Samjith Raj CP, Shreeja R (2017) Automatic brain tumor tissue detection in T-1 weighted MRI. 2017 Int Conf Innov Inf Embed Commun Syst 2018:1–4. https://doi.org/10.1109/ICIIECS.2017.8276094
Minz A, Mahobiya C (2017) MR image classification using adaboost for brain tumor type. Proc- 7th IEEE Int Adv Comput Conf IACC 2017:701–705
Sharif MI, Li JP, Khan MA, Saleem MA (2020) Active deep neural network features selection for segmentation and recognition of brain tumors using MRI images. Pattern Recognit Lett 129:181–189. https://doi.org/10.1016/j.patrec.2019.11.019
Dong H, Yang G, Liu F, Mo Y, Guo Y (2017) Automatic brain tumor detection and segmentation using U-net based fully convolutional networks. Commun Comput Inf Sci 723:506–517. https://doi.org/10.1007/978-3-319-60964-5_44/COVER
Chen W, Liu B, Peng S, Sun J, Qiao X (2019) S3D-UNET: Separable 3D U-Net for brain tumor segmentation. Lect Notes Comput Sci (including Subser Lect Notes Artif Intell Lect Notes Bioinformatics) 11384:358–368. https://doi.org/10.1007/978-3-030-11726-9_32. (LNCS)
Tuan TA, Bao PT (2019) Brain tumor segmentation using bit-plane and UNET. Lect Notes Comput Sci (including Subser Lect Notes Artif Intell Lect Notes Bioinformatics) 11384:466–475. https://doi.org/10.1007/978-3-030-11726-9_41/COVER. (LNCS)
Ibtehaz N, Rahman MS (2020) MultiResUNet : Rethinking the U-Net architecture for multimodal biomedical image segmentation. Neural Netw 121:74–87. https://doi.org/10.1016/J.NEUNET.2019.08.025
Karayegen G, Aksahin MF (2021) Brain tumor prediction on MR images with semantic segmentation by using deep learning network and 3D imaging of tumor region. Biomed Signal Process Control 66:102458. https://doi.org/10.1016/J.BSPC.2021.102458
Wang Y et al (2021) Modality-Pairing Learning for Brain Tumor Segmentation. Lect Notes Comput Sci (including Subser Lect Notes Artif Intell Lect Notes Bioinformatics) 12658:230–240. https://doi.org/10.1007/978-3-030-72084-1_21/COVER. (LNCS)
Hamada A (2020) Br35H :: Brain Tumor Detection 2020, Kaggle. https://www.kaggle.com/datasets/ahmedhamada0/brain-tumor-detection (accessed Jun. 07, 2023)
Aggarwal M, Tiwari AK, Sarathi MP, Bijalwan A (2023) An early detection and segmentation of Brain Tumor using Deep Neural Network. BMC Med Inform Decis Mak 23(1):1–2. https://doi.org/10.1186/s12911-023-02174-8
Rajendran S et al (2023) Automated Segmentation of Brain Tumor MRI Images Using Deep Learning. IEEE Access 11(June):64758–64768. https://doi.org/10.1109/ACCESS.2023.3288017
Zhu Z, He X, Qi G, Li Y, Cong B, Liu Y (2023) Brain tumor segmentation based on the fusion of deep semantics and edge information in multimodal MRI. Inf Fusion 91:376–387. https://doi.org/10.1016/J.INFFUS.2022.10.022
Santosh Kumar P, Sakthivel VP, Raju M, Satya PD (2023) Brain tumor segmentation of the FLAIR MRI images using novel ResUnet. Biomed Signal Process Control 82:104586. https://doi.org/10.1016/j.bspc.2023.104586
Mahesh Kumar G, Parthasarathy E (2023) Development of an enhanced U-Net model for brain tumor segmentation with optimized architecture. Biomed Signal Process Control 81:104427. https://doi.org/10.1016/j.bspc.2022.104427
Da Costa Nascimento JJ et al., (2023) New Health of Things Approach to Classification and Detection of Brain Tumors Using Transfer Learning for Segmentation in IMR Images, Proc Int Joint Conf Neural Netw 2023, https://doi.org/10.1109/IJCNN54540.2023.10191399
Rehman MU, Ryu J, Nizami IF, Chong KT (2023) RAAGR2-Net: A brain tumor segmentation network using parallel processing of multiple spatial frames. Comput Biol Med 152:106426. https://doi.org/10.1016/j.compbiomed.2022.106426
Ejaz K, Suaib NBM, Kamal MS, Rahim MSM, Rana N (2023) Segmentation Method of Deterministic Feature Clustering for Identification of Brain Tumor Using MRI. IEEE Access 11(February):39695–39712. https://doi.org/10.1109/ACCESS.2023.3263798
Acknowledgements
We are very thankful to Wuhan University for her generous support in conducting this research.
Funding
This work was sponsored by the National Natural Science Foundation of China General Program with grant number (No. 6 2272352).
Author information
Authors and Affiliations
Contributions
Azmat Ali proposed the ideas, collected data, and wrote the manuscript. Yulin Wang and Xiaochuan Shi provided the idea and discussed the outline of the manuscript.
Corresponding author
Ethics declarations
Conflict of interest
The authors declare that there is no conflict of interest regarding the publication of research work carried out in this paper and the order of the authors in the manuscript.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Ali, A., Wang, Y. & Shi, X. Segmentation and identification of brain tumour in MRI images using PG-OneShot learning CNN model. Multimed Tools Appl (2024). https://doi.org/10.1007/s11042-024-18596-z
Received:
Revised:
Accepted:
Published:
DOI: https://doi.org/10.1007/s11042-024-18596-z