Architecture Optimization for Hybrid Deep Residual Networks in Liver Tumor Segmentation Using a GA

Autor: Mohamed Reyad, Amany M. Sarhan, M. Arafa
Jazyk: angličtina
Rok vydání: 2024
Předmět:
Zdroj: International Journal of Computational Intelligence Systems, Vol 17, Iss 1, Pp 1-22 (2024)
Druh dokumentu: article
ISSN: 1875-6883
DOI: 10.1007/s44196-024-00542-4
Popis: Abstract Nowadays, liver tumor segmentation has been widely applied for vital medical objectives such as illness diagnosis, treatment, and evaluation of liver function. In this work, we aim to improve the performance of hybrid convolution neural network (CNN) models for liver tumor segmentation. A new automatic design method is introduced to build a hybrid CNN model with an optimal architecture using a combination of the original U_Net and the Res-UNet. In the proposed design method, the Res-UNet model is divided into different blocks, and then many different candidate architectures for the hybrid CNN model can be created using gradually these blocks to replace their corresponding blocks of the original U-Net. The replacement is executed if the candidate architecture using a residual block gives better performance than the corresponding one that uses the original U-Net block. We use a genetic algorithm (GA) to get the optimal architecture among the candidate architectures that gives the best performance for the hybrid CNN model. During the optimization process, in addition to the optimal architecture, other crucial configurations such as the learning algorithm, learning rate, and batch size are also determined. We refer to the hybrid CNN model that has the best architecture as GA_Res_UNet. Comparison has been made with other CNN models that are commonly used in image segmentation of liver tumors. We use the 3D-IRCADb01 liver tumor dataset that contains 2800 magnetic resonance (MR) images of 20 patients. Each image has a liver tumor with its corresponding mask. All MR images are resized to 256 × 256 in DICOM format. The findings indicate that GA_Res_UNet outperformed the other compared models over the considered dataset. It achieves 98.5% of dice coefficient and 99.8% of accuracy. Moreover, two other liver tumor segmentation datasets (LiTS17 and CHAOS) are utilized in another comparison to show the superior performance of our proposed model. The reported results show that the proposed model outperformed the other compared state-of-the-art models.
Databáze: Directory of Open Access Journals