Confidence Weighted Multitask Learning

Authors

  • 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

DOI:

https://doi.org/10.1609/aaai.v33i01.33015636

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.

Downloads

Published

2019-07-17

How to Cite

Yang, P., Zhao, P., Zhou, J., & Gao, X. (2019). Confidence Weighted Multitask Learning. Proceedings of the AAAI Conference on Artificial Intelligence, 33(01), 5636-5643. https://doi.org/10.1609/aaai.v33i01.33015636

Issue

Section

AAAI Technical Track: Machine Learning