Giter VIP home page Giter VIP logo

awesome-multi-task-learning's Introduction

Awesome Multi-Task Learning

By Jialong Wu.

A curated list of datasets, codebases and papers on Multi-Task Learning (MTL), from Machine Learning perspective. I greatly appreciate those surveys below, which helped me a lot.

Please let me know if you find any mistakes or omissions! Your contribution is welcome!

Table of Contents

Awesome Multi-Task Learning

Survey

Benchmark & Dataset

Computer Vision

NLP

  • ✨ GLUE - General Language Understanding Evaluation [URL]
  • ✨ decaNLP - The Natural Language Decathlon: A Multitask Challenge for NLP [URL]
  • WMT Multilingual Machine Translation

RL & Robotics

  • ✨ MetaWorld [URL]
  • MTEnv [URL]

Graph

Recommendation

Codebase

  • General
    • LibMTL: LibMTL: A PyTorch Library for Multi-Task Learning
    • MALSAR: Multi-task learning via Structural Regularization (⚠️ Non-deep 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"
    • auto-lambda: The Implementation of "Auto-Lambda: Disentangling Dynamic Task Relationships"
    • 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

Consistency

Task Relationship Learning: Grouping, Tree (Hierarchy) & Cascading

Theory

Misc

awesome-multi-task-learning's People

Contributors

manchery avatar slyviacassell avatar

Stargazers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.