LUMOS: Universal Semi-Supervised OCT Retinal Layer Segmentation with Hierarchical Reliable Mutual Learning
Authors: Yizhou Fang, Jian Zhong, Li Lin, Xiaoying Tang
Organizations: Department of Electronic and Electrical Engineering, Southern University of Science and Technology, Shenzhen, China · Shenzhen Loop Area Institute, Shenzhen, China · Jiaxing Research Institute, Southern University of Science and Technology, Jiaxing, China
Optical Coherence Tomography (OCT) layer segmentation faces challenges due to annotation scarcity and heterogeneous label granularities across datasets. While semi-supervised learning helps alleviate label scarcity, existing methods typically assume a fixed granularity, failing to fully exploit cross-granularity supervision. This paper presents LUMOS, a semi-supervised universal OCT retinal layer segmentation framework based on a Dual-Decoder Network with a Hierarchical Prompting Strategy (DDN-HPS) and Reliable Progressive Multi-granularity Learning (RPML). DDN-HPS combines a dual-branch architecture with a multi-granularity prompting strategy to effectively suppress pseudo-label noise propagation. Meanwhile, RPML introduces region-level reliability weighing and a progressive training approach that guides the model from easier to more difficult tasks, ensuring the reliable selection of cross-granularity consistency targets, thereby achieving stable cross-granularity alignment. Experiments on six OCT datasets demonstrate that LUMOS largely outperforms existing methods and exhibits exceptional cross-domain and cross-granularity generalization capability.
Figures & tables
Figure 1: Overview of the proposed LUMOS framework and its key components.
Figure 2: Visualization comparison results on the OIMHS, Goals, HEG and HC-MS datasets.
Setting
Internal Datasets
External Datasets
Avg.
DSC ↑
HD95 ↓
DSC ↑
HD95 ↓
DSC ↑
HD95 ↓
A1
85.13 ± 2.08
3.97 ± 9.77
82.92 ± 8.25
22.34 ± 23.04
84.03 ± 5.17
13.16 ± 16.41
A2
85.15 ± 7.38
3.98 ± 7.86
83.51 ± 7.81
16.58 ± 22.38
84.33 ± 7.60
10.28 ± 15.12
Table 3: Ablation results of the dual-branch design.
Components
Internal Datasets
External Datasets
Avg.
RLRW
MPF
DSC ↑
HD95 ↓
DSC ↑
HD95 ↓
DSC ↑
HD95 ↓
−
−
85.15 ± 7.38
3.98 ± 7.86
83.51 ± 7.81
16.58 ± 22.38
84.33 ± 7.60
10.28 ± 15.12
✓
−
85.94 ± 2.31
1.98 ± 2.02
85.73 ± 9.14
11.52 ± 19.38
85.84 ± 5.73
9.28 ± 12.20
✓
✓
86.28 ± 7.83
1.73 ± 1.94
85.94 ± 9.14
2.99 ± 6.15
86.11 ± 8.49
2.36 ± 4.05
Table 4: Ablation results of RPML’s key components.
Volumetric segmentation of optical coherence tomography (OCT) images is essential for diagnosing ocular diseases but requires labor-intensive voxel-wise annotations. While semi-supervised learning (SSL) can reduce annotation costs, most existing methods process data slice by slice and fail to exploit the inherent 3D spatial context. We propose PC-Seg, a progressive cross-view consistency framework that learns high-accuracy 3D segmentation models from sparse 2D annotations. Unlike conventional multi-view approaches, PC-Seg uses a single 2D model to learn cross-view consistency from standard B-scans and orthogonal slices, thereby generating reliable volumetric pseudo-labels. These pseudo-labels are then distilled into a 3D model, followed by a co-training stage in which the 2D and 3D models mutually refine each other through ensemble pseudo-labeling. Experiments on the MSHC and Duke DME datasets demonstrate that PC-Seg achieves accuracy comparable to fully supervised learning while using labels for only about 0.7% of the training data, outperforming state-of-the-art semi-supervised and retinal layer segmentation methods. Our code is publicly available at https://github.com/gsisaoki/pc-seg-official.
Tsubasa Konno, Takahiro Ninomiya, Yukun Zhou +6
Graduate School of Information Sciences, Tohoku University, Japan · Department of Ophthalmology, Graduate School of Medicine, Tohoku University · Institute of Ophthalmology, University College London, UK +3
For accurate glaucoma diagnosis and monitoring, reliable retinal layer segmentation in OCT images is essential. However, existing 2D segmentation methods often suffer from slice-to-slice inconsistencies due to the lack of contextual information across adjacent B-scans. 3D segmentation methods are better for capturing slice-to-slice context, but they require expensive computational resources. To address these limitations, we propose a 2.5D segmentation framework that incorporates a novel cross-slice feature fusion (CFF) module into a U-Net-like architecture. The CFF module fuses inter-slice features to effectively capture contextual information, enabling consistent boundary detection across slices and improved robustness in noisy regions. The framework was validated on both a clinical dataset and the publicly available DUKE DME dataset. Compared to other segmentation methods without the CFF module, the proposed method achieved an 8.56% reduction in mean absolute distance and a 13.92% reduction in root mean square error, demonstrating improved segmentation accuracy and robustness. Overall, the proposed 2.5D framework balances contextual awareness and computational efficiency, enabling anatomically reliable retinal layer delineation for automated glaucoma evaluation and potential clinical applications.
Hyunwoo Kim, Heesuk Kim, Wungrak Choi +1
Department of Mechanical Engineering, Yonsei University, Seoul, 03722, South Korea · Institute of Vision Research, Department of Ophthalmology, Severance Hospital, Yonsei University, Seoul, 03722, South Korea
Reliable automated analysis of Optical Coherence Tomography (OCT) imaging is crucial for diagnosing retinal disorders but faces a critical barrier: the need for expensive, labor-intensive expert annotations. Supervised deep learning models struggle to generalize across diverse pathologies, imaging devices, and patient populations due to their restricted vocabulary of annotated abnormalities. We propose an unsupervised anomaly detection framework that learns the normative distribution of healthy retinal anatomy without lesion annotations, directly addressing annotation efficiency challenges in clinical deployment. Our approach leverages a discrete latent model trained on normal B-scans to capture OCT-specific structural patterns. To enhance clinical robustness, we incorporate retinal layer-aware supervision and structured triplet learning to separate healthy from pathological representations, improving model reliability across varied imaging conditions. During inference, anomalies are detected and localized via reconstruction discrepancies, enabling both image and pixel-level identification without requiring disease-specific labels. On the Kermany dataset (AUROC: 0.799), our method substantially outperforms VAE, VQVAE, VQGAN, and f-AnoGAN baselines. Critically, cross-dataset evaluation on Srinivasan achieves AUROC 0.884 with superior generalization, demonstrating robust domain adaptation. On the external RETOUCH benchmark, unsupervised anomaly segmentation achieves competitive Dice (0.200) and mIoU (0.117) scores, validating reproducibility across institutions.
Tania Haghighi, Sina Gholami, Hamed Tabkhi +1
University of North Carolina at Charlotte Charlotte, NC, USA