arXiv reaDer
Polarity-GANを使用したクラスの不均衡の除去:不確実性サンプリングアプローチ
Removing Class Imbalance using Polarity-GAN: An Uncertainty Sampling Approach
クラスの不均衡は、従来のモデルだけでなく、深層学習モデルの実際の分類問題においても難しい問題です。合成オーバーサンプリングなどの従来成功した対策は、深層学習モデルによって処理される複雑で構造化されたデータでは限られた成功しか収めていませんでした。この作業では、視覚データセットのクラスの不均衡を取り除くために、ジェネレータネットワークG、ディスクリミネータネットワークD、および分類器ネットワークCを備えた生成的敵対的ネットワーク(GAN)を使用することを提案します。発電機ネットワークは、安定させるためにオートエンコーダーで初期化されます。弁別器Dは、Gが不均衡なクラスのクラス分布に準拠していることを確認します。ジェネレータGがミニマックスゲームで弁別器Dと競合する従来の方法では、元のネットワークに分類器ネットワークをさらに追加することを提案します。現在、ジェネレータネットワークは、Discriminatorおよび導入した新しい分類器とミニマックスゲームで競争しようとしています。追加の条件が発電機ネットワークGに適用され、目的の不均衡クラスの凸包に点が生成されます。さらに、分類器Cとの敵対的ゲームの競合は、Gによって学習された条件付き分布をそれぞれのクラスの周辺に押し出し、クラスの不均衡の問題を補償します。実験的証拠は、この初期化がネットワークの安定したトレーニングをもたらすことを示しています。 FashionMNIST、MNIST、SVHN、ExDark、MVTec異常検出データセット、胸部X線データセットなどの極端な視覚分類タスクで最先端のパフォーマンスを実現します。
Class imbalance is a challenging issue in practical classification problems for deep learning models as well as for traditional models. Traditionally successful countermeasures such as synthetic over-sampling have had limited success with complex, structured data handled by deep learning models. In this work, we propose to use a Generative Adversarial Network (GAN) equipped with a generator network G, a discriminator network D and a classifier network C to remove the class-imbalance in visual data sets. The generator network is initialized with auto-encoder to make it stable. The discriminator D ensures that G adheres to class distribution of imbalanced class. In conventional methods, where Generator G competes with discriminator D in a min-max game, we propose to further add an additional classifier network to the original network. Now, the generator network tries to compete in a min-max game with Discriminator as well as the new classifier that we have introduced. An additional condition is enforced on generator network G to produce points in the convex hull of desired imbalanced class. Further the contention of adversarial game with classifier C, pushes conditional distribution learned by G towards the periphery of the respective class, compensating the problem of class imbalance. Experimental evidence shows that this initialization results in stable training of the network. We achieve state of the art performance on extreme visual classification task on the FashionMNIST, MNIST, SVHN, ExDark, MVTec Anomaly Detection dataset, Chest X-Ray dataset and others.
updated: Wed Dec 09 2020 09:40:07 GMT+0000 (UTC)
published: Wed Dec 09 2020 09:40:07 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アソシエイト