0

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

Adaptive Task-to-Task Fusion Network (AdaTT) addresses challenges in multi-task learning by adaptively learning shared and task-specific knowledge through residual and gating mechanisms, demonstrating superior performance on benchmarks and recommendation datasets.

Year
2023
Venue
arXiv 2023
Authors
8
Hosting
Abstract onlyARXIV-DEFAULT

Cite

Notes

Only stored in your browser.

Attribution

Abstract & full text
arxiv.org/abs/2304.04959v2ARXIV-DEFAULT
TL;DR
Semantic Scholar
Attribution policy →

Abstract

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) 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.

Authors

8