Multi-Task Learning Network Optimization Based on Weight Adaptive

Multi-task learning has emerged as a significant topic in artificial intelligence research, where a singular network model performs numerous tasks. This approach simultaneously processes multiple related tasks and shared knowledge, enhancing model generalization while increasing efficiency. This methodology provides innovative solutions to complex real-world problems. However, the single-model-based approach for multi-task learning suffers from inter-task interference in practice. Therefore, the exploration of more efficient multi-task learning strategies, aimed at balancing the synergy and conflict among tasks and minimizing the dependence on computational resources, is pivotal for the field's future progress. We propose a strategy that autonomously adjusts both the parameters and structures of the model to alleviate gradient interference in multi-task learning. This method includes integrating a lightweight, weight-adaptive module that enhances the network's ability to process tasks by optimally balancing parameter sharing and isolation. This adaptation enables the model to share common features across tasks while allocating distinct spaces for each task, thereby reducing interference. Our extensive experimental validation indicates that our framework surpasses other multi-task learning approaches, achieving joint optimization of tasks more effectively. This enhancement not only bolsters performance but also maintains an equilibrium between accuracy and inference speed.

Authors

Institutions

Publication Details

Published
2024-08-16
DOI
https://doi.org/10.1109/besc64747.2024.10780578
Primary Topic
Advanced Computing and Algorithms
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Multi-Task Learning Network Optimization Based on Weight Adaptive

Zheng Cao, Yidong Li, Chuntao Ding, Zhuxi Zhang
Advanced Computing and Algorithms
article

Multi-Task Learning Network Optimization Based on Weight Adaptive

Zheng Cao, Yidong Li, Chuntao Ding, Zhuxi Zhang
article en

Abstract

Multi-task learning has emerged as a significant topic in artificial intelligence research, where a singular network model performs numerous tasks. This approach simultaneously processes multiple related tasks and shared knowledge, enhancing model generalization while increasing efficiency. This methodology provides innovative solutions to complex real-world problems. However, the single-model-based approach for multi-task learning suffers from inter-task interference in practice. Therefore, the exploration of more efficient multi-task learning strategies, aimed at balancing the synergy and conflict among tasks and minimizing the dependence on computational resources, is pivotal for the field's future progress. We propose a strategy that autonomously adjusts both the parameters and structures of the model to alleviate gradient interference in multi-task learning. This method includes integrating a lightweight, weight-adaptive module that enhances the network's ability to process tasks by optimally balancing parameter sharing and isolation. This adaptation enables the model to share common features across tasks while allocating distinct spaces for each task, thereby reducing interference. Our extensive experimental validation indicates that our framework surpasses other multi-task learning approaches, achieving joint optimization of tasks more effectively. This enhancement not only bolsters performance but also maintains an equilibrium between accuracy and inference speed.

Beijing Jiaotong University (CN)
National Natural Science Foundation of China
Openalex Percentile: Top 62%
Advanced Computing and Algorithms
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.