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

Download:

Current browse context:

cs.CV

Change to browse by:

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 > Computer Vision and Pattern Recognition

Title: Maximum Roaming Multi-Task Learning

Abstract: Multi-task learning has gained popularity due to the advantages it provides with respect to resource usage and performance. Nonetheless, the joint optimization of parameters with respect to multiple tasks remains an active research topic. Sub-partitioning the parameters between different tasks has proven to be an efficient way to relax the optimization constraints over the shared weights, may the partitions be disjoint or overlapping. However, one drawback of this approach is that it can weaken the inductive bias generally set up by the joint task optimization. In this work, we present a novel way to partition the parameter space without weakening the inductive bias. Specifically, we propose Maximum Roaming, a method inspired by dropout that randomly varies the parameter partitioning, while forcing them to visit as many tasks as possible at a regulated frequency, so that the network fully adapts to each update. We study the properties of our method through experiments on a variety of visual multi-task data sets. Experimental results suggest that the regularization brought by roaming has more impact on performance than usual partitioning optimization strategies. The overall method is flexible, easily applicable, provides superior regularization and consistently achieves improved performances compared to recent multi-task learning formulations.
Comments: Accepted at the 35th AAAI Conference on Artificial Intelligence (AAAI 2021)
Subjects: Computer Vision and Pattern Recognition (cs.CV); Machine Learning (cs.LG); Machine Learning (stat.ML)
Journal reference: Proceedings of the AAAI Conference on Artificial Intelligence: 35(10), 9331-9341 (2021)
Cite as: arXiv:2006.09762 [cs.CV]
  (or arXiv:2006.09762v4 [cs.CV] for this version)

Submission history

From: Lucas Pascal [view email]
[v1] Wed, 17 Jun 2020 10:25:41 GMT (500kb,D)
[v2] Wed, 2 Dec 2020 10:55:28 GMT (508kb,D)
[v3] Thu, 22 Apr 2021 12:35:19 GMT (822kb,D)
[v4] Wed, 19 May 2021 09:20:37 GMT (822kb,D)

Link back to: arXiv, form interface, contact.