Augmentation Strategy for DINO in Glaucoma Classification
Organizations: Department of Electrical Engineering, Indian Institute of Science, Bengaluru, India
Abstract
Self-supervised learning (SSL) offers a promising approach to medical image representation learning, yet augmentation strategies of SSL frameworks validated on natural images do not reliably transfer to specialized medical imaging domains. We analyze DINO, a knowledge-distillation based SSL framework, and empirically demonstrate that its default augmentation strategy produces unreliable representations for retinal fundus images, degrading downstream glaucoma classification performance. Guided by the principle that effective cropped views should share minimal redundant information while retaining clinically relevant signal, we redesign DINO's augmentation strategy. Our approach biases the learned representations toward diagnostically critical features rather than low-level image statistics. We evaluate our method for glaucoma classification by training a linear classifier on the frozen representations, achieving strong generalization across seven benchmark datasets. URL for code and weights: https://github.com/ashish-256/task_dino
Figures & tables
| Dataset | Vanilla DINO | RETFound | RetiZero | FLAIR | Ours |
| (a) REFUGE-trained classifier ( 1.2k images) | |||||
| REFUGE | 0.802 - | 0.875 - | 0.911 - | 0.761 - | 0.797 - |
| ACRIMA | 0.824 - | 0.824 - | 0.778 - | 0.717 - | 0.846 - |
| LAG | 0.852 - | 0.815 - | 0.816 - | 0.874 - | 0.890 - |
| ORIGA | 0.692 - | 0.694 - | 0.776 - | 0.789 - | 0.757 - |
| FIVES | 0.835 - | 0.804 - | 0.847 - | 0.852 - | 0.866 - |
| Trained on | No Asymmetry | Common Views | Reverse Views | With Vessel | Our Approach |
| REFUGE | |||||
| AIROGS |
Self-Supervised Learning
Self-supervised learning uses structure within unlabeled data to train representations and models for downstream tasks.
See all 259 papers