arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.08032cs.AI

结合领域知识图谱的多模态电影类型分类:基于自监督注意力和对比学习

Incorporating Domain Knowledge Graph into Multimodal Movie Genre Classification with Self-Supervised Attention and Contrastive Learning

Southeast University(东南大学);Hohai University(河海大学)

Jiaqi Li, Guilin Qi, Chuanyi Zhang, Yongrui Chen, Yiming Tan, Chenlong Xia, Ye Tian

AI总结:

针对多模态电影类型分类中元数据关系未利用、注意力不可靠和特征判别性不足的问题,提出融合领域知识图谱、自监督注意力教师和类型质心对比学习的框架,在MM-IMDb和MM-IMDb 2.0上达到最优性能。

摘要:

多模态电影类型分类一直被视为一项要求很高的多标签分类任务,因为其涉及多种模态数据,如海报、剧情摘要、预告片和元数据。尽管现有工作在建模和组合每种模态方面取得了巨大进展,但它们仍面临三个问题:1)元数据中的群体关系未被利用,2)注意力分配不可靠,3)融合特征缺乏判别性。鉴于知识图谱已被证明包含丰富的信息,我们提出了一种新颖的框架,从多个角度利用知识图谱来解决上述问题。作为准备工作,元数据被处理成领域知识图谱。采用一种用于知识图谱嵌入的翻译模型来捕获实体之间的关系。首先,我们利用元数据中的群体关系从知识图谱中检索相关嵌入,然后将其与其他模态集成。接下来,我们引入一个基于自监督学习的注意力教师模块,以实现可靠的注意力分配。该模块学习知识图谱的分布并产生合理的注意力权重。最后,我们提出了一种基于类型质心锚定的对比学习模块,以增强融合特征的判别能力。锚点的嵌入空间从知识图谱中的类型实体初始化。为了验证我们框架的有效性,我们收集了一个比MM-IMDb数据集更大且更具挑战性的数据集,命名为MM-IMDb 2.0。在两个数据集上的实验结果表明,我们的模型优于最先进的方法。我们将在不久的将来发布代码。

英文摘要:

Multimodal movie genre classification has always been regarded as a demanding multi-label classification task due to the diversity of multimodal data such as posters, plot summaries, trailers and metadata. Although existing works have made great progress in modeling and combining each modality, they still face three issues: 1) unutilized group relations in metadata, 2) unreliable attention allocation, and 3) indiscriminative fused features. Given that the knowledge graph has been proven to contain rich information, we present a novel framework that exploits the knowledge graph from various perspectives to address the above problems. As a preparation, the metadata is processed into a domain knowledge graph. A translate model for knowledge graph embedding is adopted to capture the relations between entities. Firstly we retrieve the relevant embedding from the knowledge graph by utilizing group relations in metadata and then integrate it with other modalities. Next, we introduce an Attention Teacher module for reliable attention allocation based on self-supervised learning. It learns the distribution of the knowledge graph and produces rational attention weights. Finally, a Genre-Centroid Anchored Contrastive Learning module is proposed to strengthen the discriminative ability of fused features. The embedding space of anchors is initialized from the genre entities in the knowledge graph. To verify the effectiveness of our framework, we collect a larger and more challenging dataset named MM-IMDb 2.0 compared with the MM-IMDb dataset. The experimental results on two datasets demonstrate that our model is superior to the state-of-the-art methods. We will release the code in the near future.

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

↑