Parameterising the SA-UNet using a Genetic Algorithm

Mahsa Mahdinejad, Aidan Murphy, Patrick Healy, Conor Ryan

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

Abstract

Deep learning is an excellent way for effectively addressing image processing, and several Neural Networks designs have been explored in this area. The Spatial Attention U-Net architecture, a version of the famous U-Net but which uses DropBlock and an attention block as well as the common U-Net convolutional blocks, is one notable example. Finding the best combination of hyper-parameters is expensive, time consuming and needs expert input. We show the genetic algorithm can be utilized to automatically determine the optimal combination of Spatial Attention U-Net hyper-parameters to train a model to solve a Retinal Blood Vessel Segmentation problem. Our new approach is able to find a model with an accuracy measure of 0.9855, an improvement from our previous experimentation which found a model with accuracy measure of 0.9751. Our new methods exhibit competitive performance with other state-of-the-art Retinal Blood Vessel Segmentation techniques.

Original languageEnglish
Title of host publicationIJCCI 2022 - Proceedings of the 14th International Joint Conference on Computational Intelligence
EditorsThomas Back, Bas van Stein, Christian Wagner, Jonathan Garibaldi, H. K. Lam, Marie Cottrell, Faiyaz Doctor, Joaquim Filipe, Kevin Warwick, Janusz Kacprzyk
PublisherScience and Technology Publications, Lda
Pages97-104
Number of pages8
ISBN (Electronic)9789897586118
Publication statusPublished - 2022
Event14th International Joint Conference on Computational Intelligence, IJCCI 2022 - Valletta, Malta
Duration: 24 Oct 202226 Oct 2022

Publication series

NameICETE International Conference on E-Business and Telecommunication Networks (International Joint Conference on Computational Intelligence)
Volume2022-October
ISSN (Print)2184-2825

Conference

Conference14th International Joint Conference on Computational Intelligence, IJCCI 2022
Country/TerritoryMalta
CityValletta
Period24/10/2226/10/22

Keywords

  • Deep Learning
  • Evolutionary Algorithm
  • Image Segmentation

Fingerprint

Dive into the research topics of 'Parameterising the SA-UNet using a Genetic Algorithm'. Together they form a unique fingerprint.

Cite this