English

Macro Graph of Experts for Billion-Scale Multi-Task Recommendation

Information Retrieval 2026-01-09 v4 Machine Learning

Abstract

Graph-based multi-task learning at billion-scale presents a significant challenge, as different tasks correspond to distinct billion-scale graphs. Traditional multi-task learning methods often neglect these graph structures, relying solely on individual user and item embeddings. However, disregarding graph structures overlooks substantial potential for improving performance. In this paper, we introduce the Macro Graph of Experts (MGOE) framework, the first approach capable of leveraging macro graph embeddings to capture task-specific macro features while modeling the correlations between task-specific experts. Specifically, we propose the concept of a Macro Graph Bottom, which, for the first time, enables multi-task learning models to incorporate graph information effectively. We design the Macro Prediction Tower to dynamically integrate macro knowledge across tasks. MGOE has been deployed at scale, powering multi-task learning for a leading billion-scale recommender system, Alibaba. Extensive offline experiments conducted on three public benchmark datasets demonstrate its superiority over state-of-the-art multi-task learning methods, establishing MGOE as a breakthrough in multi-task graph-based recommendation. Furthermore, online A/B tests confirm the superiority of MGOE in billion-scale recommender systems.

Keywords

Cite

@article{arxiv.2506.10520,
  title  = {Macro Graph of Experts for Billion-Scale Multi-Task Recommendation},
  author = {Hongyu Yao and Zijin Hong and Hao Chen and Zhiqing Li and Qijie Shen and Zuobin Ying and Qihua Feng and Huan Gong and Feiran Huang},
  journal= {arXiv preprint arXiv:2506.10520},
  year   = {2026}
}

Comments

Accepted to KDD2026

R2 v1 2026-07-01T03:12:54.111Z