AdaTT: Adaptive Task-to-Task Fusion Network for Multitask Learning in Recommendations

被引:8
作者
Li, Danwei [1 ]
Zhang, Zhengyu [2 ]
Yuan, Siyang [1 ]
Gao, Mingze [1 ]
Zhang, Weilin [1 ]
Yang, Chaofei [1 ]
Liu, Xi [1 ]
Yang, Jiyan [1 ]
机构
[1] Meta AI, Menlo Pk, CA 94025 USA
[2] Meta Platforms Inc, Menlo Pk, CA USA
来源
PROCEEDINGS OF THE 29TH ACM SIGKDD CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, KDD 2023 | 2023年
关键词
multi-task learning; neural network; recommender systems;
D O I
10.1145/3580305.3599769
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Multi-task learning (MTL) aims to enhance the performance and efficiency of machine learning models by simultaneously training them on multiple tasks. However, MTL research faces two challenges: 1) effectively modeling the relationships between tasks to enable knowledge sharing, and 2) jointly learning task-specific and shared knowledge. In this paper, we present a novel model called Adaptive Task-to-Task Fusion Network (AdaTT)(1) to address both challenges. AdaTT is a deep fusion network built with task-specific and optional shared fusion units at multiple levels. By leveraging a residual mechanism and a gating mechanism for task-to-task fusion, these units adaptively learn both shared knowledge and task-specific knowledge. To evaluate AdaTT's performance, we conduct experiments on a public benchmark and an industrial recommendation dataset using various task groups. Results demonstrate AdaTT significantly outperforms existing state-of-the-art baselines. Furthermore, our end-to-end experiments reveal that the model exhibits better performance compared to alternatives.
引用
收藏
页码:4370 / 4379
页数:10
相关论文
共 36 条
[1]  
Aoki Raquel, 2021, BIOKDD
[2]   Multitask learning [J].
Caruana, R .
MACHINE LEARNING, 1997, 28 (01) :41-75
[3]  
Chen Z, 2018, PR MACH LEARN RES, V80
[4]  
Chen Zhao, 2020, ADV NEUR IN, V33
[5]  
Collobert R., 2008, P 25 INT C MACH LEAR, P160, DOI [10.1145/1390156.1390177, DOI 10.1145/1390156.1390177]
[6]  
Deng L, 2013, IEEE INT NEW CIRC
[7]  
Dua D, 2017, UCI MACHINE LEARNING
[8]  
Guo P., 2020, P INT C MACH LEARN, P3854
[9]   Rank and Rate: Multi-task Learning for Recommender Systems [J].
Hadash, Guy ;
Shalom, Oren Sar ;
Osadchy, Rita .
12TH ACM CONFERENCE ON RECOMMENDER SYSTEMS (RECSYS), 2018, :451-454
[10]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778