arXiv reaDer
一貫性の正則化により、ラベルノイズに対するロバスト性を向上させることができます
Consistency Regularization Can Improve Robustness to Label Noise
一貫性の正則化は、半教師あり学習と自己教師あり学習で一般的に使用される手法です。これは、観測されたトレーニングサンプルの近くでネットワークの予測が類似するように促す補助的な目的関数です。ヘンドリックス他(2020)は最近、そのような正則化が破損したデータにテスト時の堅牢性を自然にもたらし、キャリブレーションに役立つことを示しました。この論文は、ノイズの多いラベルに対するトレーニング時間のロバスト性に対する一貫性の正則化の関連性を経験的に研究しています。まず、ノイズの多いデータセットで標準のクロスエントロピー損失を使用してトレーニングされたネットワークの一貫性に関して、2つの興味深い有用な観察を行います。(i)ノイズの多いデータでトレーニングされたネットワークは、クリーンなデータでトレーニングされたネットワークよりも一貫性が低くなります。一貫性は、正しくラベル付けされたものよりも、ノイズの多いラベルが付けられたトレーニングデータポイントの周囲で大幅に減少します。次に、一貫性を促進する単純な損失関数により、合成(CIFAR-10、CIFAR-100)と実世界(WebVision)の両方のノイズ、およびさまざまなノイズレートとタイプのノイズにラベルを付けるために、モデルの堅牢性が向上することを示します。最先端の結果を実現します。
Consistency regularization is a commonly-used technique for semi-supervised and self-supervised learning. It is an auxiliary objective function that encourages the prediction of the network to be similar in the vicinity of the observed training samples. Hendrycks et al. (2020) have recently shown such regularization naturally brings test-time robustness to corrupted data and helps with calibration. This paper empirically studies the relevance of consistency regularization for training-time robustness to noisy labels. First, we make two interesting and useful observations regarding the consistency of networks trained with the standard cross entropy loss on noisy datasets which are: (i) networks trained on noisy data have lower consistency than those trained on clean data, and(ii) the consistency reduces more significantly around noisy-labelled training data points than correctly-labelled ones. Then, we show that a simple loss function that encourages consistency improves the robustness of the models to label noise on both synthetic (CIFAR-10, CIFAR-100) and real-world (WebVision) noise as well as different noise rates and types and achieves state-of-the-art results.
updated: Mon Oct 04 2021 08:15:08 GMT+0000 (UTC)
published: Mon Oct 04 2021 08:15:08 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アソシエイト