Trace Norm Regularised Deep Multi-Task LearningDownload PDF

02 Jun 2023 (modified: 17 Feb 2017)ICLR 2017 workshop submissionReaders: Everyone
TL;DR: Extending the idea of trace norm regularised matrix-based multi-task learning to deep nets
Abstract: We propose a framework for training multiple neural networks simultaneously. The parameters from all models are regularised by the tensor trace norm, so that each neural network is encouraged to reuse others' parameters if possible -- this is the main motivation behind multi-task learning. In contrast to many deep multi-task learning models, we do not predefine a parameter sharing strategy by specifying which layers have tied parameters. Instead, our framework considers sharing for all shareable layers, and the sharing strategy is learned in a data-driven way.
3 Replies