Meta transfer learning for brain tumor segmentation using nnUNet in meningioma and metastasis cases.
Authors
Affiliations (4)
Affiliations (4)
- Centre for Health Informatics, Australian Institute of Health Innovation, Macquarie University, 75, Talavera Road, North Ryde, Sydney, 2113, NSW, Australia. [email protected].
- Computational Neurosurgery Lab, Macquarie Medical School, Macquarie University, 75, Talavera Road, North Ryde, Sydney, 2113, NSW, Australia. [email protected].
- Computational Neurosurgery Lab, Macquarie Medical School, Macquarie University, 75, Talavera Road, North Ryde, Sydney, 2113, NSW, Australia.
- Centre for Health Informatics, Australian Institute of Health Innovation, Macquarie University, 75, Talavera Road, North Ryde, Sydney, 2113, NSW, Australia.
Abstract
In recent years, numerous algorithms have emerged for the segmentation of brain tumors, driven by advancements in deep learning techniques, where the objective is to identify and delineate various tumor sub-regions. While deep learning models like nnUNet have shown promising results in glioma segmentation, their effectiveness in segmenting other brain tumor subtypes, such as meningiomas and metastases, remains uncertain, especially when the available dataset lacks representative examples. To address this challenge, we propose a meta-transfer learning approach, which involves fine-tuning the nnUNet model on datasets containing meningiomas and metastases while leveraging the knowledge acquired from glioma segmentation. This approach aims to enhance the adaptability of nnUNet, allowing it to generalize better to diverse brain tumor types and potentially improving the accuracy of diagnosis and treatment planning for patients with meningiomas and metastases. Our proposed method significantly improves segmentation performance, achieving Dice coefficients of 0.8621 ± 0.2413 for Whole Tumor (WT) in meningiomas and 0.8141 ± 0.0562 for WT in metastases. These results set a new benchmark in brain tumor segmentation and pave the way for more robust and generalizable medical image analysis tools.