Confidence Weighted Multitask Learning

  • Peng Yang King Abdullah University of Science and Technology
  • Peilin Zhao Tencent AI Lab
  • Jiayu Zhou Michigan State University
  • Xin Gao King Abdullah University of Science and Technology

Abstract

Traditional online multitask learning only utilizes the firstorder information of the datastream. To remedy this issue, we propose a confidence weighted multitask learning algorithm, which maintains a Gaussian distribution over each task model to guide online learning process. The mean (covariance) of the Gaussian Distribution is a sum of a local component and a global component that is shared among all the tasks. In addition, this paper also addresses the challenge of active learning on the online multitask setting. Instead of requiring labels of all the instances, the proposed algorithm determines whether the learner should acquire a label by considering the confidence from its related tasks over label prediction. Theoretical results show the regret bounds can be significantly reduced. Empirical results demonstrate that the proposed algorithm is able to achieve promising learning efficacy, while simultaneously minimizing the labeling cost.

Published
2019-07-17
Section
AAAI Technical Track: Machine Learning