ICMLNov, 2017
GradNorm:用于深度多任务网络中自适应损失平衡的梯度归一化
GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks
Zhao Chen, Vijay Badrinarayanan, Chen-Yu Lee, Andrew Rabinovich
TL;DR本研究提出了一种使用梯度规范化的算法(GradNorm),在多任务网络的训练中能够自动平衡不同任务的训练误差,从而避免过拟合现象,提高准确率和性能,并且能够适用于各种网络结构、回归和分类任务,证明了梯度操作是解锁多任务学习潜力的关键之一。