Abstract:Layer segmentation is important to quantitative analysis of retinal optical coherence tomography (OCT). Recently, deep learning based methods have been developed to automate this task and yield remarkable performance. However, due to the large spatial gap and potential mismatch between the B-scans of an OCT volume, all of them were based on 2D segmentation of individual B-scans, which may lose the continuity and diagnostic information of the retinal layers in 3D space. Besides, most of these methods required dense annotation of the OCT volumes, which is labor-intensive and expertise-demanding. This work presents a novel framework based on hybrid 2D-3D convolutional neural networks (CNNs) to obtain continuous 3D retinal layer surfaces from OCT volumes, which works well with both full and sparse annotations. The 2D features of individual B-scans are extracted by an encoder consisting of 2D convolutions. These 2D features are then used to produce the alignment displacement vectors and layer segmentation by two 3D decoders coupled via a spatial transformer module. Two losses are proposed to utilize the retinal layers' natural property of being smooth for B-scan alignment and layer segmentation, respectively, and are the key to the semi-supervised learning with sparse annotation. The entire framework is trained end-to-end. To the best of our knowledge, this is the first work that attempts 3D retinal layer segmentation in volumetric OCT images based on CNNs. Experiments on a synthetic dataset and three public clinical datasets show that our framework can effectively align the B-scans for potential motion correction, and achieves superior performance to state-of-the-art 2D deep learning methods in terms of both layer segmentation accuracy and cross-B-scan 3D continuity in both fully and semi-supervised settings, thus offering more clinical values than previous works.

Retinal Automatic Segmentation Method Based on Prior Information and Optimized Boundary Tracking Algorithm

Intra-retinal Layers Segmentation of Macular OCT Images Based on the Graph Optimal Approach

Simultaneous Alignment and Surface Regression Using Hybrid 2D-3D Networks for 3D Coherent Layer Segmentation of Retinal OCT Images with Full and Annotations

Automated Layer Segmentation of Macular OCT Images Via Graph-Based SLIC Superpixels and Manifold Ranking Approach

Intra-Retinal Layer Segmentation of Macular Oct Images Using Edge Superpixels and Manifold Ranking Method

Retinal layer segmentation using gradient feature calculation in OCT

Automated Retinal Layer Segmentation in Optical Coherence Tomography Images with Intraretinal Fluid

Three dimensional segmentation to detect retinal boundary surfaces from OCT volume data

Automated Layer Segmentation of Retinal Optical Coherence Tomography Images Using a Deep Feature Enhanced Structured Random Forests Classifier

Automatic Choroid Layer Segmentation : A Slicing Approach

DeepRetina: Layer Segmentation of Retina in OCT Images Using Deep Learning

3D Automatic Segmentation Method for Retinal Optical Coherence Tomography Volume Data Using Boundary Surface Enhancement

Automated retinal boundary segmentation of optical coherence tomography images using an improved Canny operator

Automated Segmentation of Retinal Layers from Optical Coherent Tomography Images Using Geodesic Distance

Automatic Choroid Layer Segmentation from Optical Coherence Tomography Images Using Deep Learning

Retinal Nerve Fiber Layer Segmentation of Spectral Domain Optical Coherence Tomography Images Based on Random Forest

Automatic segmentation of layers in chorio-retinal complex using Graph-based method for ultra-speed 1.7 MHz wide field swept source FDML optical coherence tomography

Retinal Layer Segmentation in OCT images with Boundary Regression and Feature Polarization

Inter-frame information based choroid segmentation in enhanced optical coherence tomography images

Deep Learning Based Retinal Layer Segmentation in Optical Coherence Tomography Scans of Patients with Inherited Retinal Diseases

A combined machine-learning and graph-based framework for the segmentation of retinal surfaces in SD-OCT volumes