arXiv reaDer
変動係数を使用したマルチロス重み付け
Multi-Loss Weighting with Coefficient of Variations
機械学習とコンピュータービジョンの多くの興味深いタスクは、複数の損失の加重線形結合として定義された目的関数を最適化することによって学習されます。最終的なパフォーマンスは、これらの損失に対して正しい(相対的な)重みを選択することに敏感です。重みの適切なセットを見つけることは、多くの場合、広範なグリッド検索を使用して設定されるハイパーパラメーターのセットにそれらを採用することによって行われます。これは計算コストがかかります。本論文では、変動係数に基づく重み付けスキームを提案し、モデルのトレーニング中に観察された特性に基づいて重みを設定します。提案された方法は、損失のバランスをとるために不確実性の尺度を組み込んでおり、その結果、損失の重みは、別の(学習ベースの)最適化を必要とせずにトレーニング中に進化します。文献の多くの損失重み付け方法とは対照的に、単眼深度推定やセマンティックセグメンテーションなどの単一タスクの複数損失問題に焦点を当て、損失重み付けのマルチタスクアプローチがこれらの単一タスクでは機能しないことを示します。このアプローチの有効性は、複数のデータセットでの深度推定とセマンティックセグメンテーションについて経験的に示されています。
Many interesting tasks in machine learning and computer vision are learned by optimising an objective function defined as a weighted linear combination of multiple losses. The final performance is sensitive to choosing the correct (relative) weights for these losses. Finding a good set of weights is often done by adopting them into the set of hyper-parameters, which are set using an extensive grid search. This is computationally expensive. In this paper, we propose a weighting scheme based on the coefficient of variations and set the weights based on properties observed while training the model. The proposed method incorporates a measure of uncertainty to balance the losses, and as a result the loss weights evolve during training without requiring another (learning based) optimisation. In contrast to many loss weighting methods in literature, we focus on single-task multi-loss problems, such as monocular depth estimation and semantic segmentation, and show that multi-task approaches for loss weighting do not work on those single-tasks. The validity of the approach is shown empirically for depth estimation and semantic segmentation on multiple datasets.
updated: Tue Nov 10 2020 10:41:03 GMT+0000 (UTC)
published: Thu Sep 03 2020 14:51:19 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アソシエイト