Skip to content

2024 up-to-date list of DATASETS, CODEBASES and PAPERS on Multi-Task Learning (MTL), from Machine Learning perspective.

Notifications You must be signed in to change notification settings

thuml/awesome-multi-task-learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

51 Commits
 
 
 
 
 
 

Repository files navigation

Awesome Multi-Task Learning

By Jialong Wu.

I greatly appreciate those surveys below, which help me a lot. Please let me know if you find any mistakes or omissions!

Table of Contents

Survey

Benchmark

Computer Vision

NLP

  • GLUE - General Language Understanding Evaluation [URL]

RL & Robotics

Graph

Recommendation

Codebase

  • General

    • LibMTL: LibMTL: A PyTorch Library for Multi-Task Learning
  • Computer Vision

    • Multi-Task-Learning-PyTorch: PyTorch implementation of multi-task learning architectures
    • mtan: The implementation of "End-to-End Multi-Task Learning with Attention"
    • astmt: Attentive Single-tasking of Multiple Tasks
  • NLP

    • mt-dnn: Multi-Task Deep Neural Networks for Natural Language Understanding
  • Recommendation System

    • MTReclib: MTReclib provides a PyTorch implementation of multi-task recommendation models and common datasets.
  • RL

    • mtrl: Multi Task RL Baselines

Architecture

Hard Parameter Sharing

client-demo

Soft Parameter Sharing

Decoder-focused Model

Modulation & Adapters

Modularity, MoE, Routing & NAS

Task Representation

Others

Optimization

Loss & Gradient Strategy

Note:

  • We find that AdaLoss, IMTL-l, and Uncertainty are quite similiar in form.

Task Sampling

Adversarial Training

Pareto

Distillation

Task Relationship Learning: Grouping, Hierarchy & Cascading

Theory

Misc