Towards Personalized Federated Multi-Scenario Multi-Task Recommendation

被引:1
作者
Ding, Yue [1 ]
Ji, Yanbiao [1 ]
Cai, Xun [1 ]
Xin, Xin [2 ]
Lu, Yuxiang [1 ]
Huang, Suizhi [1 ]
Liu, Chang [1 ]
Gao, Xiaofeng [1 ]
Murata, Tsuyoshi [3 ]
Lu, Hongtao [1 ]
机构
[1] Shanghai Jiao Tong Univ, Shanghai, Peoples R China
[2] Shandong Univ, Qingdao, Peoples R China
[3] Inst Sci Tokyo, Tokyo, Japan
来源
PROCEEDINGS OF THE EIGHTEENTH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, WSDM 2025 | 2025年
关键词
Multi-task Recommendation; Federated Learning; Collaborative Filtering;
D O I
10.1145/3701551.3703523
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In modern recommender systems, especially in e-commerce, predicting multiple targets such as click-through rate (CTR) and postview conversion rate (CTCVR) is common. Multi-task recommender systems are increasingly popular in both research and practice, as they leverage shared knowledge across diverse business scenarios to enhance performance. However, emerging real-world scenarios and data privacy concerns complicate the development of a unified multi-task recommendation model. In this paper, we propose PF-MSMTrec, a novel framework for personalized federated multi-scenario multi-task recommendation. In this framework, each scenario is assigned to a dedicated client utilizing the Multi-gate Mixture-of-Experts (MMoE) structure. To address the unique challenges of multiple optimization conflicts, we introduce a bottom-up joint learning mechanism. First, we design a parameter template to decouple the expert network parameters, distinguishing scenario-specific parameters as shared knowledge for federated parameter aggregation. Second, we implement personalized federated learning for each expert network during a federated communication round, using three modules: federated batch normalization, conflict coordination, and personalized aggregation. Finally, we conduct an additional round of personalized federated parameter aggregation on the task tower network to obtain prediction results for multiple tasks. Extensive experiments on two public datasets demonstrate that our proposed method outperforms state-of-the-art approaches. The source code and datasets will be released as open-source for public access.
引用
收藏
页码:429 / 438
页数:10
相关论文
共 52 条
[1]  
Agrawal N, 2024, AAAI CONF ARTIF INTE, P10775
[2]  
Ammad-ud-din M, 2019, Arxiv, DOI arXiv:1901.09888
[3]   A Contrastive Sharing Model for Multi-Task Recommendation [J].
Bai, Ting ;
Xiao, Yudong ;
Wu, Bin ;
Yang, Guojun ;
Yu, Hongyong ;
Nie, Jian-Yun .
PROCEEDINGS OF THE ACM WEB CONFERENCE 2022 (WWW'22), 2022, :3239-3247
[4]  
Cai Ruisi, 2023, P IEEE CVF C COMP VI, P5036
[5]   PEPNet: Parameter and Embedding Personalized Network for Infusing with Personalized Prior Information [J].
Chang, Jianxin ;
Zhang, Chenbin ;
Hui, Yiqun ;
Leng, Dewei ;
Niu, Yanan ;
Song, Yang ;
Gai, Kun .
PROCEEDINGS OF THE 29TH ACM SIGKDD CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, KDD 2023, 2023, :3795-3804
[6]  
Chen YQ, 2023, Arxiv, DOI [arXiv:2306.17465, DOI 10.48550/ARXIV.2306.17465]
[7]   DeepRec: On-device Deep Learning for Privacy-Preserving Sequential Recommendation in Mobile Commerce [J].
Han, Jialiang ;
Ma, Yun ;
Mei, Qiaozhu ;
Liu, Xuanzhe .
PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE 2021 (WWW 2021), 2021, :900-911
[8]   Adversarial Feature Translation for Multi-domain Recommendation [J].
Hao, Xiaobo ;
Liu, Yudan ;
Xie, Ruobing ;
Ge, Kaikai ;
Tang, Linyao ;
Zhang, Xu ;
Lin, Leyu .
KDD '21: PROCEEDINGS OF THE 27TH ACM SIGKDD CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2021, :2964-2973
[9]  
He CY, 2022, AAAI CONF ARTIF INTE, P6865
[10]   MetaBalance: Improving Multi-Task Recommendations via Adapting Gradient Magnitudes of Auxiliary Tasks [J].
He, Yun ;
Feng, Xue ;
Cheng, Cheng ;
Ji, Geng ;
Guo, Yunsong ;
Caverlee, James .
PROCEEDINGS OF THE ACM WEB CONFERENCE 2022 (WWW'22), 2022, :2205-2215