We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Deep Multi-task Representation Learning: A Tensor Factorisation Approach

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.
Comments: 9 pages, Accepted to ICLR 2017 Conference Track. This is a conference version of the paper. For the multi-domain learning part (not in this version), please refer to this https URL
Subjects: Machine Learning (cs.LG)
Cite as: arXiv:1605.06391 [cs.LG]
  (or arXiv:1605.06391v2 [cs.LG] for this version)

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.