arXiv reaDer
未知のラベルのノイズが多く不均衡なデータセットに対する全能の敵対的トレーニング
Omnipotent Adversarial Training for Unknown Label-noisy and Imbalanced Datasets
敵対的トレーニングは堅牢な深層学習において重要なトピックですが、コミュニティはその実際の使用法に注目していません。このペーパーでは、私たちが提案する全能敵対的トレーニング (OAT) を使用して、現実世界のアプリケーションの課題、つまり、不均衡でノイズの多いデータセットでモデルをトレーニングして、クリーンな精度と堅牢性を達成することを目指します。私たちの戦略は、トレーニング セット内のラベル ノイズとデータの不均衡に対処するための 2 つの革新的な方法論で構成されています。まず、敵対的トレーニング プロセスにオラクルを導入して、モデルが正しいデータ ラベルの条件付き分布を学習できるようにします。この慎重に設計されたオラクルは、敵対的トレーニングに正しいラベルの注釈を提供できます。さらに、データの不均衡の課題を克服するためのロジット調整の敵対的トレーニングを提案します。これは、モデルがベイズ最適分布を学習するのに役立ちます。当社の包括的な評価結果は、データの不均衡とラベル ノイズ シナリオの複雑な組み合わせの下で、OAT が他のベースラインよりも 20% 以上のクリーンな精度の向上と 10% 以上のロバストな精度の向上で優れていることを示しています。コードは https://github.com/GuanlinLee/OAT にあります。
Adversarial training is an important topic in robust deep learning, but the community lacks attention to its practical usage. In this paper, we aim to resolve a real-world application challenge, i.e., training a model on an imbalanced and noisy dataset to achieve high clean accuracy and robustness, with our proposed Omnipotent Adversarial Training (OAT). Our strategy consists of two innovative methodologies to address the label noise and data imbalance in the training set. We first introduce an oracle into the adversarial training process to help the model learn a correct data-label conditional distribution. This carefully-designed oracle can provide correct label annotations for adversarial training. We further propose logits adjustment adversarial training to overcome the data imbalance challenge, which can help the model learn a Bayes-optimal distribution. Our comprehensive evaluation results show that OAT outperforms other baselines by more than 20% clean accuracy improvement and 10% robust accuracy improvement under the complex combinations of data imbalance and label noise scenarios. The code can be found in https://github.com/GuanlinLee/OAT.
updated: Fri Jul 14 2023 07:09:57 GMT+0000 (UTC)
published: Fri Jul 14 2023 07:09:57 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アソシエイト