用最优部分传输提升图文匹配精度与效率
Learning Image-Text Matching with Optimal Partial Transport
- 基于最优传输理论,用Sinkhorn迭代高效计算图文细粒度相似度
- 在Flickr30K和MS-COCO上均超越现有方法,召回率显著提升
- 支持局部到全局相似性建模,适合需要精准匹配的多模态任务
跨模态匹配是连接视觉与语言的基础任务,近年来受到广泛关注。尽管已有诸多方法试图量化图像-文本对之间的语义相关性,但往往难以兼顾高性能与高效率。本文提出跨模态Sinkhorn匹配网络(OMIT),在最优传输理论基础上,利用跨模态移动者距离精确计算细粒度视觉与文本片段间的相似性,并通过Sinkhorn迭代实现高效近似。为缓解冗余对齐问题,我们进一步将部分匹配融入OMIT,借助局部到全局相似性消除无关片段干扰。在Flickr30K和MS-COCO两个基准数据集上进行了广泛评估,OMIT在两项数据集上均表现出卓越性能,充分证明其有效性。通过可视化分析,揭示了OMIT内在的聚焦匹配倾向,进一步解释其优越性。代码已开源:https://github.com/ppanzx/OMIT。
原文摘要 · Abstract (English)
Cross-modal matching, a fundamental task in bridging vision and language, has recently garnered substantial research interest. Despite the development of numerous methods aimed at quantifying the semantic relatedness between image-text pairs, these methods often fall short of achieving both outstanding performance and high efficiency. In this paper, we propose the crOss-Modal sInkhorn maTching (OMIT) network as an effective solution to effectively improving performance while maintaining efficiency. Rooted in the theoretical foundations of Optimal Transport, OMIT harnesses the capabilities of Cross-modal Mover's Distance to precisely compute the similarity between fine-grained visual and textual fragments, utilizing Sinkhorn iterations for efficient approximation. To further alleviate the issue of redundant alignments, we seamlessly integrate partial matching into OMIT, leveraging local-to-global similarities to eliminate the interference of irrelevant fragments. We conduct extensive evaluations of OMIT on two benchmark image-text retrieval datasets, namely Flickr30K and MS-COCO. The superior performance achieved by OMIT on both datasets unequivocally demonstrates its effectiveness in cross-modal matching. Furthermore, through comprehensive visualization analysis, we elucidate OMIT's inherent tendency towards focal matching, thereby shedding light on its efficacy. Our code is publicly available at https://github.com/ppanzx/OMIT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。