Files and links (1)
Published VersionCC BY V4.0, Open Access
Journal article
Enhancing OCT patch-based segmentation with improved GAN data augmentation and semi-supervised learning
Neural Computing & Applications, Vol.36, pp.18087-18105
2024
Abstract
For optimum performance, deep learning methods, such as those applied for retinal and choroidal layer segmentation in optical coherence tomography (OCT) images, require sufficiently large and diverse labelled datasets for training. However, the acquisition and labelling of such data can be difficult or infeasible due to privacy reasons (particularly in the medical domain), accessing patient images such as those with specific pathologies, and the cost and time investment to annotate large volumes of data by clinical experts. Data augmentation is one solution to address this issue, either using simple variations and transformations of the images (e.g. flips, brightness) or using synthetic data from sophisticated generative methods such as generative adversarial networks (GANs). Semi-supervised learning (SSL) is another technique which aims to utilise unlabelled data to enhance the performance of deep learning methods and is beneficial where significant amounts of data may be available but are not labelled. In this study, we aim to enhance patch-based OCT retinal and choroidal layer segmentation with both GAN-based data augmentation and SSL. In particular, we employ a conditional StyleGAN2 to generate synthetic patches for data augmentation and a similar unconditional GAN for pre-training the patch classifier to perform SSL. In doing so, we propose a new patch classifier architecture based on the discriminator architecture to improve performance, in addition to the SSL benefit. Compared to previous methods, the proposed data augmentation approach provides an improved data augmentation performance for patch classification with its effectiveness widespread, particularly in the case of low data, across three different OCT datasets encompassing a range of scanning parameters, noise levels, pathology and participant variability. The method provides some subsequent improvements in boundary delineation which is of high importance from a clinical perspective. Additionally, the proposed SSL approach boosts classification performance and boundary delineation performance in some cases which provides further usefulness in the case of low data. The proposed methods can be utilised to enhance OCT segmentation methods, which may be of considerable benefit for both clinicians and researchers.
Details
- Title
- Enhancing OCT patch-based segmentation with improved GAN data augmentation and semi-supervised learning
- Authors
- Jason Kugelman (Corresponding Author) - Queensland University of TechnologyDavid Alonso-Caneiro - Queensland University of TechnologyScott A. Read - Queensland University of TechnologyStephen J. Vincent - Queensland University of TechnologyMichael J. Collins - Queensland University of Technology
- Publication details
- Neural Computing & Applications, Vol.36, pp.18087-18105
- Publisher
- Springer UK
- Date published
- 2024
- DOI
- 10.1007/s00521-024-10044-1
- ISSN
- 1433-3058; 0941-0643
- Copyright note
- This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/.
- Data Availability
- The dataset used within this study is currently not publicly available.
- Organisation Unit
- School of Science, Technology and Engineering
- Language
- English
- Record Identifier
- 991048886502621
- Output Type
- Journal article
Metrics
21 Record Views