arXiv reaDer
ノイズにラベルを付けるためのロバスト性のための多目的補間トレーニング
Multi-Objective Interpolation Training for Robustness to Label Noise
標準のクロスエントロピー損失でトレーニングされたディープニューラルネットワークは、ノイズの多いラベルを記憶し、パフォーマンスを低下させます。この暗記を軽減するためのほとんどの研究は、新しいロバストな分類損失関数を提案しています。逆に、ラベルノイズの下での教師あり対照学習の動作を調査して、これらのシナリオで画像分類を改善する方法を理解します。特に、対照的な学習と分類を共同で活用する多目的補間トレーニング(MOIT)アプローチを提案します。標準的な対照学習がラベルノイズの存在下で低下することを示し、この動作を軽減するための補間トレーニング戦略を提案します。さらに、対照的な学習によって学習されたロバストな特徴表現を活用して、元のラベルとの不一致がノイズの多いサンプルを正確に識別するサンプルごとのソフトラベルを推定する新しいラベルノイズ検出方法を提案します。この検出により、ノイズの多いサンプルをラベルなしとして扱い、半教師ありの方法で分類器をトレーニングできます。さらに、検出されたクリーンなサンプルを微調整することでMOITを改良したMOIT +を提案します。ハイパーパラメータとアブレーションの研究は、私たちの方法の重要な要素を検証します。合成および実世界のノイズベンチマークに関する実験は、MOIT / MOIT +が最先端の結果を達成することを示しています。コードはhttps://git.io/JI40Xで入手できます。
Deep neural networks trained with standard cross-entropy loss memorize noisy labels, which degrades their performance. Most research to mitigate this memorization proposes new robust classification loss functions. Conversely, we explore the behavior of supervised contrastive learning under label noise to understand how it can improve image classification in these scenarios. In particular, we propose a Multi-Objective Interpolation Training (MOIT) approach that jointly exploits contrastive learning and classification. We show that standard contrastive learning degrades in the presence of label noise and propose an interpolation training strategy to mitigate this behavior. We further propose a novel label noise detection method that exploits the robust feature representations learned via contrastive learning to estimate per-sample soft-labels whose disagreements with the original labels accurately identify noisy samples. This detection allows treating noisy samples as unlabeled and training a classifier in a semi-supervised manner. We further propose MOIT+, a refinement of MOIT by fine-tuning on detected clean samples. Hyperparameter and ablation studies verify the key components of our method. Experiments on synthetic and real-world noise benchmarks demonstrate that MOIT/MOIT+ achieves state-of-the-art results. Code is available at https://git.io/JI40X.
updated: Tue Dec 08 2020 15:01:54 GMT+0000 (UTC)
published: Tue Dec 08 2020 15:01:54 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アソシエイト