Choroidal Layer Analysis in OCT images via Ambiguous Boundary-aware Attention

Abstract

Optical Coherence Tomography (OCT) is a commonly used retina imaging technique, and it is capable of revealing the morphology of the choroid. However, the segmentation and quantitative analysis of the sublayers and vessels in choroid are rarely explored, primarily due to the indistinct boundaries of choroidal sublayers, and imbalanced distribution of vessels observed in OCT imagery. In this paper, we propose a novel two-stage architecture called Choroidal Layer Analysis network (CLA), that may be considered the first attempt in this research community for joint segmentation of choroidal sublayers and choroidal vessels in OCT images. CLA employs the encoder–decoder network with the residual U-shape module as the backbone. In order to empower the ability of the segmentation model to identify the inconspicuous boundaries of choroidal sublayers, we introduce an Ambiguous Boundary Attention block (ABA) into the bottleneck of the encoder–decoder network in the first stage. For more accurate segmentation of large choroidal vessels with ambiguous contours and imbalanced spatial distribution, the second stage introduces an active contour-based loss to refine the contours of choroidal vessels simultaneously with precise identification of each vessel via contextual modeling. To train, test and validate the proposed model, we conducted a choroidal segmentation dataset containing 800 OCT images, with their sublayers and large choroidal vessels manually annotated. Experimental results demonstrate the superiority of the proposed approach compared with other state-of-the-art segmentation networks in large margins. It is worth noting that we also reconstructed the large choroidal vessels in three-dimensional (3D) based on the segmentation results, and multiple 3D morphological parameters were calculated. The statistical analysis of these parameters demonstrates significant differences between the healthy control and high myopia group, and this further confirms the proposed work may facilitate subsequent disease understanding and clinical decision-making.