MMPG: MoE-based Adaptive Multi-Perspective Graph Fusion for Protein Representation Learning

Authors

  • Yusong Wang Guangdong Institute of Intelligence Science and Technology, Zhuhai, China
  • Jialun Shen Institute of Science Tokyo, Tokyo, Japan
  • Zhihao Wu Zhejiang University, Hangzhou, China
  • Yicheng Xu Institute of Science Tokyo, Tokyo, Japan
  • Shiyin Tan Institute of Science Tokyo, Tokyo, Japan
  • Mingkun Xu Guangdong Institute of Intelligence Science and Technology
  • Changshuo Wang University College London, London, U.K.
  • Zixing Song University of Cambridge, Cambridge, U.K.
  • Prayag Tiwari Halmstad University, Halmstad, Sweden

DOI:

https://doi.org/10.1609/aaai.v40i2.37096

Abstract

Graph Neural Networks (GNNs) have been widely adopted for Protein Representation Learning (PRL), as residue interaction networks can be naturally represented as graphs. Current GNN-based PRL methods typically rely on single-perspective graph construction strategies, which capture partial properties of residue interactions, resulting in incomplete protein representations. To address this limitation, we propose MMPG, a framework that constructs protein graphs from multiple perspectives and adaptively fuses them via Mixture of Experts (MoE) for PRL. MMPG constructs graphs from physical, chemical, and geometric perspectives to characterize different properties of residue interactions. To capture both perspective-specific features and their synergies, we develop an MoE module, which dynamically routes perspectives to specialized experts, where experts learn intrinsic features and cross-perspective interactions. We quantitatively verify that MoE automatically specializes experts in modeling distinct levels of interaction—from individual representations, to pairwise inter-perspective synergies, and ultimately to a global consensus across all perspectives. Through integrating this multi-level information, MMPG produces superior protein representations and achieves advanced performance on four different downstream protein tasks.

Published

2026-03-14

How to Cite

Wang, Y., Shen, J., Wu, Z., Xu, Y., Tan, S., Xu, M., … Tiwari, P. (2026). MMPG: MoE-based Adaptive Multi-Perspective Graph Fusion for Protein Representation Learning. Proceedings of the AAAI Conference on Artificial Intelligence, 40(2), 1240–1248. https://doi.org/10.1609/aaai.v40i2.37096

Issue

Section

AAAI Technical Track on Application Domains II