Intelligent optic disc segmentation using improved particle swarm optimization and evolving ensemble models

Zhang, Li and Lim, Chee Peng (2020) Intelligent optic disc segmentation using improved particle swarm optimization and evolving ensemble models. Applied Soft Computing, 92. p. 106328. ISSN 1568-4946

[img] Text
journal_optimized.pdf - Accepted Version
Restricted to Repository staff only until 20 April 2021.
Available under License Creative Commons Attribution Non-commercial No Derivatives 4.0.

Download (1MB) | Request a copy
Official URL: https://doi.org/10.1016/j.asoc.2020.106328

Abstract

In this research, we propose Particle Swarm Optimization (PSO)-enhanced ensemble deep neural networks for optic disc (OD) segmentation using retinal images. An improved PSO algorithm with six search mechanisms to diversify the search process is introduced. It consists of an accelerated super-ellipse action, a refined super-ellipse operation, a modified PSO operation, a random leader-based search operation, an average leader-based search operation and a spherical random walk mechanism for swarm leader enhancement. Owing to the superior segmentation capabilities of Mask R-CNN, transfer learning with a PSO-based hyper-parameter identification method is employed to generate the fine-tuned segmenters for OD segmentation. Specifically, we optimize the learning parameters, which include the learning rate and momentum of the transfer learning process, using the proposed PSO algorithm. To overcome the bias of single networks, an ensemble segmentation model is constructed. It incorporates the results of distinctive base segmenters using a pixel-level majority voting mechanism to generate the final segmentation outcome. The proposed ensemble network is evaluated using the Messidor and Drions data sets and is found to significantly outperform other deep ensemble networks and hybrid ensemble clustering models that are incorporated with both the original and state-of-the-art PSO variants. Additionally, the proposed method statistically outperforms existing studies on OD segmentation and other search methods for solving diverse unimodal and multimodal benchmark optimization functions and the detection of Diabetic Macular Edema.

Item Type: Article
Uncontrolled Keywords: Image segmentation, Particle swarm optimization, Evolutionary algorithm, Convolutional neural network and ensemble segmentation model
Subjects: G400 Computer Science
G500 Information Systems
G600 Software Engineering
Department: Faculties > Engineering and Environment > Computer and Information Sciences
Depositing User: Elena Carlaw
Date Deposited: 27 Apr 2020 11:08
Last Modified: 27 Apr 2020 11:15
URI: http://nrl.northumbria.ac.uk/id/eprint/42920

Actions (login required)

View Item View Item

Downloads

Downloads per month over past year

View more statistics