arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.00448cs.CVcs.MM
收藏

结合常识优化的风格Transformer用于图文检索

The style transformer with common knowledge optimization for image-text retrieval

Harbin Institute of Technology(哈尔滨工业大学);Peng Cheng Laboratory(鹏城实验室);Beijing University of Posts and Telecommunications(北京邮电大学)

Wenrui Li, Zhengyu Ma, Jinqiao Shi, Xiaopeng Fan

AI总结:

提出结合常识优化的风格Transformer网络(CKSTN),通过风格嵌入提取器和常识优化模块动态捕获多模态高级语义与常识,在MSCOCO和Flickr30K上取得优越性能且参数更少。

摘要:

图文检索将不同模态关联起来,因其卓越的研究价值和广泛的现实应用而备受关注。然而,现有大多数方法未能充分考虑高级语义关系(“风格嵌入”)以及来自多模态的常识。为此,我们提出了一种结合常识优化的风格Transformer网络(CKSTN)用于图文检索。其主要模块是常识适配器(CKA),包含风格嵌入提取器(SEE)和常识优化(CKO)模块。具体而言,SEE采用序列更新策略,有效连接SEE中不同阶段的特征。引入CKO模块以动态捕获不同模态的常识潜在概念。此外,为获得泛化的时序常识,我们提出序列更新策略,将SEE中不同层的特征与先前的公共特征单元有效融合。CKSTN在MSCOCO和Flickr30K数据集上的图文检索任务中展现了优于现有先进方法的性能。此外,CKSTN基于轻量级Transformer构建,凭借更优性能和更少参数,更便于实际场景应用。

英文摘要:

Image-text retrieval which associates different modalities has drawn broad attention due to its excellent research value and broad real-world application. However, most of the existing methods haven't taken the high-level semantic relationships ("style embedding") and common knowledge from multi-modalities into full consideration. To this end, we introduce a novel style transformer network with common knowledge optimization (CKSTN) for image-text retrieval. The main module is the common knowledge adaptor (CKA) with both the style embedding extractor (SEE) and the common knowledge optimization (CKO) modules. Specifically, the SEE uses the sequential update strategy to effectively connect the features of different stages in SEE. The CKO module is introduced to dynamically capture the latent concepts of common knowledge from different modalities. Besides, to get generalized temporal common knowledge, we propose a sequential update strategy to effectively integrate the features of different layers in SEE with previous common feature units. CKSTN demonstrates the superiorities of the state-of-the-art methods in image-text retrieval on MSCOCO and Flickr30K datasets. Moreover, CKSTN is constructed based on the lightweight transformer which is more convenient and practical for the application of real scenes, due to the better performance and lower parameters.

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

↑