近年、生成的敵対的ネットワーク(GAN)に基づくハイパースペクトル画像(HSI)分類が大きな進歩を遂げました。 GANベースの分類方法は、限られたトレーニングサンプルのジレンマをある程度軽減することができます。ただし、いくつかの研究では、既存のGANベースのHSI分類方法は、不均衡なトレーニングデータの問題の影響を大きく受けることが指摘されています。 GANの弁別子は常に矛盾し、偽のラベルを少数派クラスのサンプルに関連付けようとするため、分類のパフォーマンスが低下します。もう1つの重要な問題は、GANベースのメソッドのモード崩壊です。ジェネレータは、データ空間の狭い範囲内でのみサンプルを生成することができます。これは、GANベースのHSI分類方法の進歩を著しく妨げます。この論文では、HSI分類のためのAdaptive DropBlock-enhanced Generative Adversarial Networks(ADGAN)を提案しました。まず、不均衡なトレーニングデータの問題を解決するために、弁別器を単一の分類器になるように調整します。これは、それ自体と矛盾しません。次に、モード崩壊の問題を軽減するために、ジェネレーターとディスクリミネーターで採用されている正則化手法として、アダプティブDropBlock(AdapDrop)が提案されています。 AdapDropは、固定サイズの領域ではなく適応形状のドロップマスクを生成し、さまざまな形状の地面オブジェクトを処理する際のDropBlockの制限を緩和します。 3つのHSIデータセットでの実験結果は、提案されたADGANが最先端のGANベースの方法よりも優れたパフォーマンスを達成したことを示しました。私たちのコードはhttps://github.com/summitgao/HC_ADGANで入手できます
In recent years, hyperspectral image (HSI) classification based on generative adversarial networks (GAN) has achieved great progress. GAN-based classification methods can mitigate the limited training sample dilemma to some extent. However, several studies have pointed out that existing GAN-based HSI classification methods are heavily affected by the imbalanced training data problem. The discriminator in GAN always contradicts itself and tries to associate fake labels to the minority-class samples, and thus impair the classification performance. Another critical issue is the mode collapse in GAN-based methods. The generator is only capable of producing samples within a narrow scope of the data space, which severely hinders the advancement of GAN-based HSI classification methods. In this paper, we proposed an Adaptive DropBlock-enhanced Generative Adversarial Networks (ADGAN) for HSI classification. First, to solve the imbalanced training data problem, we adjust the discriminator to be a single classifier, and it will not contradict itself. Second, an adaptive DropBlock (AdapDrop) is proposed as a regularization method employed in the generator and discriminator to alleviate the mode collapse issue. The AdapDrop generated drop masks with adaptive shapes instead of a fixed size region, and it alleviates the limitations of DropBlock in dealing with ground objects with various shapes. Experimental results on three HSI datasets demonstrated that the proposed ADGAN achieved superior performance over state-of-the-art GAN-based methods. Our codes are available at https://github.com/summitgao/HC_ADGAN