arXiv reaDer
クラスレベルの勾配操作による一般化されたデータの重み付け
Generalized Data Weighting via Class-level Gradient Manipulation
ラベルノイズとクラスの不均衡は、実際のデータセットに共存する2つの主要な問題です。 2つの問題を軽減するために、最先端の方法では、少量のクリーンで偏りのないデータを活用して、各インスタンスの重みを再設定します。ただし、これらのメソッドは、各インスタンス内のクラスレベルの情報を見落とします。これらの情報は、パフォーマンスを向上させるためにさらに利用できます。この目的のために、この論文では、クラスレベルで勾配を操作することにより、ラベルノイズとクラスの不均衡を同時に軽減する一般化データ均等化(GDW)を提案します。具体的には、GDWは、連鎖律によって損失勾配をクラスレベルの勾配に展開し、各勾配のフローを個別に再重み付けします。このようにして、GDWは両方の問題で顕著なパフォーマンスの向上を実現します。パフォーマンスの向上に加えて、GDWは、インスタンスの重み付け方法と比較して余分な計算コストを導入することなく、クラスレベルの重みを効率的に取得します。具体的には、GDWは、中間勾配のみに依存するクラスレベルの重みに対して勾配降下ステップを実行します。さまざまな設定での広範な実験により、GDWの有効性が検証されます。たとえば、GDWは、CIFAR10の60%の均一ノイズ設定の下で、最先端の方法を2.56%上回っています。私たちのコードはhttps://github.com/GGchen1997/GDW-NIPS2021で入手できます。
Label noise and class imbalance are two major issues coexisting in real-world datasets. To alleviate the two issues, state-of-the-art methods reweight each instance by leveraging a small amount of clean and unbiased data. Yet, these methods overlook class-level information within each instance, which can be further utilized to improve performance. To this end, in this paper, we propose Generalized Data Weighting (GDW) to simultaneously mitigate label noise and class imbalance by manipulating gradients at the class level. To be specific, GDW unrolls the loss gradient to class-level gradients by the chain rule and reweights the flow of each gradient separately. In this way, GDW achieves remarkable performance improvement on both issues. Aside from the performance gain, GDW efficiently obtains class-level weights without introducing any extra computational cost compared with instance weighting methods. Specifically, GDW performs a gradient descent step on class-level weights, which only relies on intermediate gradients. Extensive experiments in various settings verify the effectiveness of GDW. For example, GDW outperforms state-of-the-art methods by 2.56% under the 60% uniform noise setting in CIFAR10. Our code is available at https://github.com/GGchen1997/GDW-NIPS2021.
updated: Fri Oct 29 2021 19:30:01 GMT+0000 (UTC)
published: Fri Oct 29 2021 19:30:01 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アソシエイト