Novel hybrid optimized clustering schemes with genetic algorithm and PSO for segmentation and classification of articular cartilage loss from MR images

Abstract

Medical image segmentation plays an indispensable role in the identification of articular cartilage, tibial and femoral bones from magnetic resonance imaging (MRI). There are various image segmentation strategies that can be used to identify the knee structures of interest. Among the most popular are the methods based on non-hierarchical clustering, including the algorithms K-means and fuzzy C-means (FCM). Although these algorithms have been used in many studies for regional image segmentation, they have two essential drawbacks that limit their performance and accuracy of segmentation. Firstly, they rely on a precise selection of initial centroids, which is usually conducted randomly, and secondly, these algorithms are sensitive enough to image noise and artifacts, which may deteriorate the segmentation performance. Based on such limitations, we propose, in this study, two novel alternative metaheuristic hybrid schemes: non-hierarchical clustering, driven by a genetic algorithm, and Particle Swarm Optimization (PSO) with fitness function, which utilizes Kapur’s entropy and statistical variance. The goal of these optimization elements is to find the optimal distribution of centroids for the knee MR image segmentation model. As a part of this study, we provide comprehensive testing of the robustness of these novel segmentation algorithms upon the image noise generators. This includes Gaussian, Speckle, and impulsive Salt and Pepper noise with dynamic noise to objectively report the robustness of the proposed segmentation strategies in contrast with conventional K-means and FCM. This study reveals practical applications of the proposed algorithms for articular cartilage extraction and the consequent classification performance of early osteoarthritis based on segmentation models and convolutional neural networks (CNN). Here, we provide a comparative analysis of GoogLeNet and ResNet 18 with various hyperparameter settings, where we achieved 99.92% accuracy for the best classification configuration for early cartilage loss recognition.