Skip to main content

Research Repository

Advanced Search

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

Authors

Anh Vu Luong

Alan Wee Chung Liew



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

Files





You might also like



Downloadable Citations