arXiv reaDer
ノイズの多いラベルを使用した学習のための対照表現の学習について
On Learning Contrastive Representations for Learning with Noisy Labels
ディープニューラルネットワークは、softmaxクロスエントロピー(CE)損失を伴う、ノイズの多いラベルを簡単に記憶することができます。以前の研究では、CE損失にノイズに強い損失関数を組み込むことに焦点を当ててこの問題に対処しようとしました。ただし、暗記の問題は軽減されますが、堅牢でないCE損失のために残ります。この問題に対処するために、CE損失の下で分類器がラベルノイズを記憶するのが難しいデータのロバストな対照表現の学習に焦点を当てます。ラベルノイズが表現学習を支配しないノイズの多いデータ上でそのような表現を学習するために、新しい対照的な正則化関数を提案します。提案された正則化関数によって誘発される表現を理論的に調査することにより、学習された表現が真のラベルに関連する情報を保持し、破損したラベルに関連する情報を破棄することを明らかにします。さらに、私たちの理論的結果は、学習された表現がラベルノイズに対してロバストであることも示しています。この方法の有効性は、ベンチマークデータセットでの実験で実証されています。
Deep neural networks are able to memorize noisy labels easily with a softmax cross-entropy (CE) loss. Previous studies attempted to address this issue focus on incorporating a noise-robust loss function to the CE loss. However, the memorization issue is alleviated but still remains due to the non-robust CE loss. To address this issue, we focus on learning robust contrastive representations of data on which the classifier is hard to memorize the label noise under the CE loss. We propose a novel contrastive regularization function to learn such representations over noisy data where label noise does not dominate the representation learning. By theoretically investigating the representations induced by the proposed regularization function, we reveal that the learned representations keep information related to true labels and discard information related to corrupted labels. Moreover, our theoretical results also indicate that the learned representations are robust to the label noise. The effectiveness of this method is demonstrated with experiments on benchmark datasets.
updated: Sat Jul 23 2022 21:54:03 GMT+0000 (UTC)
published: Thu Mar 03 2022 15:58:05 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アソシエイト