PCGrad | Code for `` Gradient Surgery for Multi-Task Learning | Machine Learning library

PCGrad is a Python library typically used in Artificial Intelligence, Machine Learning, Deep Learning, Pytorch applications. PCGrad has no bugs, it has no vulnerabilities, it has a Permissive License and it has low support. However PCGrad build file is not available. You can download it from GitHub.

This repository contains code for Gradient Surgery for Multi-Task Learning in TensorFlow v1.0+ (PyTorch implementation forthcoming). PCGrad is a form of gradient surgery that projects a task’s gradient onto the normal plane of the gradient of any other task that has a conflicting gradient, which achieves substantial gains in efficiency and performance on a range of supervised multi-task learning and multi-task reinforcement learning domains. Moreover, it is model-agnostic and can be combined with previously-proposed multitask architectures for enhanced performance.