Multiple Heads are Better than One: Mixture of Modality Knowledge Experts for Entity Representation Learning

Yichi Zhang, Zhuo Chen, Lingbing Guo, Yajing Xu, Binbin Hu +3 more
2/12/2026

No official code found. Best implementation: zjukg/MoMoK (52 stars, last commit 2025-06-12).

Abstract

Learning high-quality multi-modal entity representations is an important goal of multi-modal knowledge graph (MMKG) representation learning, which can enhance reasoning tasks within the MMKGs, such as MMKG completion (MMKGC). The main challenge is to collaboratively model the structural information concealed in massive triples and the multi-modal features of the entities. Existing methods focus on crafting elegant entity-wise multi-modal fusion strategies, yet they overlook the utilization of multi-perspective features concealed within the modalities under diverse relational contexts. To address this issue, we introduce a novel framework with Mixture of Modality Knowledge experts (MoMoK for short) to learn adaptive multi-modal entity representations for better MMKGC. We design relation-guided modality knowledge experts to acquire relation-aware modality embeddings and integrate the predictions from multi-modalities to achieve joint decisions. Additionally, we disentangle the experts by minimizing their mutual information. Experiments on four public MMKG benchmarks demonstrate the outstanding performance of MoMoK under complex scenarios.

View on arXivView PDF

Code Implementations(1)

zjukg/MoMoKCommunity59%

[Paper][ICLR 2025] Multiple Heads are Better than One: Mixture of Modality Knowledge Experts for Entity Representation Learning

524PythonMay 28, 20241 years ago
contrastive-learningknowledge-graphknowledge-graph-completionknowledge-graph-embeddingsmixture-of-experts+3 more

Similar Papers

Cite this paper

@article{zhang2026multiple,
  title  = {Multiple Heads are Better than One: Mixture of Modality Knowledge Experts for Entity Representation Learning},
  author = {Yichi Zhang and Zhuo Chen and Lingbing Guo and Yajing Xu and Binbin Hu and Ziqi Liu and Wen Zhang and Hua-zeng Chen},
  year   = {2026},
  eprint = {2405.16869},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2405.16869},
  journal = {ICLR 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.16869.svg)](https://codeofpaper.com/paper/2405.16869)

Discussion