核心概念
Aligning learning progress across tasks through adaptive group-based task interactions to improve overall multi-task performance while maintaining computational efficiency.
摘要
The paper proposes a novel multi-task optimization method called GO4Align that tackles the task imbalance issue in multi-task learning. The key idea is to align the learning progress across tasks by exploiting group-based task interactions.
The main contributions are:
- GO4Align recasts the task imbalance issue as a bi-level optimization problem, yielding an adaptive group risk minimization principle for multi-task optimization. This principle allocates weights over task losses at a group level to achieve learning progress alignment among relevant tasks.
- The paper develops a heuristic optimization pipeline in GO4Align to tractably achieve the principle, involving dynamical group assignment and risk-guided group indicators. This pipeline incorporates beneficial task interactions into the group assignments and exploits task correlations for multi-task alignment, improving the overall multi-task performance.
Experimental results show that GO4Align can outperform most existing state-of-the-art baselines in extensive benchmarks without sacrificing computational efficiency.
統計資料
The paper does not contain any explicit numerical data or statistics. The key results are presented in the form of performance comparisons on various multi-task learning benchmarks.
引述
"To improve the overall performance and maintain computational and memory efficiency, we propose Group Optimization for multi-task Alignment (GO4Align), a novel and effective loss-oriented method in MTO."
"GO4Align dynamically aligns learning progress across tasks by exploiting group-based task interactions for multi-task empirical risk minimization."
"The rationale behind this idea is that groupings can implicitly capture task correlations and encourage beneficial task interactions among relevant tasks."