References & Citations
Computer Science > Machine Learning
Title: Deep Multi-task Representation Learning: A Tensor Factorisation Approach
(Submitted on 20 May 2016 (v1), last revised 16 Feb 2017 (this version, v2))
Abstract: Most contemporary multi-task learning methods assume linear models. This setting is considered shallow in the era of deep learning. In this paper, we present a new deep multi-task representation learning framework that learns cross-task sharing structure at every layer in a deep network. Our approach is based on generalising the matrix factorisation techniques explicitly or implicitly used by many conventional MTL algorithms to tensor factorisation, to realise automatic learning of end-to-end knowledge sharing in deep networks. This is in contrast to existing deep learning approaches that need a user-defined multi-task sharing strategy. Our approach applies to both homogeneous and heterogeneous MTL. Experiments demonstrate the efficacy of our deep multi-task representation learning in terms of both higher accuracy and fewer design choices.
Submission history
From: Yongxin Yang [view email][v1] Fri, 20 May 2016 15:05:58 GMT (273kb,D)
[v2] Thu, 16 Feb 2017 20:40:41 GMT (252kb,D)
Link back to: arXiv, form interface, contact.