Comparative Analysis of Retinal Fluid Segmentation in OCT Images : Evaluating CNN-Based Models vs. Transformer-Based Models

  • Hala Al-Jarajrah

Research output: ThesisMaster's / Diploma thesis

Abstract

Semantic segmentation of retinal fluids in optical coherence tomography (OCT) images plays a crucial role in the diagnosis and treatment of retinal diseases. This study presents a comprehensive comparative analysis of transformer-based models and convolutional neural network (CNN)-based models for retinal fluid segmentation, focusing the comparison between the CNN-based models, DeepLab and UNet, and the transformer-based model, Swin UNetR. The evaluation metrics employed include F1 score and Absolute Volume Difference (AVD). The RETOUCH challenge dataset from three different OCT vendors, Spectralis, Cirrus, and Topcon, was used for evaluating the models. The results indicate that Swin UNetR outperformed the other models, achieving the best segmentation results for both F1 score and AVD. Visual analysis further supported these findings, revealing Swin UNetR's superior capability in capturing intricate fluid structures. Statistical analysis using paired t-tests and ANOVA showed significance between Swin UNetR and UNet for the Serous Retinal Fluid (SRF) class. The combination of the Swin Transformer and U-Net architecture in Swin UNetR proved to be a potent approach, capturing both global contextual information and precise local details. This research provides valuable insights into the strengths and weaknesses of different segmentation methods, guiding future advancements in the field of retinal fluid segmentation and contributing to improved clinical diagnosis and treatment of retinal diseases.
Original languageEnglish
QualificationMaster
Awarding Institution
  • Johannes Kepler University Linz
Supervisors/Reviewers
  • Klambauer, Günter, Supervisor
  • Bogunović, Hrvoje, Co-supervisor, External person
Publication statusPublished - Aug 2023

Fields of science

  • 102019 Machine learning
  • 102018 Artificial neural networks
  • 102032 Computational intelligence
  • 102020 Medical informatics
  • 305901 Computer-aided diagnosis and therapy
  • 101016 Optimisation
  • 101028 Mathematical modelling
  • 202037 Signal processing
  • 101019 Stochastics
  • 102003 Image processing
  • 103029 Statistical physics
  • 101018 Statistics
  • 101017 Game theory
  • 102001 Artificial intelligence
  • 202017 Embedded systems
  • 101015 Operations research
  • 101014 Numerical mathematics
  • 101029 Mathematical statistics
  • 101026 Time series analysis
  • 101024 Probability theory
  • 102004 Bioinformatics
  • 102013 Human-computer interaction
  • 101027 Dynamical systems
  • 305907 Medical statistics
  • 101004 Biomathematics
  • 305905 Medical informatics
  • 101031 Approximation theory
  • 102033 Data mining
  • 102 Computer Sciences
  • 106007 Biostatistics
  • 106005 Bioinformatics
  • 202036 Sensor systems
  • 202035 Robotics

JKU Focus areas

  • Digital Transformation

Cite this