arXiv reaDer
ノイズの多いラベルを使用した深層学習のための信頼性適応正則化
Confidence Adaptive Regularization for Deep Learning with Noisy Labels
ノイズの多いラベルに対するディープニューラルネットワークの記憶効果に関する最近の研究では、ネットワークは、誤ってラベル付けされたサンプルを記憶する前に、正しくラベル付けされたトレーニングサンプルに最初に適合します。この早期学習現象に動機付けられて、誤ってラベル付けされたサンプルの記憶を防ぐための新しい方法を提案します。モデル出力を使用して誤ってラベル付けされたサンプルを識別または無視する既存のアプローチとは異なり、元のモデルにインジケーターブランチを導入し、モデルが各サンプルの信頼値を生成できるようにします。信頼値は、正しくラベル付けされたサンプルに大きな信頼値を割り当て、誤ってラベル付けされたサンプルに小さな信頼値を割り当てることを学習した損失関数に組み込まれています。また、モデルのロバスト性をさらに向上させるために、補助正則化項を提案します。パフォーマンスを向上させるために、適切に設計されたターゲット推定戦略を使用して、ノイズの多いラベルを徐々に修正します。私たちは理論的分析を提供し、合成データセットと実世界のデータセットで実験を行い、私たちのアプローチが最先端の方法に匹敵する結果を達成することを示しています。
Recent studies on the memorization effects of deep neural networks on noisy labels show that the networks first fit the correctly-labeled training samples before memorizing the mislabeled samples. Motivated by this early-learning phenomenon, we propose a novel method to prevent memorization of the mislabeled samples. Unlike the existing approaches which use the model output to identify or ignore the mislabeled samples, we introduce an indicator branch to the original model and enable the model to produce a confidence value for each sample. The confidence values are incorporated in our loss function which is learned to assign large confidence values to correctly-labeled samples and small confidence values to mislabeled samples. We also propose an auxiliary regularization term to further improve the robustness of the model. To improve the performance, we gradually correct the noisy labels with a well-designed target estimation strategy. We provide the theoretical analysis and conduct the experiments on synthetic and real-world datasets, demonstrating that our approach achieves comparable results to the state-of-the-art methods.
updated: Sun Sep 05 2021 22:07:23 GMT+0000 (UTC)
published: Wed Aug 18 2021 15:51:25 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アソシエイト