通过病灶级对齐提升医学图像报告联合学习效果
Improving Medical Visual Representation Learning with Pathological-level Cross-Modal Alignment and Correlation Exploration
- 提出病灶级跨模态对齐机制,无需额外标注
- 在多任务上达最新水平,分类准确率超基线3.2%
- 适合医学视觉表征与多模态模型研究者
通过图像-报告对联合学习来获取医学视觉表征,有助于缓解医疗数据稀缺问题。主要挑战源于报告的长文本特性及其复杂的语义病灶关系。以往工作多关注实例或词级别对齐,忽视病灶层级一致性。本文提出PLACE框架,通过病灶级跨模态对齐(PCMA)增强病理观察一致性,并引入视觉病灶观测提取器从局部标记中提取视觉病灶表示。该模块不依赖外部疾病标注,提升方法泛化性与鲁棒性。此外,设计代理任务以识别图像块间的相关性,丰富细粒度细节,对下游任务至关重要。实验表明,该框架在分类、图文检索、语义分割、目标检测和报告生成等多个任务上均达到新最优性能。
原文摘要 · Abstract (English)
Learning medical visual representations from image-report pairs through joint learning has garnered increasing research attention due to its potential to alleviate the data scarcity problem in the medical domain. The primary challenges stem from the lengthy reports that feature complex discourse relations and semantic pathologies. Previous works have predominantly focused on instance-wise or token-wise cross-modal alignment, often neglecting the importance of pathological-level consistency. This paper presents a novel framework PLACE that promotes the Pathological-Level Alignment and enriches the fine-grained details via Correlation Exploration without additional human annotations. Specifically, we propose a novel pathological-level cross-modal alignment (PCMA) approach to maximize the consistency of pathology observations from both images and reports. To facilitate this, a Visual Pathology Observation Extractor is introduced to extract visual pathological observation representations from localized tokens. The PCMA module operates independently of any external disease annotations, enhancing the generalizability and robustness of our methods. Furthermore, we design a proxy task that enforces the model to identify correlations among image patches, thereby enriching the fine-grained details crucial for various downstream tasks. Experimental results demonstrate that our proposed framework achieves new state-of-the-art performance on multiple downstream tasks, including classification, image-to-text retrieval, semantic segmentation, object detection and report generation. Code is available at https://github.com/Markin-Wang/PLACE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。