arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2502.05282cs.CVcs.AI

医学图像密集对比表示学习中假阳性与假阴性问题的同胚先验

Homeomorphism Prior for False Positive and Negative Problem in Medical Image Dense Contrastive Representation Learning

Southeast University(东南大学);Centre de Recherche en Information Biomédicale Sino-Français (CRIBs)(中法生物医学信息研究中心(CRIBs));Jiangsu Provincial Joint International Research Laboratory of Medical Information Processing(江苏省医疗信息处理联合国际研究实验室);School of Instrument Science and Engineering, Southeast University(东南大学仪器科学与工程学院);Western University(西安大略大学);Case Western Reserve University(凯斯西储大学)

Yuting He, Boyu Wang, Rongjun Ge, Yang Chen, Guanyu Yang, Shuo Li

AI总结:

本文提出GEMINI方法,通过同胚先验和几何语义相似度解决医学图像密集对比表示学习中的假阳性与假阴性问题,在七个数据集上优于现有方法。

摘要:

密集对比表示学习(DCRL)极大地提高了图像密集预测任务的学习效率,显示出其在降低医学图像采集和密集标注的巨大成本方面的巨大潜力。然而,医学图像的特性使得对应关系发现不可靠,从而在DCRL中带来了大规模假阳性和假阴性(FP&N)对这一开放问题。在本文中,我们提出了GEoMetric vIsual deNse sImilarity(GEMINI)学习,它将同胚先验嵌入到DCRL中,并能够实现可靠的对应关系发现,从而实现有效的密集对比。我们提出了一种可变形同胚学习(DHL),它对医学图像的同胚性进行建模,并学习估计一种可变形映射,以在拓扑保持下预测像素的对应关系。它有效地缩小了配对搜索空间,并通过梯度驱动负样本对的隐式软学习。我们还提出了一种几何语义相似度(GSS),它提取特征中的语义信息来衡量对应关系学习的对齐程度。它将提高变形的学习效率和性能,可靠地构建正样本对。我们在实验中针对两个典型的表示学习任务实现了两种实用变体。我们在七个数据集上取得的优于现有方法的有前景的结果显示了我们的巨大优越性。我们将在配套链接上发布代码:https://github.com/YutingHe-list/GEMINI。

英文摘要:

Dense contrastive representation learning (DCRL) has greatly improved the learning efficiency for image-dense prediction tasks, showing its great potential to reduce the large costs of medical image collection and dense annotation. However, the properties of medical images make unreliable correspondence discovery, bringing an open problem of large-scale false positive and negative (FP&N) pairs in DCRL. In this paper, we propose GEoMetric vIsual deNse sImilarity (GEMINI) learning which embeds the homeomorphism prior to DCRL and enables a reliable correspondence discovery for effective dense contrast. We propose a deformable homeomorphism learning (DHL) which models the homeomorphism of medical images and learns to estimate a deformable mapping to predict the pixels' correspondence under topological preservation. It effectively reduces the searching space of pairing and drives an implicit and soft learning of negative pairs via a gradient. We also propose a geometric semantic similarity (GSS) which extracts semantic information in features to measure the alignment degree for the correspondence learning. It will promote the learning efficiency and performance of deformation, constructing positive pairs reliably. We implement two practical variants on two typical representation learning tasks in our experiments. Our promising results on seven datasets which outperform the existing methods show our great superiority. We will release our code on a companion link: https://github.com/YutingHe-list/GEMINI.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑