Clinical Graph-Mediated Distillation for Unpaired MRI-to-CFI Hypertension Prediction
Organizations: Department of Computer Science and Engineering, Sungkyunkwan University, Suwon, South Korea · Convergence Research Institute, Sungkyunkwan University, Suwon, South Korea · Department of AI Systems Engineering, Sungkyunkwan University, Suwon, South Korea · Department of Electrical and Computer Engineering, Sungkyunkwan University, Suwon, South Korea
Abstract
Retinal fundus imaging enables low-cost and scalable hypertension (HTN) screening, but HTN-related retinal cues are subtle, yielding high-variance predictions. Brain MRI provides stronger vascular and small-vessel-disease markers of HTN, yet it is expensive and rarely acquired alongside fundus images, resulting in modality-siloed datasets with disjoint MRI and fundus cohorts. We study this unpaired MRI-fundus regime and introduce Clinical Graph-Mediated Distillation (CGMD), a framework that transfers MRI-derived HTN knowledge to a fundus model without paired multimodal data. CGMD leverages shared structured biomarkers as a bridge by constructing a clinical similarity kNN graph spanning both cohorts. We train an MRI teacher, propagate its representations over the graph, and impute brain-informed representation targets for fundus patients. A fundus student is then trained with a joint objective combining HTN supervision, target distillation, and relational distillation. Experiments on our newly collected unpaired MRI-fundus-biomarker dataset show that CGMD consistently improves fundus-based HTN prediction over standard distillation and non-graph imputation baselines, with ablations confirming the importance of clinically grounded graph connectivity. Code is available at https://github.com/DillanImans/CGMD-unpaired-distillation.
Figures & tables
| Method | AUC | AUPRC | Sensitivity | Specificity | F1 |
|---|---|---|---|---|---|
| FDDM [ 18 ] | 0.715 0.051 | 0.879 0.013 | 0.815 0.179 | 0.679 0.182 | 0.709 0.032 |
| SimKD [ 1 ] | 0.732 0.134 | 0.869 0.075 | 0.705 0.234 | 0.786 0.192 | 0.761 0.155 |
| RKD [ 12 ] | 0.756 0.074 | 0.880 0.062 | 0.651 0.176 | 0.871 0.194 | 0.752 0.104 |
| FitNets [ 14 ] | 0.779 0.058 | 0.892 0.049 | 0.716 0.108 | 0.881 0.109 | 0.807 0.060 |
| KD [ 6 ] | 0.803 0.101 | 0.905 0.065 | 0.770 0.151 | 0.843 0.183 | 0.832 0.084 |
| CGMD | 0.855 0.127 | 0.937 0.055 | 0.728 0.123 | 0.933 0.149 | 0.826 0.087 |
| Distill | Smooth | Rel | AUC | AUPRC | Sensitivity | Specificity | F1 |
|---|---|---|---|---|---|---|---|
| ✓ | ✗ | ✗ | 0.830 0.133 | 0.927 0.060 | 0.766 0.202 | 0.876 0.137 | 0.831 0.136 |
| ✓ | ✓ | ✗ | 0.844 0.120 | 0.925 0.066 | 0.714 0.158 | 0.933 0.149 | 0.813 0.115 |
| ✓ | ✗ | ✓ | 0.815 0.116 | 0.907 0.064 | 0.820 0.151 | 0.790 0.170 | 0.853 0.090 |
| ✓ | ✓ | ✓ | 0.855 0.127 | 0.937 0.055 | 0.728 0.123 | 0.933 0.149 | 0.826 0.087 |