multi-task-learning

There are 46 repositories under multi-task-learning topic.

namisan / mt-dnn
Multi-Task Deep Neural Networks for Natural Language Understanding
bert deep-learning machine-reading-comprehension microsoft multi-task-learning named-entity-recognition natural-language-understanding nlp pytorch ranking
Language:Python 2214
median-research-group / LibMTL
A PyTorch Library for Multi-Task Learning
deep-learning mmoe mtl multi-domain-learning multi-objective-optimization multi-task-learning multiobjective-optimization multitask-learning ple python pytorch
Language:Python 1850
alibaba / EasyRec
A framework for large scale recommendation algorithms.
autoint automl capsule-network ctr-prediction deepfm deepmatching din dlrm dssm eges esmm mind multi-task-learning online-learning pdn recommendation-algorithms recommender-system transformers-models wide-and-deep
Language:Python 1599
NVlabs / prismer
The implementation of "Prismer: A Vision-Language Model with Multi-Task Experts".
image-captioning language-model multi-modal-learning multi-task-learning vision-and-language vision-language-model vqa
Language:Python 1292
huggingface / hmtl
🌊HMTL: Hierarchical Multi-Task Learning - A State-of-the-Art neural network model for several NLP tasks based on PyTorch and AllenNLP
multi-task-learning natural-language-processing nlp pytorch
Language:Python 1187
megvii-research / PETR
[ECCV2022] PETR: Position Embedding Transformation for Multi-View 3D Object Detection & [ICCV2023] PETRv2: A Unified Framework for 3D Perception from Multi-Camera Images
3d-position-embedding multi-camera multi-task-learning object-detection segmentation
Language:Python 823
SimonVandenhende / Multi-Task-Learning-PyTorch
PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).
multi-task-learning computer-vision pytorch nyud pascal segmentation scene-understanding eccv2020
Language:Python 751
SinclairCoder / Instruction-Tuning-Papers
Reading list of Instruction-tuning. A trend starts from Natrural-Instruction (ACL 2022), FLAN (ICLR 2022) and T0 (ICLR 2022).
cross-task-generalization generalization instruction instruction-following large-language-models multi-task-learning natural-language-generation natural-language-processing natural-language-understanding prompt-learning
736
lambdaji / tf_repos
TensorFlow Script
ctr-prediction deep-learning deepfm din multi-task-learning pnn tensorflow wide-and-deep
Language:Python 681
drawbridge / keras-mmoe
A TensorFlow Keras implementation of "Modeling Task Relationships in Multi-task Learning with Multi-gate Mixture-of-Experts" (KDD 2018)
machine-learning deep-learning data-science deep-neural-networks kdd2018 keras tensorflow multi-task-learning mixture-of-experts
Language:Python 680
lorenmt / mtan
The implementation of "End-to-End Multi-Task Learning with Attention" [CVPR 2019].
attention-model multi-task-learning pytorch
Language:Python 654
mbs0221 / Multitask-Learning
Awesome Multitask Learning Resources
multi-task-learning deep-learning sparse-learning domain-adaptation transfer-learning meta-learning few-shot-learning convex-optimization federated-learning machine-learning
626
thuiar / MMSA
MMSA is a unified framework for Multimodal Sentiment Analysis.
dataset framework multi-task-learning multimodal-sentiment-analysis
Language:Python 607
codefuse-ai / MFTCoder
High Accuracy and efficiency multi-task fine-tuning framework for Code LLMs. This work has been accepted by KDD 2024.
multi-task-learning multi-task-fine-tuning multi-model-support customizable user-friendly
Language:Python 596
Manchery / awesome-multi-task-learning
2023 up-to-date list of DATASETS, CODEBASES and PAPERS on Multi-Task Learning (MTL), from Machine Learning perspective.
adapter awesome-list computer-vision deep-learning deep-neural-networks loss-strategy machine-learning multi-domain-learning multi-task-architecture multi-task-learning multi-task-optimization neural-language-processing transfer-learning
586
TTNet-Real-time-Analysis-System-for-Table-Tennis-Pytorch
maudzung / TTNet-Real-time-Analysis-System-for-Table-Tennis-Pytorch
Unofficial implementation of "TTNet: Real-time temporal and spatial video analysis of table tennis" (CVPR 2020)
dice-loss distributed-data-parallel event-detection event-spotting multi-task-learning pytorch segmentation spatial-video-analysis sport-analytics table-tennis tabletennis ttnet
Language:Python 565
JayYip / m3tl
BERT for Multitask Learning
bert cws encoder-decoder multi-task-learning multitask-learning named-entity-recognition ner nlp part-of-speech pretrained-models text-classification transformer word-segmentation
Language:Jupyter Notebook 544
Mikoto10032 / AutomaticWeightedLoss
Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning
auxiliary-tasks deep-learning multi-task multi-task-learning pytorch weigh-losses
Language:Python 538
xlang-ai / UnifiedSKG
[EMNLP 2022] Unifying and multi-tasking structured knowledge grounding with language models
natural-language-processing pytorch huggingface-transformers huggingface-datasets structured-knowledge-grounding semantic-parsing question-answering data-to-text text-generation fact-verification multi-task-learning prompt-learning nlp
Language:Python 538
AlaaLab / InstructCV
[ ICLR 2024 ] Official Codebase for "InstructCV: Instruction-Tuned Text-to-Image Diffusion Models as Vision Generalists"
diffusion-models generative-model multi-task-learning stable-diffusion text-to-image vision-language-model
Language:Python 515
OpenBMB / CPM-Live
Live Training for Open-source Big Models
deep-learning multi-task-learning natural-language-generation natural-language-understanding nlp parameter-efficient-learning pretrained-language-model natural-language-processing
Language:Python 510
CLUEbenchmark / pCLUE
pCLUE: 1000000+多任务提示学习数据集
chinese clue datasets multi-task-learning prompt-learning zero-shot-learning promptclue
Language:Jupyter Notebook 454
santi-pdp / pase
Problem Agnostic Speech Encoder
deep-learning multi-task-learning pytorch self-supervised-learning speech-processing unsupervised-learning waveform-analysis
Language:Python 439
LongmaoTeamTf / deep_recommenders
Deep Recommenders
deep-learning multi-task-learning ranking recommendation-system retrieval
Language:Python 325
hwwang55 / MKR
A tensorflow implementation of MKR (Multi-task Learning for Knowledge Graph Enhanced Recommendation)
recommender-systems knowledge-graph multi-task-learning
Language:Python 319
tianheyu927 / PCGrad
Code for "Gradient Surgery for Multi-Task Learning"
deep-learning deep-reinforcement-learning multi-task-learning
Language:Python 292
smousavi05 / EQTransformer
EQTransformer, a python package for earthquake signal detection and phase picking using AI.
attention-mechanism deep-learning detection earthquakes global lstm-neural-networks multi-task-learning neural-network phase-picking stead transformer
Language:Python 291
prismformore / Multi-Task-Transformer
Code of ICLR2023 paper "TaskPrompter: Spatial-Channel Multi-Task Prompting for Dense Scene Understanding" and ECCV2022 paper "Inverted Pyramid Multi-task Transformer for Dense Scene Understanding"
cityscapes cityscapes-3d computer-vision deep-learning depth-estimation eccv2022 human-parsing multi-task-learning nyudv2 pascal scene-understanding segmentation
Language:Python 281
WeiChengTseng / Pytorch-PCGrad
Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"
cifar-100 deep-learning deep-reinforcement-learning gradient-surgery mnist mulit-mnist multi-task-learning multi-task-reinforcement-learning multi-task-rl pytorch pytorch-pcgrad reimplementation reinforcement-learning rl
Language:Python 281
yhlleo / DeepSegmentor
A Pytorch implementation of DeepCrack and RoadNet projects.
centerline-detection crack-detection dataset deep-learning deep-neural-networks edge-detection image-segmentation multi-task-learning road-detection
Language:Python 255
WeiHongLee / Awesome-Multi-Task-Learning
An up-to-date list of works on Multi-Task Learning
multi-task-learning multi-domain-learning computer-vision machine-learning partially-supervised loss-weight deep-learning multi-task cross-domain-learning universal-representation
243
raunak-agarwal / instruction-datasets
All available datasets for Instruction Tuning of Large Language Models
alpaca chain-of-thought chatgpt dataset gpt-3 gpt-4 gpt4all instruction-tuning large-language-models multi-task-learning open-assistant sharegpt
228
AvivNavon / nash-mtl
Official implementation of "Multi-Task Learning as a Bargaining Game" [ICML 2022]
multi-task-learning multi-task game-theory bargaining-game icml-2022
Language:Python 197
PaddlePaddle / PALM
a Fast, Flexible, Extensible and Easy-to-use NLP Large-scale Pretraining and Multi-task Learning Framework.
nlp multi-task-learning pretrain-model transformers paddlepaddle baidu
Language:Python 176
truongnmt / multi-task-learning
Multi-task learning smile detection, age and gender classification on GENKI4k, IMDB-Wiki dataset.
deep-learning multi-task-learning image-classification tensorflow jupyter-notebook imdb gender-estimation age-estimation smile-detection genki neural-network dataset age gender gender-classification ipynb face imdb-wiki mtcnn
Language:Jupyter Notebook 175
Coopercoppers / PFN
EMNLP 2021 - A Partition Filter Network for Joint Entity and Relation Extraction
relation-extraction nlp multi-task-learning
Language:Python 168