An Exhaustive Analytical Study of U-Net Architecture on Two Diverse Biomedical Imaging Datasets of Electron Microscopy Drosophila ssTEM and Brain MRI BraTS-2021 for Segmentation

Biswajit Jena,Gopal Krishna Nayak,Sudip Paul,Sanjay Saxena
DOI: https://doi.org/10.1007/s42979-022-01347-y
2022-08-05
SN Computer Science
Abstract:Biomedical image segmentation is a mechanism that distinguishes the boundaries of various lesion regions within the image, be it 2D or 3D. In the current scenario, various soft and hard computing approaches have been used for medical image segmentation purposes. This article analyzes deep learning-based U-Net architecture for efficient segmentation by possible variations of its different important hyper-parameters that affect the U-Net performance. This article also investigates U-Net architecture's performance to generate the segmented output over two diverse biomedical imaging datasets, such as brain MRI scans and serial section Transmission Electron Microscopy (ssTEM) dataset of the drosophila first instar larva VNC, hence making 48 possible combinations of execution of the U-Net model. The performance of the segmentation process has been evaluated using metrics like dice similarity coefficient and accuracy, while cross-entropy is considered the loss function. In doing so, an insight into the overall performance of U-Net on biomedical image segmentation problems has been obtained. The best results for accuracy and dice coefficient values out of all possible combinations made in this study come as 0.9881 and 0.9934 , respectively. The main motive of this work is to provide an exhaustive and efficient analysis of the U-Net architecture for biomedical image segmentation and subsequent analysis.
What problem does this paper attempt to address?