arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.03465cs.CVcs.AI

TBGC:面向多任务基础模型学习的任务级骨干导向梯度裁剪

TBGC: Task-level Backbone-Oriented Gradient Clip for Multi-Task Foundation Model Learning

Huazhong University of Science and Technology(华中科技大学)

Zelun Zhang, Xue Pan

AI总结:

针对多任务基础模型训练中的梯度偏差问题,提出任务级骨干导向梯度裁剪TBGC及多分支数据增强策略,在CVPR2023基础模型挑战赛中取得A榜第1、B榜第2的成绩。

摘要:

AllInOne训练范式以多任务学习的方式将广泛的任务压缩到一个统一模型中。然而,多任务学习中的优化比单任务学习更具挑战性,因为来自不同任务的梯度范数可能差异很大,使得骨干网络过度偏向某一特定任务。为了解决这一问题,我们提出了任务级骨干导向梯度裁剪范式,与普通的梯度裁剪方法相比,它有两个重点:1)对每个任务独立执行梯度裁剪。2)将每个任务生成的骨干梯度重新缩放到相同的范数尺度。基于实验结果,我们认为任务级骨干导向梯度裁剪范式可以在一定程度上缓解梯度偏差问题。我们还提出了一种新颖的多分支数据增强策略,其中相互冲突的增强被放置在不同的分支中。我们的方法已被证明是有效的,并最终在CVPR2023 Foundation Model Challenge的Leaderboard A中获得第1名,在Leaderboard B中获得第2名。值得注意的是,Leaderboard A评估了所有三个任务(检测、分割和细粒度分类),而Leaderboard B不评估分割任务,我们团队在该任务上具有巨大优势。

英文摘要:

The AllInOne training paradigm squeezes a wide range of tasks into a unified model in a multi-task learning manner. However, optimization in multi-task learning is more challenge than single-task learning, as the gradient norm from different tasks may vary greatly, making the backbone overly biased towards one specific task. To address this issue, we propose the task-level backbone-oriented gradient clip paradigm, compared with the vanilla gradient clip method, it has two points of emphasis:1) gradient clip is performed independently for each task. 2) backbone gradients generated from each task are rescaled to the same norm scale. Based on the experimental results, we argue that the task-level backbone-oriented gradient clip paradigm can relieve the gradient bias problem to some extent. We also propose a novel multi-branch data augmentation strategy where conflict augmentations are placed in different branches. Our approach has been shown to be effective and finally achieve 1st place in the Leaderboard A and 2nd place in the Leaderboard B of the CVPR2023 Foundation Model Challenge. It's worth noting that instead of evaluating all three tasks(detection, segmentation and fine-grained classification) in Leaderboard A, the segmentation task is not evaluated in Leaderboard B, in which our team has a huge advantage.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑