arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2507.23402cs.CVcs.AIcs.LG
收藏

AGA:一种用于结构化医学跨模态表征学习的自适应组对齐框架

AGA: An adaptive group alignment framework for structured medical cross-modal representation learning

School of Computing and Artificial Intelligence(计算机与人工智能学院);Southwest Jiaotong University(西南交通大学);Department of Gastroenterology(消化内科部);The Third People’s Hospital of Chengdu(成都第三人民医院)

Wei Li, Xun Gong, Jiao Li, Xiaobin Sun

AI总结:

针对医学视觉-语言预训练忽略报告结构、依赖大量硬负样本的问题,提出自适应组对齐框架AGA,通过双向分组、动态阈值门控及组对齐损失实现结构化跨模态表征学习,在多任务多设置下性能优异。

摘要:

从配对的图像和报告中学习医学视觉表征是表征学习中一个颇具前景的方向。然而,当前医学领域的视觉-语言预训练方法通常将临床报告简化为单个实体或碎片化的token,忽略了其固有结构。此外,对比学习框架通常依赖大量硬负样本,这对于小规模医学数据集而言并不现实。为应对这些挑战,我们提出了自适应组对齐(Adaptive Grouped Alignment,AGA)框架,这是一种从配对医学图像和报告中捕获结构化语义的新框架。AGA引入了基于稀疏相似度矩阵的双向分组机制。对于每对图像-报告,我们计算文本token与图像块之间的细粒度相似度。每个token选择匹配度最高的图像块组成视觉组,每个图像块选择相关性最高的token组成语言组。为实现自适应分组,我们设计了两个阈值门控模块,分别称为语言组阈值门(Language Grouped Threshold Gate)和视觉组阈值门(Vision Grouped Threshold Gate),可动态学习分组阈值。组表征通过基于相似度得分的加权平均计算得到。为使每个token与其组表征对齐,我们提出了实例感知组对齐损失,该损失在每个图像-文本对内部运算,无需外部负样本。最后,我们采用双向跨模态组对齐模块来增强视觉组表征与语言组表征之间的细粒度对齐。在公开和私有数据集上开展的大量实验表明,我们的方法在微调与零样本设置下的图像-文本检索和分类任务上均取得了优异性能。

英文摘要:

Learning medical visual representations from paired images and reports is a promising direction in representation learning. However, current vision-language pretraining methods in the medical domain often simplify clinical reports into single entities or fragmented tokens, ignoring their inherent structure. In addition, contrastive learning frameworks typically depend on large quantities of hard negative samples, which is impractical for small-scale medical datasets. To tackle these challenges, we propose Adaptive Grouped Alignment (AGA), a new framework that captures structured semantics from paired medical images and reports. AGA introduces a bidirectional grouping mechanism based on a sparse similarity matrix. For each image-report pair, we compute fine-grained similarities between text tokens and image patches. Each token selects its top-matching patches to form a visual group, and each patch selects its most related tokens to form a language group. To enable adaptive grouping, we design two threshold gating modules, called Language Grouped Threshold Gate and Vision Grouped Threshold Gate, which learn grouping thresholds dynamically. Group representations are computed as weighted averages based on similarity scores. To align each token with its group representation, we introduce an Instance Aware Group Alignment loss that operates within each image-text pair, removing the need for external negatives. Finally, a Bidirectional Cross-modal Grouped Alignment module is applied to enhance fine-grained alignment between visual and linguistic group representations. Extensive experiments on public and private datasets show that our method achieves strong performance on image-text retrieval and classification tasks under both fine-tuning and zero-shot settings.

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

↑