96 Open Source Multi Task Learning Software Projects
Free and open source multi task learning code projects including engines, APIs, generators, and tools.
Multi Task Learning144 ⭐
Multi-task learning smile detection, age and gender classification on GENKI4k, IMDB-Wiki dataset.
Mnist Multitask33 ⭐
6️⃣6️⃣6️⃣ Reproduce ICLR '18 under-reviewed paper "MULTI-TASK LEARNING ON MNIST IMAGE DATASETS"
Mtl Homoscedastic Uncertainty45 ⭐
Multi Task Learning Implementation with Homoscedastic Uncertainty in Tensorflow
Naacl Mpqa Srl4orl10 ⭐
SRL4ORL: Improving Opinion Role Labeling Using Multi-Task Learning With Semantic Role Labeling
[unofficial] PyTorch Implementation of image quality assessment methods: IQA-CNN++ in ICIP2015 and IQA-CNN in CVPR2014
Yi-Min Chou, Yi-Ming Chan, Jia-Hong Lee, Chih-Yi Chiu, Chu-Song Chen, "Unifying and Merging Well-trained Deep Neural Networks for Inference Stage," International Joint Conference on Artificial Intelligence (IJCAI), 2018
Code for the paper "A Multi-lingual Multi-task Architecture for Low-resource Sequence Labeling" (ACL2018)
Multi Task Cs Lm10 ⭐
Code-Switching Language Modeling using Syntax-Aware Multi-Task Learning (CALCS 2018, ACL)
Hierarchical Skill Acquisition10 ⭐
Implementation of the Hierarchical and Interpretable Skill Acquisition in Multi-task Reinforcement Learning by Tianmin Shu, Caiming Xiong, and Richard Socher
The code for our proposed neural models which give state-of-the-art performance for emotion intensity detection in tweets.
Jia-Hong Lee, Yi-Ming Chan, Ting-Yen Chen, and Chu-Song Chen, "Joint Estimation of Age and Gender from Unconstrained Face Images using Lightweight Multi-task CNN for Mobile Applications," IEEE International Conference on Multimedia Information Processing and Retrieval, MIPR 2018
Tensorflow Asp Mtl24 ⭐
A Tensorflow implementation of Adversarial Shared-Private Model for Multi-Task Learning and Transfer Learning.
A tensorflow implementation of MKR (Multi-task Learning for Knowledge Graph Enhanced Recommendation)
Multi Task Image Classification65 ⭐
Multi-task learning for image classification implemented in PyTorch.
[STACOM 2018] Multi-task Learning for Left Atrial Segmentation on GE-MRI. This repository contains training and testing code for atrial segmentation challenge 2018.
Keras Mmoe518 ⭐
A TensorFlow Keras implementation of "Modeling Task Relationships in Multi-task Learning with Multi-gate Mixture-of-Experts" (KDD 2018)
Cups Rl34 ⭐
Customisable Unified Physical Simulations (CUPS) for Reinforcement Learning. Experiments run on the ai2thor environment (http://ai2thor.allenai.org/) e.g. using A3C, RainbowDQN and A3C_GA (Gated Attention multi-modal fusion) for Task-Oriented Language Grounding (tasks specified by natural language instructions) e.g. "Pick up the Cup or else"
🌊HMTL: Hierarchical Multi-Task Learning - A State-of-the-Art neural network model for several NLP tasks based on PyTorch and AllenNLP
Mask Yolo96 ⭐
Inspired from Mask R-CNN to build a multi-task learning, two-branch architecture: one branch based on YOLOv2 for object detection, the other branch for instance segmentation. Simply tested on Rice and Shapes. MobileNet supported.
The implementation of "Self-Supervised Generalisation with Meta Auxiliary Learning" [NeurIPS 2019].
Implementation of "Multi-task Deep Learning for Satellite Image Pansharpening and Segmentation"
Steven C. Y. Hung, Jia-Hong Lee, Timmy S. T. Wan, Chein-Hung Chen, Yi-Ming Chan and Chu-Song Chen. "Increasingly Packing Multiple Facial-Informatics Modules in A Unified Deep-Learning Model via Lifelong Learning" 2019 ACM on International Conference on Multimedia Retrieval
Official PyTorch implementation of paper <Hierarchical Multi-task Network For Race, Gender and Facial Attractiveness Recognition> (IEEE International Conference on Image Processing (ICIP) 2019)
Multi Task Conditional Attention Networks20 ⭐
A prototype version of our submitted paper: Conversion Prediction Using Multi-task Conditional Attention Networks to Support the Creation of Effective Ad Creatives.
Yanyusong Han47 ⭐
使用分层注意力机制 HAN + 多任务学习 解决 AI Challenger 细粒度用户评论情感分析 。https://challenger.ai/competition/fsauor2018
Multi Head Attention Labeller13 ⭐
Joint text classification on multiple levels with multiple labels, using a multi-head attention mechanism to wire two prediction tasks together.
Temporal Depth Segmentation18 ⭐
Source code (train/test) accompanying the paper entitled "Veritatem Dies Aperit - Temporally Consistent Depth Prediction Enabled by a Multi-Task Geometric and Semantic Scene Understanding Approach" in CVPR 2019 (https://arxiv.org/abs/1903.10764).
Merging Mobilenets For Multitask11 ⭐
Cheng-En Wu, Yi-Ming Chan and Chu-Song Chen "On Merging MobileNets for Efficient Multitask Inference", International Symposium on High-Performance Computer Architecture(HPCA) on Workshop on Energy Efficient Machine Learning and Cognitive Computing for Embedded Applications(EMC2), 2019
Lic2019 Dureader2.0 Rank2124 ⭐
Rank2 solution (no-BERT) for 2019 Language and Intelligence Challenge - DuReader2.0 Machine Reading Comprehension.
Thyroid Us26 ⭐
Evaluation of Thyroid Nodules Seen on Ultrasound Using Multi-task Deep Convolutional Neural Network (CNN)
Hav4ik Hydra112 ⭐
Multi-Task Learning Framework on PyTorch. State-of-the-art methods are implemented to effectively train models on multiple tasks.
An easy-to-use wrapper for work with dense per-pixel tasks in PyTorch (including multi-task learning)
[ACL 19] Fine-tuning Pre-Trained Transformer Language Models to Distantly Supervised Relation Extraction
Nddr Cnn Pytorch10 ⭐
PyTorch Implementation of "NDDR-CNN: Layerwise Feature Fusing in Multi-Task CNNs by Neural Discriminative Dimensionality Reduction"
Steven C. Y. Hung, Cheng-Hao Tu, Cheng-En Wu, Chien-Hung Chen, Yi-Ming Chan, and Chu-Song Chen, "Compacting, Picking and Growing for Unforgetting Continual Learning," Thirty-third Conference on Neural Information Processing Systems, NeurIPS 2019
a Fast, Flexible, Extensible and Easy-to-use NLP Large-scale Pretraining and Multi-task Learning Framework.
Multi-Task Learning package built with tensorflow 2 (Multi-Gate Mixture of Experts, Cross-Stitch, Ucertainty Weighting)
Taan Mtl13 ⭐
Adaptive Activation Network and Functional Regularization for Efficient and Flexible Deep Multi-Task Learning
EQTransformer, a python package for earthquake signal detection and phase picking using AI.
[CVPR2020] Multi-task Collaborative Network for Joint Referring Expression Comprehension and Segmentation, CVPR2020 (oral)
CH-SIMS: A Chinese Multimodal Sentiment Analysis Dataset with Fine-grained Annotations of Modality (ACL2020)
Ttnet Real Time Analysis System For Table Tennis Pytorch317 ⭐
Unofficial implementation of "TTNet: Real-time temporal and spatial video analysis of table tennis" (CVPR 2020)
Code for the SIGIR 2020 paper "A Unified Dual-view Model for Review Summarization and Sentiment Classification with Inconsistency Loss"
Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning
Multi Task Learning Pytorch389 ⭐
PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).
Menelaoskanakis Rcm26 ⭐
The official repository (in PyTorch) for the ECCV 2020 paper "Reparameterizing Convolutions for Incremental Multi-Task Learning without Task Interference".
MultiMix: Sparingly Supervised, Extreme Multitask Learning From Medical Images (ISBI 2021, MELBA 2021)
PhoNLP: A BERT-based multi-task learning model for part-of-speech tagging, named entity recognition and dependency parsing (NAACL 2021)
Codes and Datasets for our SIGIR 2021 Paper: "Understanding the Role of Affect Dimensions in Detecting Emotions from Tweets: A Multi-task Approach"
Fine Grained Or Not29 ⭐
Code release for Your “Flamingo” is My “Bird”: Fine-Grained, or Not (CVPR 2021 Oral)
Manchery Awesome Multi Task Learning77 ⭐
2021 up-to-date list of papers on Multi-Task Learning (MTL), from Machine Learning perspective.
Detection Attributes Fields19 ⭐
PyTorch implementation of "Detecting 32 Pedestrian Attributes for Autonomous Vehicles"
Har Joint Model24 ⭐
AROMA: A Deep Multi-Task Learning Based Simple and Complex Human Activity Recognition Method Using Wearable Sensors