An improved particle swarm optimization for multilevel thresholding medical image segmentation

PLoS One. 2024 Dec 31;19(12):e0306283. doi: 10.1371/journal.pone.0306283. eCollection 2024.

Abstract

Multilevel thresholding image segmentation is one of the widely used image segmentation methods, and it is also an important means of medical image preprocessing. Replacing the original costly exhaustive search approach, swarm intelligent optimization algorithms are recently used to determine the optimal thresholds for medical image, and medical images tend to have higher bit depth. Aiming at the drawbacks of premature convergence of existing optimization algorithms for high-bit depth image segmentation, this paper presents a pyramid particle swarm optimization based on complementary inertia weights (CIWP-PSO), and the Kapur entropy is employed as the optimization objective. Firstly, according to the fitness value, the particle swarm is divided into three-layer structure. To accommodate the larger search range caused by higher bit depth, the particles in the layer with the worst fitness value are employed random opposition learning strategy. Secondly, a pair of complementary inertia weights are introduced to balance the capability of exploitation and exploration. In the part of experiments, this paper used nine high-bit depth benchmark images to test the CIWP-PSO effectiveness. Then, a group of Brain Magnetic Resonance Imaging (MRI) images with 12-bit depth are utilized to validate the advantages of CIWP-PSO compared with other segmentation algorithms based on other optimization algorithms. According to the segmentation experimental results, thresholds optimized by CIWP-PSO could achieve higher Kapur entropy, and the multi-level thresholding segmentation algorithm based on CIWP-PSO outperforms the similar algorithms in high-bit depth image segmentation. Besides, we used image segmentation quality metrics to evaluate the impact of different segmentation algorithms on images, and the experimental results show that the MRI images segmented by the CIWP-PSO has achieved the best fitness value more times than images segmented by other comparison algorithm in terms of Structured Similarity Index and Feature Similarity Index, which explains that the images segmented by CIWP-PSO has higher image quality.

MeSH terms

  • Algorithms*
  • Brain* / diagnostic imaging
  • Humans
  • Image Processing, Computer-Assisted* / methods
  • Magnetic Resonance Imaging* / methods

Grants and funding

The author(s) received no specific funding for this work.