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

Loading...
Thumbnail Image
File version

Accepted Manuscript (AM)

Author(s)
Truong, Dang
Anh, Vu Luong
Liew, Alan Wee Chung
McCall, John
Tien, Thanh Nguyen
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
2022
Size
File type(s)
Location

Padua, Italy

License
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.

Journal Title
Conference Title

2022 IEEE Congress on Evolutionary Computation (CEC)

Book Title
Edition
Volume
Issue
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement

© 2022 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.

Item Access Status
Note
Access the data
Related item(s)
Subject

Deep learning

Biomedical imaging

Science & Technology

Technology

Life Sciences & Biomedicine

Computer Science, Artificial Intelligence

Mathematical & Computational Biology

Persistent link to this record
Citation

Truong, D; Anh, VL; Liew, AWC; McCall, J; Tien, TN, Ensemble of deep learning models with surrogate-based optimization for medical image segmentation, 2022 IEEE Congress on Evolutionary Computation (CEC), 2022