Alternatives and similar repositories for GradNorm:

Users that are interested in GradNorm are comparing it to the libraries listed below

• brianlan / pytorch-grad-norm
  Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…
  ☆264Updated 2 years ago
• lorenmt / auto-lambda
  The Implementation of "Auto-Lambda: Disentangling Dynamic Task Relationships" [TMLR 2022].
  ☆134Updated 2 years ago
• AvivNavon / nash-mtl
  Official implementation of "Multi-Task Learning as a Bargaining Game" [ICML 2022]
  ☆223Updated 11 months ago
• WeiChengTseng / Pytorch-PCGrad
  Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"
  ☆337Updated 3 years ago
• Cranial-XIX / CAGrad
  Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)
  ☆122Updated last year
• EnnengYang / AdaTask
  AdaTask: A Task-Aware Adaptive Learning Rate Approach to Multi-Task Learning. AAAI, 2023.
  ☆28Updated last year
• hosseinshn / GradNorm
  This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018
  ☆176Updated 3 years ago
• Pengxin-Guo / Awesome-Multi-Task-Learning
  Paper List for Multi-Task Learning (focus on architectures and optimization for MTL)
  ☆45Updated last year
• mit-gfx / ContinuousParetoMTL
  [ICML 2020] Efficient Continuous Pareto Exploration in Multi-Task Learning
  ☆143Updated 3 years ago
• xiaofangxd / Multi-objective-optimization-and-Multi-task-Learning
  ☆70Updated 7 months ago
• xxgege / StableNet
  Official repository for CVPR21 paper "Deep Stable Learning for Out-Of-Distribution Generalization".
  ☆191Updated 3 years ago
• Cranial-XIX / FAMO
  Official PyTorch Implementation for Fast Adaptive Multitask Optimization (FAMO)
  ☆84Updated 11 months ago
• deebuls / deep_evidential_regression_loss_pytorch
  Implementation of Deep evidential regression paper
  ☆55Updated 4 years ago
• sunxm2357 / AdaShare
  AdaShare: Learning What To Share For Efficient Deep Multi-Task Learning
  ☆112Updated 4 years ago
• SamsungLabs / MTL
  ☆80Updated last year
• Mikoto10032 / AutomaticWeightedLoss
  Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning
  ☆612Updated 4 years ago
• SimonVandenhende / Multi-Task-Learning-PyTorch
  PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).
  ☆804Updated 3 years ago
• oscarkey / multitask-learning
  MSc group project: Reproduction of 'Multi-Task Learning using Uncertainty to Weigh Losses for Scene Geometry and Semantics'; A. Kendall, …
  ☆89Updated 5 years ago
• tianheyu927 / PCGrad
  Code for "Gradient Surgery for Multi-Task Learning"
  ☆320Updated 4 years ago
• hav4ik / Hydra
  Multi-Task Learning Framework on PyTorch. State-of-the-art methods are implemented to effectively train models on multiple tasks.
  ☆149Updated 5 years ago
• isl-org / MultiObjectiveOptimization
  Source code for Neural Information Processing Systems (NeurIPS) 2018 paper "Multi-Task Learning as Multi-Objective Optimization"
  ☆1,026Updated 7 months ago
• XzwHan / CARD
  Official PyTorch implementation for the paper "CARD: Classification and Regression Diffusion Models"
  ☆226Updated 2 years ago
• WeiHongLee / Awesome-Multi-Task-Learning
  An up-to-date list of works on Multi-Task Learning
  ☆340Updated 5 months ago
• Hui-Li / multi-task-learning-example-PyTorch
  ☆146Updated 3 years ago
• xxgege / GAM
  The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".
  ☆82Updated last year
• BorealisAI / MMoEEx-MTL
  PyTorch Implementation of the Multi-gate Mixture-of-Experts with Exclusivity (MMoEEx)
  ☆32Updated 3 years ago
• ranandalon / mtl
  Unofficial implementation of: Multi-task learning using uncertainty to weigh losses for scene geometry and semantics
  ☆552Updated 3 years ago
• haofanwang / awesome-mlp-papers
  Recent Advances in MLP-based Models (MLP is all you need!)
  ☆115Updated 2 years ago
• thuml / awesome-multi-task-learning
  2024 up-to-date list of DATASETS, CODEBASES and PAPERS on Multi-Task Learning (MTL), from Machine Learning perspective.
  ☆739Updated 4 months ago
• needylove / OrdinalEntropy
  The code of "Improving Deep Regression with Ordinal Entropy" in ICLR 2023
  ☆50Updated last year