通过选择性融合强弱教师,实现糖尿病视网膜病变轻量化筛查
OrthKD: Extracting Generalized Clinical Knowledge from Heterogeneous Teachers for Lightweight Deployment

- 强教师全监督+弱教师特征蒸馏,结合正交约束避免冗余
- 学生模型仅540万参数,在眼底图像上达0.885 QWK
- 适合资源受限设备部署,零样本迁移性能提升显著
在基层医疗中部署糖尿病视网膜病变(DR)筛查模型需要边缘高效且可靠的系统,以应对领域偏移。多教师知识蒸馏是自然的压缩策略,但现有方法假设所有教师提供等可信监督,这在实际中不成立:一个强教师(EfficientNet-B3,0.876 QWK)和一个弱教师(Swin-Base,0.830 QWK)互补,但后者输出仍可能误导学生。为此,我们提出OrthKD,一种选择性信任蒸馏框架:从强教师接收完整监督,仅从弱视觉变换器提取特征,并强制教师专属的学生投影正交,以促进互补而非冗余信息。该设计保留局部病灶精度,注入全局结构上下文,增强分布偏移鲁棒性。在132,049张眼底图像上,540万参数的MobileNetV3学生模型在EyePACS上达到0.885 QWK,零样本迁移至Messidor-2的QWK从0.507提升至0.728,同时保持优异的转诊AUC与校准性能。结果表明,选择性蒸馏异构教师可实现在资源受限设备上的实用化DR筛查。
原文摘要 · Abstract (English)
Deploying diabetic retinopathy (DR) screening models in primary care requires edge-efficient systems that remain accurate, safe, and reliable under domain shift. Multi-teacher knowledge distillation (KD) is a natural compression strategy, but existing approaches largely assume that all teachers provide equally trustworthy supervision. In our setting, this assumption fails: a strong CNN teacher (EfficientNet-B3, 0.876 QWK) and a weaker Transformer teacher (Swin-Base, 0.830 QWK) are complementary, yet the Transformer's logits can still mislead the student. We therefore propose OrthKD, a selective-trust distillation framework that transfers full supervision from the strong CNN, uses feature-only distillation from the weak ViT, and enforces orthogonality between teacher-specific student projections to encourage complementary rather than redundant evidence. This design preserves local lesion precision, injects global structural context, and improves robustness to distribution shift. On 132,049 retinal images, a 5.4M-parameter MobileNetV3 student reaches 0.885 QWK on EyePACS and improves zero-shot Messidor-2 performance from 0.507 to 0.728 QWK, while also achieving strong referral AUC and calibration. These results show that selectively distilling heterogeneous teachers can enable practical DR screening on resource-constrained devices.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。