Organizations: Zhejiang Key Laboratory of Accessible Perception and Intelligent Systems, College of Computer Science and Technology, Zhejiang University · Hangzhou Pujian Medical Technology Co., Ltd · Hangzhou Dianzi University · Tsinghua University
Diabetic Retinopathy (DR) progresses as a continuous and irreversible deterioration of the retina, following a well-defined clinical trajectory from mild to severe stages. However, most existing ordinal regression approaches model DR severity as a set of static, symmetric ranks, capturing relative order while ignoring the inherent unidirectional nature of disease progression. As a result, the learned feature representations may violate biological plausibility, allowing implausible proximity between non-consecutive stages or even reverse transitions. To bridge this gap, we propose Directed Ordinal Diffusion Regularization (D-ODR), which explicitly models the feature space as a directed flow by constructing a progression-constrained directed graph that strictly enforces forward disease evolution. By performing multi-scale diffusion on this directed structure, D-ODR imposes penalties on score inversions along valid progression paths, thereby effectively preventing the model from learning biologically inconsistent reverse transitions. This mechanism aligns the feature representation with the natural trajectory of DR worsening. Extensive experiments demonstrate that D-ODR yields superior grading performance compared to state-of-the-art ordinal regression and DR-specific grading methods, offering a more clinically reliable assessment of disease severity. Our code is available on https://github.com/HovChen/D-ODR.
Figures & tables
Figure 1: Overview of the proposed D-ODR framework. The model constructs a directed graph in the feature space using a directional mask and applies multi-scale diffusion to capture extended progression chains, therefore penalizing score inversion along valid forward trajectories. The regularization module is discarded during inference, ensuring no extra computational cost.
Model
APTOS
Messidor
DeepDR
DDR
QWK
F1
QWK
F1
QWK
F1
QWK
F1
ViT-B/16 [ 21 ]
87.6 ± 1.1
65.8 ± 1.7
57.4 ± 2.3
51.2 ± 3.8
74.4 ± 0.5
56.6 ± 0.4
75.5 ± 1.0
54.2 ± 0.3
RETFound [ 25 ]
92.1 ± 0.2
72.3 ± 0.5
81.2 ± 0.8
70.5 ± 0.4
81.6 ± 1.2
62.8 ± 0.5
85.5 ± 0.5
60.2 ± 1.7
CORN † [ 19 ]
88.9 ± 0.2
68.8 ± 0.7
60.9 ± 2.3
49.2 ± 5.0
74.6 ± 2.4
56.8 ± 0.6
73.5 ± 1.8
53.8 ± 2.4
POEs † [ 12 ]
89.6 ± 0.7
67.4 ± 0.5
62.7 ± 1.1
48.1 ± 4.3
78.8 ± 1.4
56.4 ± 1.3
74.8 ± 0.8
52.4 ± 1.0
GOL † [ 10 ]
88.3 ± 1.1
64.6 ± 1.9
52.6 ± 2.9
45.6 ± 1.8
74.9 ± 1.2
51.3 ± 0.5
75.3 ± 1.5
50.7 ± 0.3
Table 1: Performance comparison on four datasets. † and ‡ indicate ViT-B/16 and RETFound backbones, respectively.
Model
APTOS
Messidor
DeepDR
DDR
QWK
F1
QWK
F1
QWK
F1
QWK
F1
Backbone: ViT-B/16
Baseline
87.7 ± 1.1
65.7 ± 1.7
57.4 ± 2.3
51.2 ± 3.8
74.4 ± 0.5
56.6 ± 0.4
75.5 ± 1.0
54.2 ± 0.3
w/o Dir.
88.7 ± 0.7
62.5 ± 1.2
64.7 ± 4.4
49.8 ± 0.4
79.0 ± 1.2
54.7 ± 0.3
76.2 ± 1.3
49.0 ± 0.3
w/o FC
87.7 ± 2.2
58.5 ± 13.2
62.4 ± 5.2
45.9 ± 11.9
77.4 ± 3.7
51.3 ± 13.2
75.0 ± 1.0
49.2 ± 4.0
w/o MSD
88.8 ± 1.7
65.9 ± 2.3
63.0 ± 2.4
46.0 ± 8.0
80.1 ± 1.7
56.9 ± 1.7
76.3 ± 0.7
52.5 ± 1.2
Table 2: Ablation study of different regularization designs with two backbones. Dir.: direction constraint; FC: forward constraint; MSD: multi-scale diffusion.
Figure 2: Sensitivity analysis of hyperparameters λ , bs , and k on four datasets. The translucent shaded regions indicate ± one standard deviation.
Figure 3: Qualitative evaluation of feature representations and score predictions. (a) 2D UMAP projections; (b) Predicted severity score distributions.
Diabetic retinopathy (DR) is one of the main causes of impaired vision. A good and reliable automated grading system can make the screening process safer and more accurate. Because DR stages progress gradually, the task of grading disease severity naturally follows an ordinal structure in which neighboring classes share similar visual characteristics. In this study, ORViT-DR, a hybrid deep learning framework, is designed to improve DR grading from low-resolution retinal images. The proposed approach combines convolutional feature extraction with transformer-based global context modeling through a pre-trained ViT-Hybrid backbone, which integrates BiT-ResNetv2 with a Vision Transformer architecture. The approach is tested on the RetinaMNIST subset of the MedMNISTv2 dataset, which contains 28x28 retinal fundus images annotated with five levels of disease severity. To promote stable training and better feature learning, the training strategy applies progressive layer unfreezing, layer-wise learning rate decay, exponential moving average (EMA) parameter updates, and ensemble-based prediction during inference. Experimental results on the official RetinaMNIST test set show that the proposed method achieves 57.00% classification accuracy, along with a quadratic weighted kappa score of 0.5963 and a macro-F1 score of 0.4293. These results suggest that hybrid CNN-Transformer architectures can provide effective representations for ordinal retinal image analysis.
Soumit Kumar Kundu, Nabil Ashab, Bidhan Biswas +4
Department of Computer Science and Engineering, Dhaka International University, Dhaka, Bangladesh · KUET, Khulna, Bangladesh · Bangladesh Bank, Dhaka, Bangladesh
Diabetic retinopathy (DR) is a leading cause of vision impairment worldwide, and automated grading systems play a crucial role in large-scale screening programs. However, deep learning models often exhibit degraded performance when deployed across datasets acquired under different imaging conditions. This study presents a robust dual-resolution deep learning framework for DR grading that integrates attention-based feature fusion with ordinal regression to improve cross-dataset generalization. The proposed method employs two parallel EfficientNet backbones operating at different spatial resolutions to capture complementary retinal features. A learnable attention mechanism adaptively fuses multi-resolution representations, while an ordinal regression formulation based on the cumulative link model (CORAL) explicitly accounts for the ordered nature of DR severity levels. To mitigate domain discrepancies between datasets, a preprocessing strategy combining circular cropping, contrast enhancement, and histogram matching is applied. The model was trained on the APTOS 2019 dataset and evaluated on both an internal validation split and an external Messidor-2 test set. Experimental results demonstrate strong grading performance, achieving a quadratic weighted kappa (QWK) of 0.88 on the APTOS validation set and 0.68 on the unseen Messidor-2 dataset, indicating improved robustness for cross-dataset DR grading applications.
Afshan Hashmi
Tuwaiq Academy, Tuwaiq Research and Development Centre, Riyadh, Saudi Arabia
Diabetic retinopathy (DR) remains a leading cause of vision loss among working-age adults worldwide, making reliable severity grading clinically important. Despite strong performance, most deep models formulate DR grading as image-level classification and do not explicitly model clinically grounded evidence, such as lesion types and spatial relations. In this paper, we propose PARAGraph, a Pathology-Anatomy-Aware Hierarchical Graph framework for DR grading. PARAGraph represents each image as a three-level hierarchical graph with lesion-level nodes, intermediate category and region nodes, and global anatomical and semantic nodes. To incorporate medical priors into nodes, we construct an optic disc-fovea-anchored coordinate frame that provides a scale- and rotation-normalized retinal reference system. Within this frame, lesion nodes are encoded with category, normalized area, and anatomical coordinates. To mitigate noisy lesion segmentation, PARAGraph uses a dual-fusion strategy that introduces global visual context into a graph semantic node and a decision-level prediction branch, improving robustness when lesion evidence is unreliable. Extensive experiments on Messidor-2, APTOS, and DDR show that PARAGraph achieves consistent DR grading performance over state-of-the-art methods. Interpretability and robustness analyses further demonstrate that its predictions are clinically grounded, closely associated with lesion evidence and robust to lesion segmentation noise.
Ziyang Zhang, Yuankai Huo, Yalin Zheng +1
Vanderbilt University, USA · University of Liverpool, UK