Joint-Seg: Treat Foveal Avascular Zone and Retinal Vessel Segmentation in OCTA Images as a Joint Task

Kai Hu1*      Shuai Jiang1       Yuan Zhang1†      Xuanya Li2      Xieping Gao3†     
1XTU                2Baidu Inc.                3HNNU
* M.Eng. Supervisor        † Corresponding Author
IEEE TIM 2022
Comparison of our proposed method and the existing FAZ and RV segmentation methods in OCTA images. (a) and (b) Single tasks. (c) Parallel multitask. (d) Our proposed joint task.

Joint-Seg innovatively regards the segmentation of two biomarkers [fovea avascular zone (FAZ) and retinal vessels (RV)] as a joint task , characterized by:

  • Unified encoding and modeling of OCTA images
  • FAZ-specific feature alignment decoder block (FADB) for restoring boundary information
  • RV-specific multiscale soft fusion module (MSFM) for capturing multiscale vessels
  • State-of-the-art performance with 90.51%/98.43% and 89.72%/91.13% dice scores for FAZ and RV, respectively

Abstract

Optical coherence tomography angiography (OCTA) has been widely used in ophthalmology in recent years due to its noninvasive and high resolution. In OCTA images, two biomarkers are extremely important for clinical diagnosis, i.e., foveal avascular zone (FAZ) and retinal vessel (RV), and RV has an implicit constraint on FAZ in position. In previous studies, the segmentation of the two biomarkers is naturally separated, which undoubtedly leads to the omission of such constraints between them. In this article, we propose a joint segmentation framework (Joint-Seg), a single-encoder and dual-decoder architecture, through which simultaneous FAZ and RV extractions from en-face OCTA images can be achieved. Specifically, the OCTA image is encoded through joint encoding and provides FAZ- or RV-related information to separate decoding branches through a feature adaptive filter (FAF). In the FAZ segmentation branch, we propose a feature alignment decoder block (FADB) to recover image details, especially boundaries. While in the RV segmentation branch, a multiscale soft fusion module (MSFM) is designed to adapt to different vessel thicknesses. Finally, we evaluate the proposed Joint-Seg on the OCTA-500 dataset, and the experimental results show that our Joint-Seg outperforms the state-of-the-art methods on both FAZ and RV segmentations and has fewer floating point operations (FLOPs) and parameters. The generalization experiments on four other datasets, i.e., OCTAGON, sFAZ, OCTA-25K, and ROSE, also demonstrate the portability and scalability of the proposed Joint-Seg framework. In addition, the noise analysis further shows good robustness of the proposed method against noise.

Framework

pipeline

Results

FAZ Segmentation

pipeline

RV Segmentation

pipeline

Visualization

pipeline
Ablation study of the proposed FADB and MSFM on the OCTA-500 dataset in joint segmentation task (top: OCTA_6M and bottom: OCTA_3M). (a) Raw OCTA images. (b) Ground truths. Segmentation results obtained by (c) Baseline, (d) Baseline + FAF + FADB, (e) Baseline + FAF + MSFM, and (f) Baseline + FAF + FADB + MSFM

Noise Robustness

pipeline
Variation trends in Dice scores of (a) FAZ segmentation and (b) RV segmentation for each dataset (subset) at different strip noise levels

Citation

@article{hu2022joint,
  title={Joint-seg: Treat foveal avascular zone and retinal vessel segmentation in octa images as a joint task},
  author={Hu, Kai and Jiang, Shuai and Zhang, Yuan and Li, Xuanya and Gao, Xieping},
  journal={IEEE Transactions on Instrumentation and Measurement},
  volume={71},
  pages={1--13},
  year={2022},
  publisher={IEEE}
}