Ensemble of deep learning models with surrogate-based optimization for medical image segmentation.

Dang, Truong; Luong, Anh Vu; Liew, Alan Wee Chung; McCall, John; Nguyen, Tien Thanh

Abstract

Deep Neural Networks (DNNs) have created a breakthrough in medical image analysis in recent years. Because clinical applications of automated medical analysis are required to be reliable, robust and accurate, it is necessary to devise effective DNNs based models for medical applications. In this paper, we propose an ensemble framework of DNNs for the problem of medical image segmentation with a note that combining multiple models can obtain better results compared to each constituent one. We introduce an effective combining strategy for individual segmentation models based on swarm intelligence, which is a family of optimization algorithms inspired by biological processes. The problem of expensive computational time of the optimizer during the objective function evaluation is relieved by using a surrogate-based method. We train a surrogate on the objective function information of some populations and then use it to predict the objective values of each candidate in the subsequent populations. Experiments run on a number of public datasets indicate that our framework achieves competitive results within reasonable computation time.

Citation

DANG, T., LUONG, A.V., LIEW, A.W.C., MCCALL, J. and NGUYEN, T.T. 2022. Ensemble of deep learning models with surrogate-based optimization for medical image segmentation. In 2022 IEEE (Institute of Electrical and Electronics Engineers) Congress on evolutionary computation (CEC 2022), co-located with 2022 IEEE International joint conferences on neural networks (IJCNN 2022), 2022 IEEE International conference on fuzzy systems (FUZZ-IEEE 2022), 18-23 July 2022, Padua, Italy. Piscataway: IEEE (online), article #1030. Available from: https://doi.org/10.1109/CEC55065.2022.9870389

• Conference Name: 2022 IEEE (Institute of Electrical and Electronics Engineers) Congress on evolutionary computation (CEC 2022), co-located with 2022 IEEE International joint conferences on neural networks (IJCNN 2022), 2022 IEEE International conference on fuzzy systems (
• Conference Location: Padua, Italy
• Start Date: Jul 18, 2022
• End Date: Jul 23, 2022
• Acceptance Date: Apr 26, 2022
• Online Publication Date: Jul 23, 2022
• Publication Date: Sep 6, 2022
• Deposit Date: Sep 9, 2022
• Publicly Available Date: Sep 9, 2022
• Publisher: IEEE Institute of Electrical and Electronics Engineers
• ISBN: 9781665467087
• DOI: https://doi.org/10.1109/CEC55065.2022.9870389
• Keywords: Image segmentation; Deep learning, Ensemble learning; Particle swarm optimization; Surrogate models; Surrogate-assisted evolutionary algorithms
• Public URL: https://rgu-repository.worktribe.com/output/1745054