Welcome to the AdaTT repository! This repository provides a PyTorch library for multitask learning, specifically focused on the models evaluated in the paper "AdaTT: Adaptive Task-to-Task Fusion Network for Multitask Learning in Recommendations" (KDD'23)".
This repository implements the following models:
- AdaTT [Paper]
- MMoE [Paper]
- Multi-level MMoE (an extension of MMoE)
- PLE [Paper]
- Cross-stitch [Paper]
- Shared-bottom [Paper]
To facilitate the integration and selection of these models, we have implemented a class called CentralTaskArch.
AdaTT is MIT-licensed.