Training Hacks and a Frugal Man's Net with Application to Glioblastoma Segmentation

Abdallah J. B., Marrakchi-Kacem L., Rekık I.

6th International Conference on Advanced Technologies for Signal and Image Processing, ATSIP 2022, Sfax, Tunisia, 24 - 27 May 2022 identifier

  • Publication Type: Conference Paper / Full Text
  • Doi Number: 10.1109/atsip55956.2022.9805950
  • City: Sfax
  • Country: Tunisia
  • Keywords: Brain Tumour, Low-Budget, Multiclass-Segmentation, U-Net
  • Istanbul Technical University Affiliated: Yes


© 2022 IEEE.In this paper, we investigate the effectiveness of training a sparse Neural Network on a limited number of samples in the context of brain tumor segmentation. Nowadays, Deep Learning architectures are getting deeper, more sophisticated and environmentally unfriendly in an effort to improve their segmentation performance. We use a brain tumor segmentation dataset and apply simple practices to reduce the needed computational resources to allow cheap and fast training. We also present a lighter, cheaper version of the U-Net dubbed Frugal U-Net stemming from our investigation on how far we can push the original U-Net by decreasing its parameter count using Depth-Wise Separable Convolutions instead of regular ones, all the while preserving the minimum levels of accuracy required in Medical Imaging. Our methodology is useful in clinical facilities where high-computation resources are limited.