A regularization approach to learning task relationships in multitask learning

Yu Zhang, Dit Yan Yeung

Research output: Contribution to journalJournal articlepeer-review

99 Citations (Scopus)

Abstract

Multitask learning is a learning paradigm that seeks to improve the generalization performance of a learning task with the help of some other related tasks. In this article, we propose a regularization approach to learning the relationships between tasks inmultitask learning. This approach can be viewed as a novel generalization of the regularized formulation for single-task learning. Besides modeling positive task correlation, our approach-multitask relationship learning (MTRL)-can also describe negative task correlation and identify outlier tasks based on the same underlying principle. By utilizing a matrix-variate normal distribution as a prior on the model parameters of all tasks, our MTRL method has a jointly convex objective function. For efficiency, we use an alternating method to learn the optimal model parameters for each task as well as the relationships between tasks. We study MTRL in the symmetric multitask learning setting and then generalize it to the asymmetric setting as well.We also discuss some variants of the regularization approach to demonstrate the use of other matrix-variate priors for learning task relationships. Moreover, to gain more insight into our model, we also study the relationships between MTRL and some existing multitask learning methods. Experiments conducted on a toy problem as well as several benchmark datasets demonstrate the effectiveness of MTRL as well as its high interpretability revealed by the task covariance matrix.

Original languageEnglish
Article number12
JournalACM Transactions on Knowledge Discovery from Data
Volume8
Issue number3
DOIs
Publication statusPublished - Jun 2014

Scopus Subject Areas

  • Computer Science(all)

User-Defined Keywords

  • Multitask learning
  • Regularization framework
  • Task relationship

Fingerprint

Dive into the research topics of 'A regularization approach to learning task relationships in multitask learning'. Together they form a unique fingerprint.

Cite this