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

用于一次性医学地标检测的结构引导自监督匹配

Structure-Guided Self-Supervised Matching for One-Shot Medical Landmark Detection

Department of Computer Science(计算机科学系);Renmin University of China(中国人民大学);University of Science and Technology of China(中国科学技术大学);School of Biomedical Engineering(生物医学工程学院);School of Artificial Intelligence(人工智能学院);Shanghai Jiaotong University(上海交通大学);Department of Orthopedics(骨科系);Affiliated Hospital of Guizhou Medical University(贵州医科大学附属医院);Yale University(耶鲁大学)

Qingsong Yao, Zhen Huang, Ao Wang, Rongsheng Wang, Hanxue Zhang, Jiangji Wang, S. Kevin Zhou

AI总结:

研究仅单张标注模板图像的一次性医学地标检测,提出SGB-Match框架,通过自监督从无标注图像对学习对应关系,经结构引导偏差改进对比目标,采用全局到局部设计,实验证明其性能优异。

摘要:

医学地标检测通常需要准确的专家注释,这既费力又难以跨解剖区域扩展。在这项工作中,我们研究了一种极端的注释高效设置,即只有一个带注释的模板图像可用。我们提出了SGB-Match,一种用于一次性医学地标检测的结构引导的从粗到细的自监督匹配框架。该框架首先从未标记的增强图像对中学习密集的解剖对应关系,然后通过特征匹配将地标定义从带注释的模板转移到每个目标图像。与标准的对比对应学习不同,在标准的对比对应学习中,负候选通过与结构无关的规则进行惩罚,我们在对比目标中引入了结构引导偏差。该偏差由相对距离和边缘感知解剖线索构建,并明确重新加权负梯度:与结构相关的附近候选被弱排斥,而与结构无关的远处或负候选被强烈抑制。结果,学习到的特征空间更好地保留了模板地标周围的局部解剖结构,并减少了重复纹理的混淆响应。我们进一步采用全局到局部的设计,其中全局编码器提供粗略的地标定位,局部编码器在裁剪区域中细化预测。在四个二维放射地标数据集上进行的广泛实验表明,SGB-Match在公共数据集和新收集的数据集中都实现了强大的一次性性能,并且始终受益于结构引导偏差和两阶段细化。

英文摘要:

Medical landmark detection usually requires accurate expert annotations, which are laborious and difficult to scale across anatomical regions. In this work, we study an extreme annotation-efficient setting where only a single annotated template image is available. We propose SGB-Match, a structure-guided coarse-to-fine self-supervised matching framework for one-shot medical landmark detection. The framework first learns dense anatomical correspondence from unlabeled augmented image pairs, and then transfers the landmark definition from the annotated template to each target image through feature matching. Different from standard contrastive correspondence learning, where negative candidates are penalized by a structure-agnostic rule, we introduce a structure-guided bias into the contrastive objective. The bias is constructed from relative distance and edge-aware anatomical cues, and explicitly reweights the negative gradients: nearby structure-relevant candidates are weakly repelled, while distant or structure-irrelevant negatives are strongly suppressed. As a result, the learned feature space better preserves local anatomical structures around template landmarks and reduces confusing responses from repeated textures. We further adopt a global-to-local design, where a global encoder provides coarse landmark localization and a local encoder refines the prediction in a cropped region. Extensive experiments on four 2D radiological landmark datasets demonstrate that SGB-Match achieves strong one-shot performance across both public and newly collected datasets and consistently benefits from both structure-guided bias and two-stage refinement.

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

↑