arXiv reaDer
限られたデータのハイパースペクトルリモートセンシング画像分類のためのベイズ畳み込みニューラルネットワーク
Bayesian Convolutional Neural Networks for Limited Data Hyperspectral Remote Sensing Image Classification
ハイパースペクトルリモートセンシング(HSRS)画像分類にディープニューラルネットワークを採用することは、困難な作業です。 HSRS画像は、高次元で多数のチャネルを持ち、チャネル間にかなりの冗長性があります。さらに、HSRS画像を分類するためのトレーニングデータは限られており、利用可能なトレーニングデータの量は他の分類タスクと比較してはるかに少なくなっています。これらの要因により、多くのパラメータを持つディープニューラルネットワークのトレーニングプロセスが複雑になり、従来のモデルと比較してもパフォーマンスが低下します。さらに、畳み込みニューラルネットワークは自信過剰な予測を生成します。これは、前述の問題を考慮すると非常に望ましくありません。この作業では、HSRS画像分類に、特殊なクラスのディープニューラルネットワーク、つまりベイジアンニューラルネットワーク(BNN)を使用します。私たちの知る限り、HSRS画像分類でBNNが使用されるのはこれが初めてです。 BNNは、本質的に不確実性の尺度を提供します。パヴィアセンター、サリナス、ボツワナのデータセットで広範な実験を行います。 BNNが標準の畳み込みニューラルネットワーク(CNN)および既製のランダムフォレスト(RF)よりも優れていることを示します。さらなる実験は、BNNがモデルの剪定に対してより安定していてロバストであり、予想される予測誤差がより高いサンプルの不確実性がより高いことを強調しています。
Employing deep neural networks for Hyperspectral remote sensing (HSRS) image classification is a challenging task. HSRS images have high dimensionality and a large number of channels with substantial redundancy between channels. In addition, the training data for classifying HSRS images is limited and the amount of available training data is much smaller compared to other classification tasks. These factors complicate the training process of deep neural networks with many parameters and cause them to not perform well even compared to conventional models. Moreover, convolutional neural networks produce over-confident predictions, which is highly undesirable considering the aforementioned problem. In this work, we use for HSRS image classification a special class of deep neural networks, namely a Bayesian neural network (BNN). To the extent of our knowledge, this is the first time that BNNs are used in HSRS image classification. BNNs inherently provide a measure for uncertainty. We perform extensive experiments on the Pavia Centre, Salinas, and Botswana datasets. We show that a BNN outperforms a standard convolutional neural network (CNN) and an off-the-shelf Random Forest (RF). Further experiments underline that the BNN is more stable and robust to model pruning, and that the uncertainty is higher for samples with higher expected prediction error.
updated: Mon May 30 2022 21:34:04 GMT+0000 (UTC)
published: Thu May 19 2022 00:02:16 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アソシエイト