An evolutionary Chameleon Swarm Algorithm based network for 3D medical image segmentation

Abstract

Developing 3D Convolutional Neural Networks (CNNs) for medical image segmentation is challenging due to the limited number of available labeled medical images and computational resources. Effective design of a 3D CNN requires a better structure and appropriate hyperparameters. Manually designing a neural network and choosing appropriate hyperparameters requires extensive knowledge, a time-consuming and challenging process, resulting in complex and over-parameterized networks. Neuroevolution is an automatic and effective approach to finding an optimal network structure and hyperparameters. This paper proposes an evolutionary based method using Chameleon Swarm Algorithm (CSA) to develop 3D CNN networks for volume segmentation. We also design a search space to build optimal 3D block-based encoder–decoder structure networks for medical image segmentation. We evaluated the proposed method on two different datasets (CT Spleen and MRI Heart) and compared the results of the obtained model with state-of-the-art models. Furthermore, the comparative results show that our proposed method yields 3D CNN networks with superior performance and fewer parameters than other state-of-the-art models utilized in 3D medical image segmentation.