摘要
arXiv:2405.05336v3 通告类型: replace-cross
摘要: 尽管当前的深度学习模型在处理不同域的图像方面非常有效,但在处理具有不同外观和内容的图像时仍面临挑战。我们介绍了SegCLR,这是一个多功能框架,旨在跨不同域分割图像,通过同时使用监督学习和对比学习来有效利用标记和未标记数据进行学习。我们通过一项全面的评估证明了SegCLR的优越性能,该评估涉及三种不同的临床数据集,其中包括3D视网膜光学相干断层扫描(OCT)图像,对不同网络配置的切片分割进行评估,并且在10种不同的网络初始化下进行验证。在无监督域适应的背景下,SegCLR的表现与在目标域上训练的监督上限模型相当。值得注意的是,我们发现SegCLR框架的分割性能受目标域未标记数据的丰富程度的影响不大,因此我们还提出了一种有效的SegCLR域泛化扩展,称为零样本域适应,这消除了对目标域信息的任何需求。这表明,在分割的标准监督训练中加入对比损失可以生成更优秀的模型,且这些模型能够更自然地适用于域内和域外测试数据。此外,我们还为在多个包含标记数据的域中实际部署SegCLR提出了一个实用的解决方案。因此,我们的框架在多域应用中推动了基于深度学习的分割技术的边界,无论数据是否可用、是标记的、未标记的还是不存在的。