arrow
Return

Robust Multitask Learning With Sample Gradient Similarity

delete2024-01-01
delete3
PRE
AI
X
Xinyu Peng
C
Cheng Chang
F
Fei‐Yue Wang
李
李力 (Li Li) *
DOI:10.1109/TSMC.2023.3315541delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Multitask learning has led to great success in many deep learning applications during the last decade. However, recent experiments have demonstrated that the performance of multitask learning depends on how to balance the relationship between different tasks. Therefore, many approaches have been proposed to adjust per-task gradient directions or design a more appropriate task reweighting scheme based on task-level statistics. In this article, we discuss how to boost the performance of multitask learning by using more fine-grained sample gradient information. To this end, we propose the concept of sample gradient similarity, which measures the agreement between the sample gradient for a task and the true gradient. Based on this concept, greater weight is assigned to more consistent tasks and more robust training samples to improve the training process of multitask learning. Extensive experimental results show that our proposed method outperforms the state-of-the-art algorithms on a series of challenging multitask datasets.
Keywords:
Deep learning
Automation
multitask learning
sample gradient
sample reweighting
task reweighting

Journal

IEEE Transactions on Cybernetics cover
IEEE Transactions on Cybernetics
IF:
10.5
Papers:
1.1W
Citations:
5.0W

Organization

T
tsinghua university
Scholars:
11.9W
Papers: 10.0W
Citations: 137
I
institute of automation, cas
Scholars:
2.2K
Papers: 2.1K
Citations: 2
C
chinese academy of sciences
Scholars:
56.7W
Papers: 45.0W
Citations: 704
researcher View more organizations
Cited Papers

Cited Papers

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
errShare
errSave
Sampling Methods for Efficient Training of Graph Convolutional Networks: A Survey
err2022-02-01
err57
errOAAI
errLiu, Xin; Yan, Mingyu; Deng, Lei; Li, Guoqi; Ye, Xiaochun; Fan, Dongrui
errShare
errSave
Multi-Modal Domain Adaptation Variational Auto-encoder for EEG-Based Emotion Recognition
err2022-09-01
err40
PREAI
errWang, Yixin; Qiu, Shuang; Li, Dan; Du, Changde; Lu, Bao-Liang; He, Huiguang
errShare
errSave
SwinFusion: Cross-domain Long-range Learning for General Image Fusion via Swin Transformer
err2022-07-01
err482
PREAI
errMa, Jiayi; Tang, Linfeng; Fan, Fan; Huang, Jun; Mei, Xiaoguang; Ma, Yong
errShare
errSave
Multitask TSK Fuzzy System Modeling by Jointly Reducing Rules and Consequent Parameters
err2021-07-01
err15
PREAI
errWang, Jun; Lin, Defu; Deng, Zhaohong; Jiang, Yizhang; Zhu, Jihua; Chen, Lei; Li, Zuoyong; Gong, Lejun; Wang, Shitong
errShare
errSave
Multifactorial Genetic Programming for Symbolic Regression Problems
err2020-11-01
err103
PREAI
errZhong, Jinghui; Feng, Liang; Cai, Wentong; Ong, Yew-Soon
errShare
errSave
Adversarial Learning and Self-Teaching Techniques for Domain Adaptation in Semantic Segmentation
err2020-09-01
err42
errOAAI
errMichieli, Umberto; Biasetton, Matteo; Agresti, Gianluca; Zanuttigh, Pietro
errShare
errSave
3-D LiDAR + Monocular Camera: An Inverse-Depth-Induced Fusion Framework for Urban Road Detection
err2018-09-01
err47
PREAI
errGu, Shuo; Lu, Tao; Zhang, Yigong; Alvarez, Jose M.; Yang, Jian; Kong, Hui
errShare
errSave
researcher View more