arXiv reaDer
DeepSMOTE:不均衡なデータのためのディープラーニングとSMOTEの融合
DeepSMOTE: Fusing Deep Learning and SMOTE for Imbalanced Data
20年以上の進歩にもかかわらず、不均衡なデータは、現代の機械学習モデルにとって依然として重要な課題と見なされています。ディープラーニングの最近の進歩により、不均衡なデータ問題の重要性が拡大しています。この問題に対処するための2つの主なアプローチは、損失関数の変更とインスタンスのリサンプリングに基づいています。インスタンスのサンプリングは通常、生成的敵対的ネットワーク(GAN)に基づいており、モードの崩壊に悩まされる可能性があります。したがって、ディープラーニングモデルに合わせて特別に調整され、プロパティを保持しながら生の画像を処理でき、マイノリティクラスを強化してトレーニングセットのバランスをとることができる高品質の人工画像を生成できるオーバーサンプリング方法が必要です。 DeepSMOTEを提案します。これは、深層学習モデル用の新しいオーバーサンプリングアルゴリズムです。シンプルでありながらデザイン性に優れています。これは、次の3つの主要コンポーネントで構成されています。(i)エンコーダー/デコーダーフレームワーク。 (ii)SMOTEベースのオーバーサンプリング。 (iii)ペナルティ項で強化された専用の損失関数。 GANベースのオーバーサンプリングに対するDeepSMOTEの重要な利点は、DeepSMOTEが弁別器を必要とせず、情報が豊富で目視検査に適した高品質の人工画像を生成することです。 DeepSMOTEコードは、https://github.com/dd1github/DeepSMOTEで公開されています。
Despite over two decades of progress, imbalanced data is still considered a significant challenge for contemporary machine learning models. Modern advances in deep learning have magnified the importance of the imbalanced data problem. The two main approaches to address this issue are based on loss function modifications and instance resampling. Instance sampling is typically based on Generative Adversarial Networks (GANs), which may suffer from mode collapse. Therefore, there is a need for an oversampling method that is specifically tailored to deep learning models, can work on raw images while preserving their properties, and is capable of generating high quality, artificial images that can enhance minority classes and balance the training set. We propose DeepSMOTE - a novel oversampling algorithm for deep learning models. It is simple, yet effective in its design. It consists of three major components: (i) an encoder/decoder framework; (ii) SMOTE-based oversampling; and (iii) a dedicated loss function that is enhanced with a penalty term. An important advantage of DeepSMOTE over GAN-based oversampling is that DeepSMOTE does not require a discriminator, and it generates high-quality artificial images that are both information-rich and suitable for visual inspection. DeepSMOTE code is publicly available at: https://github.com/dd1github/DeepSMOTE
updated: Wed May 05 2021 21:49:37 GMT+0000 (UTC)
published: Wed May 05 2021 21:49:37 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アソシエイト