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

Abstract

Learning high-quality multi-modal entity representations is an important goal of multi-modal knowledge graph (MMKG) representation learning, which can en- hance 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 over- look 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. Our code and data are available at https://github.com/zjukg/MoMoK.

Cite

Text

Zhang et al. "Multiple Heads Are Better than One: Mixture of Modality Knowledge Experts for Entity Representation Learning." International Conference on Learning Representations, 2025.

Markdown

[Zhang et al. "Multiple Heads Are Better than One: Mixture of Modality Knowledge Experts for Entity Representation Learning." International Conference on Learning Representations, 2025.](https://mlanthology.org/iclr/2025/zhang2025iclr-multiple/)

BibTeX

@inproceedings{zhang2025iclr-multiple,
  title     = {{Multiple Heads Are Better than One: Mixture of Modality Knowledge Experts for Entity Representation Learning}},
  author    = {Zhang, Yichi and Chen, Zhuo and Guo, Lingbing and Xu, Yajing and Hu, Binbin and Liu, Ziqi and Zhang, Wen and Chen, Huajun},
  booktitle = {International Conference on Learning Representations},
  year      = {2025},
  url       = {https://mlanthology.org/iclr/2025/zhang2025iclr-multiple/}
}