医学视觉语言预训练新框架,提升细粒度任务性能
Representation Learning with Semantic-aware Instance and Sparse Token Alignments
- 基于图像-报告与片段-词元两级语义对齐,增强跨模态匹配
- 引入报告间相似性缓解误判负样本,提升表征质量
- 在少量标注数据下仍显著提升分类/分割/检测效果
医学对比视觉语言预训练(VLP)在下游任务中展现出巨大潜力。传统方法采用对比学习,将配对的图像-报告视为正例,未配对样本视为负例。然而,医疗数据中不同患者间的图像或报告可能存在显著相似性,硬性将所有未配对样本作为负例会破坏潜在语义结构,影响表征质量。本文提出多层级对齐框架SISTA,通过图像-报告和图像块-词元两个层面的语义对应关系,改进传统对比学习。具体地,利用报告间相似性消除错误负样本,并提出有效对齐图像块与相关词元的方法。实验表明,该框架在三个下游任务——图像分类、图像分割和目标检测——上均显著提升迁移性能,尤其在细粒度任务且标注数据有限时表现突出。代码与预训练模型将公开。
原文摘要 · Abstract (English)
Medical contrastive vision-language pre-training (VLP) has demonstrated significant potential in improving performance on downstream tasks. Traditional approaches typically employ contrastive learning, treating paired image-report samples as positives and unpaired ones as negatives. However, in medical datasets, there can be substantial similarities between images or reports from different patients. Rigidly treating all unpaired samples as negatives, can disrupt the underlying semantic structure and negatively impact the quality of the learned representations. In this paper, we propose a multi-level alignment framework, Representation Learning with Semantic-aware Instance and Sparse Token Alignments (SISTA) by exploiting the semantic correspondence between medical image and radiology reports at two levels, i.e., image-report and patch-word levels. Specifically, we improve the conventional contrastive learning by incorporating inter-report similarity to eliminate the false negatives and introduce a method to effectively align image patches with relevant word tokens. Experimental results demonstrate the effectiveness of the proposed framework in improving transfer performance across different datasets on three downstream tasks: image classification, image segmentation, and object detection. Notably, our framework achieves significant improvements in fine-grained tasks even with limited labeled data. Codes and pre-trained models will be made available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。