Preserving Ordinality in Diabetic Retinopathy Grading through a Distribution-Based Loss Function

Lena Stelter, Valentina Corbetta, Soufyan Lakbir, Regina Beets-Tan, Ricardo P. M. Cruz, Jaime S Cardoso, Wilson Silva

Keywords: Ordinal Learning, Diabetic Retinopathy, Medical Image Analysis

Abstract:

Diabetic Retinopathy (DR) is a neurovascular complication of diabetes and the leading cause of blindness in adults in developed countries. Because DR progresses through ordered severity levels, its grading is naturally an ordinal classification problem. Yet, most deep learning methods treat it as a categorical task, disregarding the inherent class order and worsening performance under class imbalance.

In this work, we introduce a novel ordinal loss function that emphasizes the predictive tendencies of the whole model output rather than the class output probabilities individually. This design promotes unimodal predictions aligned with the underlying severity scale and is particularly robust to class imbalance. To place our method in context, we also evaluate a range of existing ordinal approaches on five publicly available DR datasets. with cross-entropy serving as a nominal baseline.

Extensive experiments demonstrate that our proposed loss function consistently preserves the ordinal structure of DR grades, even under severe imbalance, outperforming both nominal and alternative ordinal formulations.

Code will be released upon acceptance.

Serve As Reviewer: ~Valentina_Corbetta1

Submission Number: 40