TY - JOUR
T1 - Hybrid genetic and variational expectation-maximization algorithm for Gaussian-mixture-model-based brain MR image segmentation
AU - Tian, Guangjian
AU - Xia, Yong
AU - Zhang, Yanning
AU - Feng, Dagan
PY - 2011/5
Y1 - 2011/5
N2 - The expectation-maximization (EM) algorithm has been widely applied to the estimation of Gaussian mixture model (GMM) in brain MR image segmentation. However, the EM algorithm is deterministic and intrinsically prone to overfitting the training data and being trapped in local optima. In this paper, we propose a hybrid genetic and variational EM (GA-VEM) algorithm for brain MR image segmentation. In this approach, the VEM algorithm is performed to estimate the GMM, and the GA is employed to initialize the hyperparameters of the conjugate prior distributions of GMM parameters involved in the VEM algorithm. Since GA has the potential to achieve global optimization and VEM can steadily avoid overfitting, the hybrid GA-VEM algorithm is capable of overcoming the drawbacks of traditional EM-based methods. We compared our approach to the EM-based, VEM-based, and GA-EM based segmentation algorithms, and the segmentation routines used in the statistical parametric mapping package and FMRIB Software Library in 20 low-resolution and 17 high-resolution brain MR studies. Our results show that the proposed approach can improve substantially the performance of brain MR image segmentation.
AB - The expectation-maximization (EM) algorithm has been widely applied to the estimation of Gaussian mixture model (GMM) in brain MR image segmentation. However, the EM algorithm is deterministic and intrinsically prone to overfitting the training data and being trapped in local optima. In this paper, we propose a hybrid genetic and variational EM (GA-VEM) algorithm for brain MR image segmentation. In this approach, the VEM algorithm is performed to estimate the GMM, and the GA is employed to initialize the hyperparameters of the conjugate prior distributions of GMM parameters involved in the VEM algorithm. Since GA has the potential to achieve global optimization and VEM can steadily avoid overfitting, the hybrid GA-VEM algorithm is capable of overcoming the drawbacks of traditional EM-based methods. We compared our approach to the EM-based, VEM-based, and GA-EM based segmentation algorithms, and the segmentation routines used in the statistical parametric mapping package and FMRIB Software Library in 20 low-resolution and 17 high-resolution brain MR studies. Our results show that the proposed approach can improve substantially the performance of brain MR image segmentation.
KW - Gaussian mixture model (GMM)
KW - MRI
KW - genetic algorithm (GA)
KW - image segmentation
KW - variational Bayes inference
UR - http://www.scopus.com/inward/record.url?scp=79955655739&partnerID=8YFLogxK
U2 - 10.1109/TITB.2011.2106135
DO - 10.1109/TITB.2011.2106135
M3 - 文章
C2 - 21233052
AN - SCOPUS:79955655739
SN - 1089-7771
VL - 15
SP - 373
EP - 380
JO - IEEE Transactions on Information Technology in Biomedicine
JF - IEEE Transactions on Information Technology in Biomedicine
IS - 3
M1 - 5688239
ER -