Prototype-Based Approach for One-Shot Segmentation of Brain Tumors using Few-Shot Learning

TL;DR

This paper introduces a one-shot learning model for brain tumor segmentation in MRI images, utilizing a prototype similarity score and metric learning to improve performance on unseen classes with limited data.

Contribution

The proposed approach is a novel one-shot segmentation method that leverages prototype similarity and metric learning, addressing data scarcity and generalization issues in brain tumor segmentation.

Findings

Achieved competitive mean dice scores on BraTS 2021 dataset.

Demonstrated robustness in segmenting unseen tumor classes.

Reduced reliance on large annotated datasets for effective segmentation.

Abstract

The potential for augmenting the segmentation of brain tumors through the use of few-shot learning is vast. Although several deep learning networks (DNNs) demonstrate promising results in terms of segmentation, they require a substantial quantity of training data in order to produce suitable outcomes. Furthermore, a major issue faced by most of these models is their ability to perform well when faced with unseen classes. To address these challenges, we propose a one-shot learning model for segmenting brain tumors in magnetic resonance images (MRI) of the brain, based on a single prototype similarity score. Leveraging the recently developed techniques of few-shot learning, which involve the utilization of support and query sets of images for training and testing purposes, we strive to obtain a definitive tumor region by focusing on slices that contain foreground classes. This approach differs from other recent DNNs that utilize the entire set of images. The training process for this model is carried out iteratively, with each iteration involving the selection of random slices that contain foreground classes from randomly sampled data as the query set, along with a different random slice from the same sample as the support set. In order to distinguish the query images from the class prototypes, we employ a metric learning-based approach that relies on non-parametric thresholds. We employ the multimodal Brain Tumor Image Segmentation (BraTS) 2021 dataset, which comprises 60 training images and 350 testing images. The effectiveness of the model is assessed using the mean dice score and mean Intersection over Union (IoU) score.