arrow
返回

Rethinking Resource Competition in Multi-Task Learning: From Shared Parameters to Shared Representation

delete2024-01-01
delete0
delete
OA
AI
D
Dayou Mao
Y
Yuhao Chen *
Y
Yifan Wu
M
Maximilian Gilles
A
Alexander Wong
DOI:10.1109/ACCESS.2024.3429281delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
The core idea of Multi-Task Learning (MTL) is to develop neural networks with a shared feature extraction backbone and multiple prediction heads, each capable of inferring a different task simultaneously. Parameters in the backbone contribute to all tasks while those in the prediction heads contribute to only one or fewer tasks. Challenges arise when multiple tasks compete for resource. Existing methods focus on resource competition in shared parameters and proposed explanatory factors of task conflicts, task dominance, and gradient stability. However the fundamental nature of MTL is still understudied. In this paper, instead of following the existing methodology research directions, we carry out large-scale empirical study and provide deeper insight on understanding MTL. In particular, instead of focusing on resource competition in the shared parameters in the backbone, we shift our attention to resource competition in the backbone output, which is the embedded representation that is shared by all prediction heads. We show that the existing explanatory factors display weak causal relationship with model performance. We propose a novel measurement, which we term Feature Disentanglement, and show that understanding MTL problems from the perspective of how the shared representation is leveraged by different prediction heads, is a more faithful and reliable way than that from the perspective of how supervision signals from different tasks are interfering in the shared parameters. Additionally, it has been a commonly employed technique to replace gradients w.r.t. shared parameters with gradients w.r.t. shared representation for reduced computation. We conduct a comprehensive study and show that unless a theoretical analysis could be developed, there is not general guarantee that this fast approximation technique would work in practice.
Keyword:
Task analysis
Feature extraction
Training
Magnetic heads
Computational modeling
Explainable AI
Optimization
Attention mechanism
computer vision
deep learning
explainable AI
machine learning
multi-objective optimization
multi-task learning
neural networks
representation
robotics

期刊

IEEE Access 封面图
IEEE Access
IF:
3.6
论文数:
9.8W
被引数:
29.4W

机构

H
Helmholtz Association
学者数:
13.2W
论文数: 10.7W
被引数: 145
U
University of Waterloo
学者数:
2.2W
论文数: 2.3W
被引数: 3.3W
引用论文

引用论文

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收藏
Risk Analysis within Feasibility Studies: An Application to Cost-Benefit Analysis for the Construction of a New Road
err2014-09-01
err0
PREAI
errVincenzo del Giudice; Alfredo Passeri; Francesca Torrieri; Pierfrancesco de Paola
err分享
err收藏
Spinal cord injury transiently alters Meissner's corpuscle density in the digit pads of macaque monkeys
err2019-02-22
err0
errOAAI
errMatthew Crowley; Alayna Lilak; Jamie Ahloy‐Dallaire; Corinna Darian‐Smith
err分享
err收藏
FDG-PET/CT Imaging Predicts Histopathologic Treatment Responses after Neoadjuvant Therapy in Adult Primary Bone Sarcomas
err2010-01-01
err0
errOAAI
errMatthias R. Benz; Johannes Czernin; William D. Tap; Jeffrey J. Eckardt; Leanne L. Seeger; Martin S. Allen-Auerbach; Sarah M. Dry; Michael E. Phelps; Wolfgang A. Weber; Fritz C. Eilber
err分享
err收藏
Mechanical Characterization of SRG Composites According to AC434
err2019-08-16
err0
PREAI
errDavide Campanini; Houman A. Hadad; Christian Carloni; Claudio Mazzotti; Antonio Nanni
err分享
err收藏
Search-Based Extraction of Component-Based Architecture from Object-Oriented Systems
err2008-01-01
err0
PREAI
errSylvain Chardigny; Abdelhak Seriai; Mourad Oussalah; Dalila Tamzalit
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收藏
学者 查看更多内容