A triplanar ensemble model for brain tumor segmentation with volumetric multiparametric magnetic resonance images
Automated segmentation methods can produce faster segmentation of tumors in medical images, aiding medical professionals in diagnosis and treatment plans. A 3D U-Net method excels in this task but has high computational costs due to large model parameters, which limits their application under resour...
Main Authors: | , , , |
---|---|
Format: | Article |
Language: | English |
Published: |
Elsevier
2024-06-01
|
Series: | Healthcare Analytics |
Subjects: | |
Online Access: | http://www.sciencedirect.com/science/article/pii/S2772442524000091 |
_version_ | 1797319411766394880 |
---|---|
author | Snehal Rajput Rupal Kapdi Mohendra Roy Mehul S. Raval |
author_facet | Snehal Rajput Rupal Kapdi Mohendra Roy Mehul S. Raval |
author_sort | Snehal Rajput |
collection | DOAJ |
description | Automated segmentation methods can produce faster segmentation of tumors in medical images, aiding medical professionals in diagnosis and treatment plans. A 3D U-Net method excels in this task but has high computational costs due to large model parameters, which limits their application under resource constraints. This study targets an optimized triplanar (2.5D) model ensemble to generate accurate segmentation with fewer parameters. The proposed triplanar model uses spatial and channel attention mechanisms and information from multiple orthogonal planar views to predict segmentation labels. In particular, we studied the optimum filter size to improve the accuracy without increasing the network complexity. The model generated output is further post-processed to fine-tune the segmentation results. The Dice similarity coefficients (Dice-score) of the Brain Tumor Segmentation (BraTS) 2020 training set for enhancing tumor (ET), whole tumor (WT), and tumor core (TC) are 0.736, 0.896, and 0.841, whereas, for the validation set, they are 0.713, 0.873, and 0.778, respectively. The proposed base model has only 10.25M parameters, three times less than BraTS 2020’s best-performing model (ET 0.798, WT 0.912, TC 0.857) on the validation set. The proposed ensemble model has 93.5M parameters, 1.6 times less than the top-ranked model and two times less than the third-ranked model (ET 0.793, WT 0.911, TC 0.853 on validation set) of BraTS2020 challenge. |
first_indexed | 2024-03-08T04:06:31Z |
format | Article |
id | doaj.art-56fd457d981a4bd98003294306d4d433 |
institution | Directory Open Access Journal |
issn | 2772-4425 |
language | English |
last_indexed | 2024-03-08T04:06:31Z |
publishDate | 2024-06-01 |
publisher | Elsevier |
record_format | Article |
series | Healthcare Analytics |
spelling | doaj.art-56fd457d981a4bd98003294306d4d4332024-02-09T04:49:33ZengElsevierHealthcare Analytics2772-44252024-06-015100307A triplanar ensemble model for brain tumor segmentation with volumetric multiparametric magnetic resonance imagesSnehal Rajput0Rupal Kapdi1Mohendra Roy2Mehul S. Raval3School of Technology, Pandit Deendayal Energy University, Gandhinagar, 382007, Gujarat, IndiaInstitute of Technology, Nirma University, Ahmedabad, 382481, Gujarat, IndiaSchool of Technology, Pandit Deendayal Energy University, Gandhinagar, 382007, Gujarat, India; Corresponding author.School of Engineering and Applied Science, Ahmedabad University, Ahmedabad, 380009, Gujarat, India; Corresponding author.Automated segmentation methods can produce faster segmentation of tumors in medical images, aiding medical professionals in diagnosis and treatment plans. A 3D U-Net method excels in this task but has high computational costs due to large model parameters, which limits their application under resource constraints. This study targets an optimized triplanar (2.5D) model ensemble to generate accurate segmentation with fewer parameters. The proposed triplanar model uses spatial and channel attention mechanisms and information from multiple orthogonal planar views to predict segmentation labels. In particular, we studied the optimum filter size to improve the accuracy without increasing the network complexity. The model generated output is further post-processed to fine-tune the segmentation results. The Dice similarity coefficients (Dice-score) of the Brain Tumor Segmentation (BraTS) 2020 training set for enhancing tumor (ET), whole tumor (WT), and tumor core (TC) are 0.736, 0.896, and 0.841, whereas, for the validation set, they are 0.713, 0.873, and 0.778, respectively. The proposed base model has only 10.25M parameters, three times less than BraTS 2020’s best-performing model (ET 0.798, WT 0.912, TC 0.857) on the validation set. The proposed ensemble model has 93.5M parameters, 1.6 times less than the top-ranked model and two times less than the third-ranked model (ET 0.793, WT 0.911, TC 0.853 on validation set) of BraTS2020 challenge.http://www.sciencedirect.com/science/article/pii/S2772442524000091Automated segmentationBrain tumorEnsembleMachine learningMedical image analysis |
spellingShingle | Snehal Rajput Rupal Kapdi Mohendra Roy Mehul S. Raval A triplanar ensemble model for brain tumor segmentation with volumetric multiparametric magnetic resonance images Healthcare Analytics Automated segmentation Brain tumor Ensemble Machine learning Medical image analysis |
title | A triplanar ensemble model for brain tumor segmentation with volumetric multiparametric magnetic resonance images |
title_full | A triplanar ensemble model for brain tumor segmentation with volumetric multiparametric magnetic resonance images |
title_fullStr | A triplanar ensemble model for brain tumor segmentation with volumetric multiparametric magnetic resonance images |
title_full_unstemmed | A triplanar ensemble model for brain tumor segmentation with volumetric multiparametric magnetic resonance images |
title_short | A triplanar ensemble model for brain tumor segmentation with volumetric multiparametric magnetic resonance images |
title_sort | triplanar ensemble model for brain tumor segmentation with volumetric multiparametric magnetic resonance images |
topic | Automated segmentation Brain tumor Ensemble Machine learning Medical image analysis |
url | http://www.sciencedirect.com/science/article/pii/S2772442524000091 |
work_keys_str_mv | AT snehalrajput atriplanarensemblemodelforbraintumorsegmentationwithvolumetricmultiparametricmagneticresonanceimages AT rupalkapdi atriplanarensemblemodelforbraintumorsegmentationwithvolumetricmultiparametricmagneticresonanceimages AT mohendraroy atriplanarensemblemodelforbraintumorsegmentationwithvolumetricmultiparametricmagneticresonanceimages AT mehulsraval atriplanarensemblemodelforbraintumorsegmentationwithvolumetricmultiparametricmagneticresonanceimages AT snehalrajput triplanarensemblemodelforbraintumorsegmentationwithvolumetricmultiparametricmagneticresonanceimages AT rupalkapdi triplanarensemblemodelforbraintumorsegmentationwithvolumetricmultiparametricmagneticresonanceimages AT mohendraroy triplanarensemblemodelforbraintumorsegmentationwithvolumetricmultiparametricmagneticresonanceimages AT mehulsraval triplanarensemblemodelforbraintumorsegmentationwithvolumetricmultiparametricmagneticresonanceimages |