无需源数据,用梯度引导提升眼底图像分割精度
Grad-CL: Source Free Domain Adaptation with Gradient Guided Feature Disalignment
- 用梯度机制提取关键特征,优化伪标签质量
- 对比学习增强杯盘边界区分度,提升分割准确率
- 适合无源数据但需跨域适应的眼病诊断场景
准确分割视盘与视杯对青光眼等眼病的早期诊断至关重要。然而,基于某一数据集训练的分割模型在不同成像条件下的目标数据上常出现性能下降。为解决此问题,本文提出一种新型无源域适应框架Grad-CL,仅利用预训练源模型和未标注目标数据,在不访问原始源数据的情况下实现鲁棒适应。Grad-CL结合梯度引导伪标签精炼模块与基于余弦相似度的对比学习策略:第一阶段通过梯度机制提取显著类别特征,实现更精准的不确定性估计与原型估计,以修正噪声伪标签;第二阶段采用基于余弦相似度的对比损失,显式增强视杯与视盘梯度感知特征间的类间可分性。在多个具有挑战性的跨域眼底图像数据集上的实验表明,Grad-CL优于现有无监督及无源域适应方法,实现了更高的分割精度与更清晰的边界划分。项目与代码见https://visdomlab.github.io/GCL/
原文摘要 · Abstract (English)
Accurate segmentation of the optic disc and cup is critical for the early diagnosis and management of ocular diseases such as glaucoma. However, segmentation models trained on one dataset often suffer significant performance degradation when applied to target data acquired under different imaging protocols or conditions. To address this challenge, we propose \textbf{Grad-CL}, a novel source-free domain adaptation framework that leverages a pre-trained source model and unlabeled target data to robustly adapt segmentation performance without requiring access to the original source data. Grad-CL combines a gradient-guided pseudolabel refinement module with a cosine similarity-based contrastive learning strategy. In the first stage, salient class-specific features are extracted via a gradient-based mechanism, enabling more accurate uncertainty quantification and robust prototype estimation for refining noisy pseudolabels. In the second stage, a contrastive loss based on cosine similarity is employed to explicitly enforce inter-class separability between the gradient-informed features of the optic cup and disc. Extensive experiments on challenging cross-domain fundus imaging datasets demonstrate that Grad-CL outperforms state-of-the-art unsupervised and source-free domain adaptation methods, achieving superior segmentation accuracy and improved boundary delineation. Project and code are available at https://visdomlab.github.io/GCL/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。