arXiv reaDer
MaxGNR: マルチタスク学習のための勾配対雑音比の最大化による動的重み付け戦略
MaxGNR: A Dynamic Weight Strategy via Maximizing Gradient-to-Noise Ratio for Multi-Task Learning
コンピューター ビジョンで関連するタスクをモデル化する場合、マルチタスク学習 (MTL) は、タスク間の本質的な関連性を捉える能力があるため、シングルタスク学習 (STL) よりも優れています。ただし、MTL では不十分なトレーニングの問題が発生する可能性があります。つまり、MTL の一部のタスクでは、STL と比較して最適でない状況が発生する可能性があります。一連の研究では、勾配ノイズが多すぎると STL のパフォーマンスが低下することが指摘されていますが、MTL シナリオでは、タスク間勾配ノイズ (ITGN) が各タスクの勾配ノイズの追加ソースとなり、最適化プロセス。本稿では、トレーニング不足の問題を引き起こす主な要因として ITGN を指摘します。勾配ノイズ比 (GNR) を定義して勾配ノイズの相対的な大きさを測定し、MaxGNR アルゴリズムを設計して、各タスクの GNR を最大化することで各タスクの ITGN 干渉を軽減します。 MaxGNR アルゴリズムを 2 つの標準画像 MTL データセット (NYUv2 と Cityscapes) で慎重に評価します。結果は、アルゴリズムが同一の実験条件下でベースラインよりも優れていることを示しています。
When modeling related tasks in computer vision, Multi-Task Learning (MTL) can outperform Single-Task Learning (STL) due to its ability to capture intrinsic relatedness among tasks. However, MTL may encounter the insufficient training problem, i.e., some tasks in MTL may encounter non-optimal situation compared with STL. A series of studies point out that too much gradient noise would lead to performance degradation in STL, however, in the MTL scenario, Inter-Task Gradient Noise (ITGN) is an additional source of gradient noise for each task, which can also affect the optimization process. In this paper, we point out ITGN as a key factor leading to the insufficient training problem. We define the Gradient-to-Noise Ratio (GNR) to measure the relative magnitude of gradient noise and design the MaxGNR algorithm to alleviate the ITGN interference of each task by maximizing the GNR of each task. We carefully evaluate our MaxGNR algorithm on two standard image MTL datasets: NYUv2 and Cityscapes. The results show that our algorithm outperforms the baselines under identical experimental conditions.
updated: Sat Feb 18 2023 14:50:45 GMT+0000 (UTC)
published: Sat Feb 18 2023 14:50:45 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト