GA-based U-Net Architecture Optimization Applied to Retina Blood Vessel Segmentation

Vipul Popat, Mahsa Mahdinejad, Oscar S.Dalmau Cedeño, Enrique Naredo, Conor Ryan

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Blood vessel extraction in digital retinal images is an important step in medical image analysis for abnormality detection and also obtaining good retinopathy diabetic diagnosis; this is often referred to as the Retinal Blood Vessel Segmentation task and current state-of-the-art approaches all use some form of neural networks. Designing neural network architecture and selecting appropriate hyper-parameters for a specific task is challenging. In recent works, increasingly more complex models are starting to appear, but in this work, we present a simple and small model with a very low number of parameters with good performance compared with the state of the art algorithms. In particular, we choose a standard Genetic Algorithm (GA) for selecting the parameters of the model and we use an expert-designed U-net based model, which has become a very popular tool in image segmentation problems. Experimental results show that GA is able to find a much shorter architecture and acceptable accuracy compared to the U-net manually designed. This finding puts on the right track to be able in the future to implement these models in portable applications.

Original languageEnglish
Title of host publicationProceedings of the 12th International Joint Conference on Computational Intelligence, IJCCI 2020
EditorsJuan Julian Merelo, Jonathan Garibaldi, Christian Wagner, Thomas Bäck, Kurosh Madani, Kevin Warwick
PublisherScience and Technology Publications, Lda
Pages192-199
Number of pages8
ISBN (Electronic)9789897584756
ISBN (Print)9789897584756
DOIs
Publication statusPublished - 2020
Event12th International Joint Conference on Computational Intelligence, IJCCI 2020 - Budapest, Hungary
Duration: 2 Nov 20204 Nov 2020

Publication series

NameInternational Joint Conference on Computational Intelligence
Volume1
ISSN (Electronic)2184-3236

Conference

Conference12th International Joint Conference on Computational Intelligence, IJCCI 2020
Country/TerritoryHungary
CityBudapest
Period2/11/204/11/20

Keywords

  • Deep Learning
  • Image Segmentation
  • U-Net

Fingerprint

Dive into the research topics of 'GA-based U-Net Architecture Optimization Applied to Retina Blood Vessel Segmentation'. Together they form a unique fingerprint.

Cite this