通过结构先验提升糖尿病视网膜病变分级的泛化能力
Low-Rank Adaptive Structural Priors for Generalizable Diabetic Retinopathy Grading
- 引入低秩自适应结构先验,捕捉血管与病灶结构特征
- 在8个不同数据集上显著提升跨域诊断准确率
- 可无缝集成现有模型,适合医疗影像泛化研究者
糖尿病视网膜病变(DR)是糖尿病引发的主要致盲性眼病之一。深度学习在DR分级中广泛应用,但面对分布外数据时性能显著下降,主要因领域偏移问题。领域泛化(DG)被提出应对该挑战,但现有方法常忽略病灶特异性特征,导致精度不足。本文提出一种新方法,通过引入结构先验增强现有DG模型,其灵感源于DR分级高度依赖血管与病灶结构的事实。我们设计了低秩自适应结构先验(LoASP),一个即插即用框架,可无缝集成至现有DG模型。LoASP通过学习自适应结构表示,精准捕捉DR诊断中的复杂结构信息,显著提升泛化能力。在8个多样化的数据集上进行的大量实验验证了其在单源与多源领域场景下的有效性。可视化结果表明,学习到的结构先验与血管和病灶的精细结构高度对齐,揭示了其可解释性与诊断相关性。
原文摘要 · Abstract (English)
Diabetic retinopathy (DR), a serious ocular complication of diabetes, is one of the primary causes of vision loss among retinal vascular diseases. Deep learning methods have been extensively applied in the grading of diabetic retinopathy (DR). However, their performance declines significantly when applied to data outside the training distribution due to domain shifts. Domain generalization (DG) has emerged as a solution to this challenge. However, most existing DG methods overlook lesion-specific features, resulting in insufficient accuracy. In this paper, we propose a novel approach that enhances existing DG methods by incorporating structural priors, inspired by the observation that DR grading is heavily dependent on vessel and lesion structures. We introduce Low-rank Adaptive Structural Priors (LoASP), a plug-and-play framework designed for seamless integration with existing DG models. LoASP improves generalization by learning adaptive structural representations that are finely tuned to the complexities of DR diagnosis. Extensive experiments on eight diverse datasets validate its effectiveness in both single-source and multi-source domain scenarios. Furthermore, visualizations reveal that the learned structural priors intuitively align with the intricate architecture of the vessels and lesions, providing compelling insights into their interpretability and diagnostic relevance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。