12 - On comparing optimizer of UNet-VGG16 architecture for brain tumor image segmentation

Abstract

Image segmentation as computer-based diagnostic systems in brain tumor magnetic resonance imaging (MRI) plays important role in supporting the medical diagnosis. The clear shape of tumor visualization is the main objective in image segmentation. Several methods have been developed in the deep learning area under the U-Net architecture. The UNet-VGG16 with transfer learning + dropout is a new architecture that hybrids the U-Net with VGG-16 added by the transfer learning + dropout regularization. The dropout scenario is a way to minimize the overfitting effect as the nature of VGG-16 contains different nonlinear hidden layers and complex relationships that may result in overfitting. This new architecture is proved to be fast and accurate in segmenting the MRI of low-grade gliomas in fluid-attenuated inversion recovery sequence. However, the previous model is yet to concern the effect of the different optimizers on the overfitting issue. Therefore, this study aims to investigate the best optimizer for UNet-VGG16 with transfer learning + dropout that could deal with overfitting. This study had succeeded to recognize Adamax as the best optimizer.