Timezone: »

Do Current Multi-Task Optimization Methods in Deep Learning Even Help?
Derrick Xin · Behrooz Ghorbani · Dami Choi · Ankush Garg · Orhan Firat · Justin Gilmer
Event URL: https://openreview.net/forum?id=39rlc365gI5 »

Recent research has proposed a series of specialized optimization algorithms for deep multi-task models. It is often claimed that these multi-task optimization (MTO) methods yield solutions that are superior to the ones found by simply optimizing a weighted average of the task losses. In this paper, we perform large-scale experiments on a variety of language and vision tasks to examine the empirical validity of these claims. We show that, despite the added design and computational complexity of these algorithms, MTO methods do not yield any performance improvements beyond what is achievable via traditional optimization approaches. We highlight alternative strategies that consistently yield improvements to the performance profile and point out common training pitfalls that might cause suboptimal results. Finally, we outline challenges in reliably evaluating the performance of MTO algorithms and discuss potential solutions.

Author Information

Derrick Xin (Google)
Behrooz Ghorbani (Google Research)
Dami Choi (University of Toronto)
Ankush Garg (Google)
Orhan Firat (Google Research)
Justin Gilmer (Google Brain)

More from the Same Authors