深層学習タスクで使用されるデータセットのサイズが大きくなるにつれて、誤ってラベル付けされたデータに対して深層学習をロバストにするタスクであるノイジー ラベル問題が重要なタスクになりました。この論文では、テスト時間拡張(TTA)クロスエントロピーを使用したラベルノイズ選択とNoiseMixメソッドを使用した分類器学習を使用して、ノイズの多いラベルデータを学習する方法を提案します。ラベル ノイズ選択では、クロス エントロピーを測定してテスト時間の拡張トレーニング データを予測することにより、TTA クロス エントロピーを提案します。分類子の学習では、ノイズの多いラベル データとクリーンなラベル データのサンプルを混合することにより、MixUp および BalancedMix メソッドに基づく NoiseMix メソッドを提案します。 ISIC-18公開皮膚病変診断データセットの実験では、提案されたTTAクロスエントロピーは、ラベルノイズ選択プロセスでラベルノイズデータを検出する際に、従来のクロスエントロピーとTTAの不確実性よりも優れていました。さらに、提案されたNoiseMixは、分類性能において最先端の方法を上回っただけでなく、分類器学習におけるラベルノイズに対して最も堅牢性を示しました。
As the size of the dataset used in deep learning tasks increases, the noisy label problem, which is a task of making deep learning robust to the incorrectly labeled data, has become an important task. In this paper, we propose a method of learning noisy label data using the label noise selection with test-time augmentation (TTA) cross-entropy and classifier learning with the NoiseMix method. In the label noise selection, we propose TTA cross-entropy by measuring the cross-entropy to predict the test-time augmented training data. In the classifier learning, we propose the NoiseMix method based on MixUp and BalancedMix methods by mixing the samples from the noisy and the clean label data. In experiments on the ISIC-18 public skin lesion diagnosis dataset, the proposed TTA cross-entropy outperformed the conventional cross-entropy and the TTA uncertainty in detecting label noise data in the label noise selection process. Moreover, the proposed NoiseMix not only outperformed the state-of-the-art methods in the classification performance but also showed the most robustness to the label noise in the classifier learning.