Medical Image Clustering Based on Improved Particle Swarm Optimization and Expectation Maximization Algorithm

We proposed a hybrid clustering algorithm based on the improved particle swarm optimization algorithm and EM clustering algorithm to overcome the shortcomings of EM algorithm, which is sensitive to initial value and easy to sink into local minimum. First, get the optimal clustering number of any dataset to obtain the initial parameter of mixed model with the improved PSO algorithm, whose inertia weight increased and decreased along the fold line automatically. Then build the mixed density model of image data by multiple iterations of the EM algorithm. Finally divide all the pixel value of the image into corresponding branch of hybrid model with the Bayesian criterion to get the classification of image data. The proposed algorithm can increase the diversity of EM clustering algorithm initialization and promote optimization search in the global scope. Experimental results of simulation prove its accuracy and validity.

[1]  Wang Zong-hu.  Improved Possibilistic C-means Clustering Algorithm Based on Particle Swarm Optimization , 2012 .

[2]  Pasi Fränti,et al.  Random swap EM algorithm for Gaussian mixture models , 2012, Pattern Recognit. Lett..

[3]  Feng Zhao,et al.  Optimal-selection-based suppressed fuzzy c-means clustering algorithm with self-tuning non local spatial information for image segmentation , 2014, Expert Syst. Appl..

[4]  Yiguang Liu,et al.  TRUS image segmentation with non-parametric kernel density estimation shape prior , 2013, Biomed. Signal Process. Control..

[5]  Daoliang Li,et al.  An improved KK-means clustering algorithm for fish image segmentation , 2013, Math. Comput. Model..

[6]  Andries Petrus Engelbrecht,et al.  Data clustering using particle swarm optimization , 2003, The 2003 Congress on Evolutionary Computation, 2003. CEC '03..

[7]  Riccardo Poli,et al.  Particle swarm optimization , 1995, Swarm Intelligence.

[8]  Liu Zhe.  Image segmentation based on non-parametric mixture model of Legendre orthogonal polynomial , 2010 .

[9]  Volodymyr Melnykov,et al.  Finite mixture models and model-based clustering , 2010 .

[10]  M. Mohajeri,et al.  K-NichePSO clustering , 2008, 2008 International Conference on Machine Learning and Cybernetics.

[11]  Ben J. A. Kröse,et al.  Efficient Greedy Learning of Gaussian Mixture Models , 2003, Neural Computation.

[12]  Volodymyr Melnykov,et al.  Initializing the EM algorithm in Gaussian mixture models with an unknown number of components , 2012, Comput. Stat. Data Anal..

[13]  Mohammad Mehdi Ebadzadeh,et al.  A novel particle swarm optimization algorithm with adaptive inertia weight , 2011, Appl. Soft Comput..

[14]  Hui Zhang,et al.  A finite mixture model for detail-preserving image segmentation , 2013, Signal Process..

[15]  Miin-Shen Yang,et al.  A robust EM clustering algorithm for Gaussian mixture models , 2012, Pattern Recognit..

[16]  Geoffrey E. Hinton,et al.  SMEM Algorithm for Mixture Models , 1998, Neural Computation.

[17]  Daniel W. A. Buchan,et al.  A large-scale evaluation of computational protein function prediction , 2013, Nature Methods.

[18]  Ajith Abraham,et al.  Swarm Intelligence Algorithms for Data Clustering , 2008, Soft Computing for Knowledge Discovery and Data Mining.

[19]  Adrian E. Raftery,et al.  Model-Based Clustering, Discriminant Analysis, and Density Estimation , 2002 .