Pathological Retinal Region Segmentation From OCT Images Using Geometric Relation Based Augmentation

Abstract

Medical image segmentation is an important task for computer aided diagnosis. Pixelwise manual annotations of large datasets require high expertise and is time consuming. Conventional data augmentations have limited benefit by not fully representing the underlying distribution of the training set, thus affecting model robustness when tested on images captured from different sources. Prior work leverages synthetic images for data augmentation ignoring the interleaved geometric relationship between different anatomical labels. We propose improvements over previous GAN-based medical image synthesis methods by jointly encoding the intrinsic relationship of geometry and shape. Latent space variable sampling results in diverse generated images from a base image and improves robustness. Given those augmented images generated by our method, we train the segmentation network to enhance the segmentation performance of retinal optical coherence tomography (OCT) images. The proposed method outperforms state-of-the-art segmentation methods on the public RETOUCH dataset having images captured from different acquisition procedures. Ablation studies and visual analysis also demonstrate benefits of integrating geometry and diversity.

References in corpus (12)

• Adam: A Method for Stochastic Optimization
• Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning?
• Generative Adversarial Network in Medical Imaging: A Review
• Data Augmentation Generative Adversarial Networks
• Retinal Vasculature Segmentation Using Local Saliency Maps and Generative Adversarial Networks For Image Super Resolution
• Learning in an Uncertain World: Representing Ambiguity Through Multiple Hypotheses
• Night Time Haze and Glow Removal using Deep Dilated Convolutional Network
• Chest X-rays Classification: A Multi-Label and Fine-Grained Problem
• Progressive Generative Adversarial Networks for Medical Image Super resolution
• GAN Based Medical Image Registration
• Consensus Based Medical Image Segmentation Using Semi-Supervised Learning And Graph Cuts
• sZoom: A Framework for Automatic Zoom into High Resolution Surveillance Videos

Cited by in corpus (3)

• Registration of Histopathogy Images Using Structural Information From Fine Grained Feature Maps
• CT Image Synthesis Using Weakly Supervised Segmentation and Geometric Inter-Label Relations For COVID Image Analysis
• Multimodal Generalized Zero Shot Learning for Gleason Grading using Self-Supervised Learning