A Foreground Prototype-Based One-Shot Segmentation of Brain Tumors

Balasundaram Ananthakrishnan, Kavitha Subash Muthu, Pratheepan Yogarajah, Akshat Dhamale, MV Kaushik

Research output: Contribution to journalArticlepeer-review

22 Downloads (Pure)

Abstract

The potential for enhancing brain tumor segmentation with few-shot learning is enormous. While several deep learning networks (DNNs) show promising segmentation results, they all take a substantial amount of training data in order to yield appropriate results. Moreover, a prominent problem for most of these models is to perform well in unseen classes. To overcome these challenges, we propose a one-shot learning model to segment brain tumors on brain magnetic resonance images (MRI) based on a single prototype similarity score. With the use of recently developed few-shot learning techniques, where training and testing are carried out utilizing support and query sets of images, we attempt to acquire a definitive tumor region by focusing on slices containing foreground classes. It is unlike other recent DNNs that employed the entire set of images. The training of this model is carried out in an iterative manner where in each iteration, random slices containing foreground classes of randomly sampled data are selected as the query set, along with a different random slice from the same sample as the support set. In order to differentiate query images from class prototypes, we used a metric learning-based approach based on non-parametric thresholds. We employed the multimodal Brain Tumor Image Segmentation (BraTS) 2021 dataset with 60 training images and 350 testing images. The effectiveness of the model is evaluated using the mean dice score and mean IoU score. The experimental results provided a dice score of 83.42 which was greater than other works in the literature. Additionally, the proposed one-shot segmentation model outperforms the conventional methods in terms of computational time, memory usage, and the number of data.
Original languageEnglish
Article number1282
Pages (from-to)1-14
Number of pages15
JournalDiagnostics
Volume13
Issue number7
Early online date28 Mar 2023
DOIs
Publication statusPublished online - 28 Mar 2023

Bibliographical note

Funding Information:
This work was supported by research project grant number 3884293414, Nagasaki University. The authors wish to express their thanks to the Vellore Institute of Technology (VIT) management and Nagasaki University for their extensive support during this work.

Publisher Copyright:
© 2023 by the authors.

Keywords

  • Magnetic Resonance Imaging;
  • Few-shot learning
  • Foreground prototypes
  • magnetic resonance imaging
  • foreground prototypes
  • few-shot learning

Fingerprint

Dive into the research topics of 'A Foreground Prototype-Based One-Shot Segmentation of Brain Tumors'. Together they form a unique fingerprint.

Cite this