基于课程学习的医学图像分割无源域适应方法
Robust Source-Free Domain Adaptation for Medical Image Segmentation based on Curriculum Learning
- 设计易到难、源到目标的双课程学习框架,逐步优化模型适应过程
- 在视网膜和肠镜图像分割上超越现有方法,达到新最优性能
- 适合关注医疗数据隐私保护与高效域适应的研究者
近期研究提出无源域适应新方向,即在不使用源数据的情况下将模型适配至目标域,可缓解医学图像的数据隐私与安全问题。然而,现有方法多聚焦于目标数据伪标签优化,忽视学习过程设计。实际上,从源域到目标域的渐进式学习有助于知识迁移。为此,本文提出基于课程学习的框架LFC,包含易到难与源到目标两类课程。前者使模型从简单样本开始学习,并逐步提升样本难度以调整优化方向;后者稳定适应过程,保障模型从源域到目标域的平滑迁移。我们在公开的跨域视网膜与息肉分割数据集上评估该方法,实验结果表明其优于现有技术,达到新最佳性能。
原文摘要 · Abstract (English)
Recent studies have uncovered a new research line, namely source-free domain adaptation, which adapts a model to target domains without using the source data. Such a setting can address the concerns on data privacy and security issues of medical images. However, current source-free domain adaptation frameworks mainly focus on the pseudo label refinement for target data without the consideration of learning procedure. Indeed, a progressive learning process from source to target domain will benefit the knowledge transfer during model adaptation. To this end, we propose a curriculum-based framework, namely learning from curriculum (LFC), for source-free domain adaptation, which consists of easy-to-hard and source-to-target curricula. Concretely, the former curriculum enables the framework to start learning with `easy' samples and gradually tune the optimization direction of model adaption by increasing the sample difficulty. While, the latter can stablize the adaptation process, which ensures smooth transfer of the model from the source domain to the target. We evaluate the proposed source-free domain adaptation approach on the public cross-domain datasets for fundus segmentation and polyp segmentation. The extensive experimental results show that our framework surpasses the existing approaches and achieves a new state-of-the-art.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。