arrow
返回

Multitask learning with single gradient step update for task balancing

delete2022-01-01
delete18
delete
OA
AI
S
Sungjae Lee
Y
Youngdoo Son *
DOI:10.1016/j.neucom.2021.10.025delete
delete原文链接
delete分享
delete收藏
查看原文
摘要

摘要

En 中文
Multitask learning is a methodology to boost generalization performance and also reduce computational intensity and memory usage. However, learning multiple tasks simultaneously can be more difficult than learning a single task because it can cause imbalance among tasks. To address the imbalance problem, we propose an algorithm to balance between tasks at the gradient level by applying gradient-based meta- learning to multitask learning. The proposed method trains shared layers and task-specific layers sepa-rately so that the two layers with different roles in a multitask network can be fitted to their own pur -poses. In particular, the shared layer that contains informative knowledge shared among tasks is trained by employing single gradient step update and inner/outer loop training to mitigate the imbalance problem at the gradient level. We apply the proposed method to various multitask computer vision prob-lems and achieve state-of-the-art performance. CO 2021 Elsevier B.V. All rights reserved.
Keyword:
Convolution neural network
Deep learning
Gradient-based meta learning
Multitask learning
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

Neurocomputing 封面图
Neurocomputing
IF:
6.5
论文数:
2.5W
被引数:
6.5W

机构

D
Dongguk University
学者数:
8.2K
论文数: 9.3K
被引数: 1.0W
引用论文

引用论文

Adjuvant 5-FU-Based Chemoradiotherapy for Patients Undergoing R-1/R-2 Resections for Pancreatic Cancer
err2006-02-23
err0
PREAI
errHans G. Smeenk; Luca Incrocci; Geert Kazemier; Herman van Dekken; Khe T.C. Tran; Johannes Jeekel; Casper H.J. van Eijck
err分享
err收藏
Electrochemically assisted micro localized grafting of aptamers in a microchannel engraved in fluorinated thermoplastic polymer Dyneon THV
err2015-01-01
err0
PREAI
errC. Perréard; Y. Ladner; F. d'Orlyé; S. Descroix; V. Taniga; A. Varenne; F. Kanoufi; C. Slim; S. Griveau; F. Bedioui
err分享
err收藏
Individual distance in two species of macaque
err1964-04-01
err0
PREAI
errLeonard A. Rosenblum; I.Charles Kaufman; A.J. Stynes
err分享
err收藏
Robust texture analysis of multi-modal images using Local Structure Preserving Ranklet and multi-task learning for breast tumor diagnosis
err2017-10-01
err11
PREAI
errXi, Xiaoming; Xu, Hui; Shi, Hao; Zhang, Chunyun; Ding, Hong Yu; Zhang, Guang; Tang, Yuchun; Yin, Yilong
err分享
err收藏
Evolution of Organic Aerosols in the Atmosphere
err2009-12-11
err0
PREAI
errJ. L. Jimenez; M. R. Canagaratna; N. M. Donahue; A. S. H. Prevot; Q. Zhang; J. H. Kroll; P. F. DeCarlo; J. D. Allan; H. Coe; N. L. Ng; A. C. Aiken; K. S. Docherty; I. M. Ulbrich; A. P. Grieshop; A. L. Robinson; J. Duplissy; J. D. Smith; K. R. Wilson; V. A. Lanz; C. Hueglin; Y. L. Sun; J. Tian; A. Laaksonen; T. Raatikainen; J. Rautiainen; P. Vaattovaara; M. Ehn; M. Kulmala; J. M. Tomlinson; D. R. Collins; M. J. Cubison; J. Dunlea; J. A. Huffman; T. B. Onasch; M. R. Alfarra; P. I. Williams; K. Bower; Y. Kondo; J. Schneider; F. Drewnick; S. Borrmann; S. Weimer; K. Demerjian; D. Salcedo; L. Cottrell; R. Griffin; A. Takami; T. Miyoshi; S. Hatakeyama; A. Shimono; J. Y Sun; Y. M. Zhang; K. Dzepina; J. R. Kimmel; D. Sueper; J. T. Jayne; S. C. Herndon; A. M. Trimborn; L. R. Williams; E. C. Wood; A. M. Middlebrook; C. E. Kolb; U. Baltensperger; D. R. Worsnop
err分享
err收藏
Multitask learning多任务学习
err1997-01-01
err4.9K
errOAAI
errCaruana, R
err分享
err收藏
err分享
err收藏
学者 查看更多内容